Add projects

backend/sdk/cliproxy/service_cooldown_store_test.go

@@ -0,0 +1,68 @@
+package cliproxy
+
+import (
+	"context"
+	"path/filepath"
+	"testing"
+
+	"github.com/router-for-me/CLIProxyAPI/v7/internal/config"
+	sdkAuth "github.com/router-for-me/CLIProxyAPI/v7/sdk/auth"
+	coreauth "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/auth"
+)
+
+type cooldownProviderTokenStore struct {
+	cooldownStore coreauth.CooldownStateStore
+}
+
+func (s *cooldownProviderTokenStore) List(context.Context) ([]*coreauth.Auth, error) {
+	return nil, nil
+}
+
+func (s *cooldownProviderTokenStore) Save(context.Context, *coreauth.Auth) (string, error) {
+	return "", nil
+}
+
+func (s *cooldownProviderTokenStore) Delete(context.Context, string) error {
+	return nil
+}
+
+func (s *cooldownProviderTokenStore) CooldownStateStore() coreauth.CooldownStateStore {
+	return s.cooldownStore
+}
+
+type serviceCooldownStateStore struct{}
+
+func (*serviceCooldownStateStore) Load(context.Context) ([]coreauth.CooldownStateRecord, error) {
+	return nil, nil
+}
+
+func (*serviceCooldownStateStore) Save(context.Context, []coreauth.CooldownStateRecord) error {
+	return nil
+}
+
+func TestResolveCooldownStateStoreUsesCapturedBackendProvider(t *testing.T) {
+	originalStore := sdkAuth.GetTokenStore()
+	t.Cleanup(func() {
+		sdkAuth.RegisterTokenStore(originalStore)
+	})
+
+	providedStore := &serviceCooldownStateStore{}
+	sdkAuth.RegisterTokenStore(&cooldownProviderTokenStore{cooldownStore: providedStore})
+	cfg := &config.Config{
+		AuthDir:            t.TempDir(),
+		SaveCooldownStatus: true,
+	}
+	service, errBuild := NewBuilder().
+		WithConfig(cfg).
+		WithConfigPath(filepath.Join(t.TempDir(), "config.yaml")).
+		Build()
+	if errBuild != nil {
+		t.Fatalf("Build() error = %v", errBuild)
+	}
+
+	sdkAuth.RegisterTokenStore(&cooldownProviderTokenStore{cooldownStore: &serviceCooldownStateStore{}})
+	got := service.resolveCooldownStateStore(cfg)
+	if got != providedStore {
+		t.Fatalf("resolveCooldownStateStore() = %T, want captured backend-provided store", got)
+	}
+}