mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-09-12 23:30:35 +00:00
feat(task): replace built-in task adaptors with a sandboxed JS plugin system (#7076)
This commit is contained in:
@@ -0,0 +1,33 @@
|
||||
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)
|
||||
}
|
||||
Reference in New Issue
Block a user