Files
new-api-custom/setting/system_setting/task_artifact_test.go
T
Codex bb7968b612
Publish Docker image (Multi-arch) / Build & push (amd64) (push) Failing after 9m20s
Release (Linux, macOS, Windows) / Linux Release (push) Failing after 55m7s
Publish Docker image (Multi-arch) / Build & push (arm64) (push) Canceled after 0s
Release (Linux, macOS, Windows) / macOS Release (push) Canceled after 0s
Release (Linux, macOS, Windows) / Windows Release (push) Canceled after 0s
Publish Docker image (Multi-arch) / Create multi-arch manifests (push) Canceled after 0s
feat: publish v1.0.0-rc.36-custom.1
2026-09-11 05:07:48 +08:00

34 lines
1.2 KiB
Go

package system_setting
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestLoadTaskArtifactAccessLimitsUsesPositiveEnvironmentValues(t *testing.T) {
t.Setenv(TaskArtifactInvalidRateLimitEnv, "17")
t.Setenv(TaskArtifactGlobalLimitEnv, "23")
t.Setenv(TaskArtifactIPLimitEnv, "11")
t.Setenv(TaskArtifactObjectLimitEnv, "7")
limits := LoadTaskArtifactAccessLimits()
assert.Equal(t, 17, limits.InvalidRatePerMinute)
assert.Equal(t, 23, limits.GlobalConcurrency)
assert.Equal(t, 11, limits.IPConcurrency)
assert.Equal(t, 7, limits.ObjectConcurrency)
}
func TestLoadTaskArtifactAccessLimitsFallsBackForInvalidValues(t *testing.T) {
t.Setenv(TaskArtifactInvalidRateLimitEnv, "0")
t.Setenv(TaskArtifactGlobalLimitEnv, "-1")
t.Setenv(TaskArtifactIPLimitEnv, "invalid")
t.Setenv(TaskArtifactObjectLimitEnv, "")
limits := LoadTaskArtifactAccessLimits()
assert.Equal(t, DefaultTaskArtifactInvalidRateLimitPerMinute, limits.InvalidRatePerMinute)
assert.Equal(t, DefaultTaskArtifactGlobalConcurrency, limits.GlobalConcurrency)
assert.Equal(t, DefaultTaskArtifactIPConcurrency, limits.IPConcurrency)
assert.Equal(t, DefaultTaskArtifactObjectConcurrency, limits.ObjectConcurrency)
}