From 812135b57eeca7aa45ff81507b0a3e2b8ff63142 Mon Sep 17 00:00:00 2001 From: erio Date: Sat, 7 Mar 2026 20:13:59 +0800 Subject: [PATCH] fix: add ClearAccountAffinity to remaining GatewayCache mock implementations Co-Authored-By: Claude Opus 4.6 --- backend/internal/service/gateway_multiplatform_test.go | 3 +++ backend/internal/service/gemini_multiplatform_test.go | 3 +++ backend/internal/service/openai_ws_state_store_test.go | 3 +++ 3 files changed, 9 insertions(+) diff --git a/backend/internal/service/gateway_multiplatform_test.go b/backend/internal/service/gateway_multiplatform_test.go index 2b5803c784..d43ac6fbbc 100644 --- a/backend/internal/service/gateway_multiplatform_test.go +++ b/backend/internal/service/gateway_multiplatform_test.go @@ -250,6 +250,9 @@ func (m *mockGatewayCacheForPlatform) GetAccountAffinityClientsBatch(_ context.C func (m *mockGatewayCacheForPlatform) GetAccountAffinityClientsWithScores(_ context.Context, _ int64, _ []int64, _ time.Duration) ([]AffinityClient, error) { return nil, nil } +func (m *mockGatewayCacheForPlatform) ClearAccountAffinity(_ context.Context, _ int64, _ []int64) error { + return nil +} type mockGroupRepoForGateway struct { groups map[int64]*Group diff --git a/backend/internal/service/gemini_multiplatform_test.go b/backend/internal/service/gemini_multiplatform_test.go index f53bf5c17c..7299261429 100644 --- a/backend/internal/service/gemini_multiplatform_test.go +++ b/backend/internal/service/gemini_multiplatform_test.go @@ -303,6 +303,9 @@ func (m *mockGatewayCacheForGemini) GetAccountAffinityClientsBatch(_ context.Con func (m *mockGatewayCacheForGemini) GetAccountAffinityClientsWithScores(_ context.Context, _ int64, _ []int64, _ time.Duration) ([]AffinityClient, error) { return nil, nil } +func (m *mockGatewayCacheForGemini) ClearAccountAffinity(_ context.Context, _ int64, _ []int64) error { + return nil +} // TestGeminiMessagesCompatService_SelectAccountForModelWithExclusions_GeminiPlatform 测试 Gemini 单平台选择 func TestGeminiMessagesCompatService_SelectAccountForModelWithExclusions_GeminiPlatform(t *testing.T) { diff --git a/backend/internal/service/openai_ws_state_store_test.go b/backend/internal/service/openai_ws_state_store_test.go index d7e5d69145..40cce0dd47 100644 --- a/backend/internal/service/openai_ws_state_store_test.go +++ b/backend/internal/service/openai_ws_state_store_test.go @@ -208,6 +208,9 @@ func (c *openAIWSStateStoreTimeoutProbeCache) GetAccountAffinityClientsBatch(_ c func (c *openAIWSStateStoreTimeoutProbeCache) GetAccountAffinityClientsWithScores(_ context.Context, _ int64, _ []int64, _ time.Duration) ([]AffinityClient, error) { return nil, nil } +func (c *openAIWSStateStoreTimeoutProbeCache) ClearAccountAffinity(_ context.Context, _ int64, _ []int64) error { + return nil +} func TestOpenAIWSStateStore_RedisOpsUseShortTimeout(t *testing.T) { probe := &openAIWSStateStoreTimeoutProbeCache{}