mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-08-31 02:41:34 +00:00
37 lines
1.5 KiB
Go
37 lines
1.5 KiB
Go
package relay
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/QuantumNous/new-api/constant"
|
|
pluginruntime "github.com/QuantumNous/new-api/pkg/jsplugin"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// TaskPlatformUnavailableError feeds the client-facing rejection when no
|
|
// adaptor serves a platform. The three shapes are user-actionable and must
|
|
// stay distinguishable: system switched off, plugin disabled, unknown platform.
|
|
func TestTaskPlatformUnavailableError(t *testing.T) {
|
|
t.Run("master switch off", func(t *testing.T) {
|
|
pluginruntime.DefaultRegistry.SetEnabled(false)
|
|
t.Cleanup(func() { pluginruntime.DefaultRegistry.SetEnabled(true) })
|
|
code, message := TaskPlatformUnavailableError(constant.TaskPlatform("17"))
|
|
assert.Equal(t, "task_plugin_system_disabled", code)
|
|
assert.Contains(t, message, "task plugin system is disabled")
|
|
})
|
|
|
|
t.Run("factory plugin disabled resolves legacy channel type to plugin key", func(t *testing.T) {
|
|
pluginruntime.DefaultRegistry.SetDisabledFactoryKeys([]string{"alibaba"})
|
|
t.Cleanup(func() { pluginruntime.DefaultRegistry.SetDisabledFactoryKeys(nil) })
|
|
code, message := TaskPlatformUnavailableError(constant.TaskPlatform("17"))
|
|
assert.Equal(t, "task_plugin_disabled", code)
|
|
assert.Contains(t, message, `task plugin "alibaba" is disabled`)
|
|
})
|
|
|
|
t.Run("unknown platform keeps the legacy message", func(t *testing.T) {
|
|
code, message := TaskPlatformUnavailableError(constant.TaskPlatform("no-such-platform"))
|
|
assert.Equal(t, "invalid_api_platform", code)
|
|
assert.Contains(t, message, "invalid api platform: no-such-platform")
|
|
})
|
|
}
|