Add projects

backend/sdk/cliproxy/auth/conductor_selection_cooldown_test.go

@@ -0,0 +1,60 @@
+package auth
+
+import (
+	"context"
+	"errors"
+	"testing"
+	"time"
+
+	cliproxyexecutor "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/executor"
+)
+
+func TestBuiltInSelectorCooldownErrorPreservesRouteModel(t *testing.T) {
+	t.Parallel()
+
+	const routeModel = "client-opus(high)"
+	next := time.Now().Add(time.Hour)
+	auth := &Auth{
+		ID:             "cooling-auth",
+		Unavailable:    true,
+		NextRetryAfter: next,
+		Quota: QuotaState{
+			Exceeded:      true,
+			NextRecoverAt: next,
+		},
+		ModelStates: map[string]*ModelState{
+			"other-model": {Status: StatusActive},
+		},
+	}
+
+	selectors := map[string]Selector{
+		"round-robin":          &RoundRobinSelector{},
+		"weighted-round-robin": &WeightedRoundRobinSelector{},
+		"fill-first":           &FillFirstSelector{},
+	}
+	for name, selector := range selectors {
+		t.Run(name, func(t *testing.T) {
+			t.Parallel()
+
+			_, errPick := selector.Pick(
+				context.Background(),
+				"mixed",
+				selectionArgForSelector(selector, routeModel),
+				cliproxyexecutor.Options{},
+				[]*Auth{auth},
+			)
+			if errPick == nil {
+				t.Fatal("Pick() error = nil, want model cooldown")
+			}
+
+			errPick = restoreModelCooldownErrorModel(errPick, routeModel)
+			var cooldownErr *modelCooldownError
+			if !errors.As(errPick, &cooldownErr) {
+				t.Fatalf("Pick() error = %T, want *modelCooldownError", errPick)
+			}
+			if cooldownErr.model != routeModel {
+				t.Fatalf("cooldown model = %q, want %q", cooldownErr.model, routeModel)
+			}
+		})
+	}
+}