feat: publish v1.0.0-rc.36-custom.1

relay/task_platform_error_test.go

@@ -0,0 +1,36 @@
+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")
+	})
+}