Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/service/composite_platform_test.go

@@ -0,0 +1,76 @@
+package service
+
+import (
+	"context"
+	"testing"
+
+	"github.com/Wei-Shaw/sub2api/internal/pkg/ctxkey"
+	"github.com/stretchr/testify/require"
+)
+
+func TestDetectModelPlatform(t *testing.T) {
+	tests := []struct {
+		name     string
+		model    string
+		platform string
+		ok       bool
+	}{
+		{name: "claude", model: "claude-sonnet-4-5", platform: PlatformAnthropic, ok: true},
+		{name: "anthropic prefix", model: "anthropic/claude-opus-4-5", platform: PlatformAnthropic, ok: true},
+		{name: "gpt", model: "gpt-5.1", platform: PlatformOpenAI, ok: true},
+		{name: "o series", model: "o3-mini", platform: PlatformOpenAI, ok: true},
+		{name: "embedding", model: "text-embedding-3-large", platform: PlatformOpenAI, ok: true},
+		{name: "gemini", model: "gemini-3-pro", platform: PlatformGemini, ok: true},
+		{name: "gemini models prefix", model: "models/gemini-2.5-flash", platform: PlatformGemini, ok: true},
+		{name: "learnlm", model: "learnlm-2.0-flash-experimental", platform: PlatformGemini, ok: true},
+		{name: "grok", model: "grok-4", platform: PlatformGrok, ok: true},
+		{name: "xai prefix", model: "xai/grok-4", platform: PlatformGrok, ok: true},
+		{name: "kimi", model: "kimi-k2-thinking", platform: PlatformKimi, ok: true},
+		{name: "moonshot prefix", model: "moonshot/moonshot-v1-32k", platform: PlatformKimi, ok: true},
+		{name: "zhipu", model: "glm-5.2", platform: PlatformZhipu, ok: true},
+		{name: "deepseek", model: "deepseek-v4-pro", platform: PlatformDeepseek, ok: true},
+		{name: "unknown", model: "llama-4-maverick", ok: false},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			platform, ok := DetectModelPlatform(tt.model)
+			require.Equal(t, tt.ok, ok)
+			require.Equal(t, tt.platform, platform)
+		})
+	}
+}
+
+func TestQuotaPlatformCompositeUsesResolvedOrForceOnly(t *testing.T) {
+	apiKey := &APIKey{Group: &Group{Platform: PlatformComposite}}
+
+	require.Equal(t, "", QuotaPlatform(context.Background(), apiKey))
+	require.Equal(t, PlatformGemini, QuotaPlatform(WithResolvedTargetPlatform(context.Background(), PlatformGemini), apiKey))
+	require.Equal(t, PlatformAntigravity, QuotaPlatform(context.WithValue(context.Background(), ctxkey.ForcePlatform, PlatformAntigravity), apiKey))
+
+	ctx := WithResolvedTargetPlatform(context.Background(), PlatformAnthropic)
+	ctx = context.WithValue(ctx, ctxkey.ForcePlatform, PlatformAntigravity)
+	require.Equal(t, PlatformAntigravity, QuotaPlatform(ctx, apiKey))
+}
+
+func TestCompositeGroupSchedulerHasAllCanonicalPlatformBuckets(t *testing.T) {
+	seen := make(map[string]struct{})
+	for _, bucket := range schedulerCanonicalBuckets(99) {
+		seen[bucket.Platform] = struct{}{}
+	}
+	platforms := make([]string, 0, len(seen))
+	for platform := range seen {
+		platforms = append(platforms, platform)
+	}
+	require.ElementsMatch(t,
+		[]string{PlatformAnthropic, PlatformGemini, PlatformOpenAI, PlatformAntigravity, PlatformGrok, PlatformKimi, PlatformZhipu, PlatformDeepseek},
+		platforms,
+	)
+}
+
+func TestCompositeConcretePlatformsIncludeCNProviders(t *testing.T) {
+	for _, platform := range []string{PlatformKimi, PlatformZhipu, PlatformDeepseek} {
+		require.True(t, isConcreteRequestPlatform(platform))
+		require.True(t, canCopyAccountsFromGroupPlatform(PlatformComposite, platform))
+	}
+}