53 lines
1.9 KiB
Go
53 lines
1.9 KiB
Go
|
package utils
|
||
|
|
|
||
|
|
import "testing"
|
||
|
|
|
||
|
|
func TestGetMaxSkillBundleSizeMB(t *testing.T) {
|
||
|
|
t.Run("supports the documented 20 GB knowledge upload ceiling", func(t *testing.T) {
|
||
|
|
t.Setenv("MAX_FILE_SIZE_MB", "20480")
|
||
|
|
if got := GetMaxFileSizeMB(); got != 20480 {
|
||
|
|
t.Fatalf("GetMaxFileSizeMB() = %d, want 20480", got)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
t.Run("caps an unsafe upload setting", func(t *testing.T) {
|
||
|
|
t.Setenv("MAX_FILE_SIZE_MB", "999999")
|
||
|
|
if got := GetMaxFileSizeMB(); got != maxFileSizeMBCeiling {
|
||
|
|
t.Fatalf("GetMaxFileSizeMB() = %d, want %d", got, maxFileSizeMBCeiling)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
t.Run("defaults above the knowledge upload cap", func(t *testing.T) {
|
||
|
|
t.Setenv("MAX_FILE_SIZE_MB", "")
|
||
|
|
t.Setenv("MAX_SKILL_BUNDLE_SIZE_MB", "")
|
||
|
|
if got := GetMaxSkillBundleSizeMB(); got != defaultMaxSkillBundleSizeMB {
|
||
|
|
t.Fatalf("GetMaxSkillBundleSizeMB() = %d, want %d", got, defaultMaxSkillBundleSizeMB)
|
||
|
|
}
|
||
|
|
if GetMaxFileSizeMB() != defaultMaxFileSizeMB {
|
||
|
|
t.Fatalf("knowledge cap must stay %d MB", defaultMaxFileSizeMB)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("honours an explicit skill cap above the knowledge cap", func(t *testing.T) {
|
||
|
|
t.Setenv("MAX_FILE_SIZE_MB", "50")
|
||
|
|
t.Setenv("MAX_SKILL_BUNDLE_SIZE_MB", "300")
|
||
|
|
if got := GetMaxSkillBundleSizeMB(); got != 300 {
|
||
|
|
t.Fatalf("GetMaxSkillBundleSizeMB() = %d, want 300", got)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("never below the knowledge upload cap", func(t *testing.T) {
|
||
|
|
t.Setenv("MAX_FILE_SIZE_MB", "200")
|
||
|
|
t.Setenv("MAX_SKILL_BUNDLE_SIZE_MB", "64")
|
||
|
|
if got := GetMaxSkillBundleSizeMB(); got != 200 {
|
||
|
|
t.Fatalf("GetMaxSkillBundleSizeMB() = %d, want 200", got)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("caps at the uncompressed archive ceiling", func(t *testing.T) {
|
||
|
|
t.Setenv("MAX_FILE_SIZE_MB", "50")
|
||
|
|
t.Setenv("MAX_SKILL_BUNDLE_SIZE_MB", "4096")
|
||
|
|
if got := GetMaxSkillBundleSizeMB(); got != maxSkillBundleSizeMBCeiling {
|
||
|
|
t.Fatalf("GetMaxSkillBundleSizeMB() = %d, want %d", got, maxSkillBundleSizeMBCeiling)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|