relay/task_platform_error_test.go

Rama main · réplica de lectura del repositorio original.

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")
	})
}