From ee01f80dc13bf5f02a74e775ea9cad5577a1700a Mon Sep 17 00:00:00 2001 From: yangjianbo Date: Fri, 6 Feb 2026 09:54:29 +0800 Subject: [PATCH] =?UTF-8?q?test(backend):=20=E4=BF=AE=E5=A4=8D=20usage=20?= =?UTF-8?q?=E7=B1=BB=E5=9E=8B=E6=96=AD=E8=A8=80=E6=9C=AA=E6=A3=80=E6=9F=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../service/gateway_cached_tokens_test.go | 16 ++++++---------- 1 file changed, 6 insertions(+), 10 deletions(-) diff --git a/backend/internal/service/gateway_cached_tokens_test.go b/backend/internal/service/gateway_cached_tokens_test.go index 359d7f58..f886c855 100644 --- a/backend/internal/service/gateway_cached_tokens_test.go +++ b/backend/internal/service/gateway_cached_tokens_test.go @@ -163,11 +163,9 @@ func TestStreamingReconcile_MessageDelta(t *testing.T) { require.Equal(t, "message_delta", eventType) // 模拟 processSSEEvent 中的 reconcile 逻辑 - if u, ok := event["usage"].(map[string]any); ok { - reconcileCachedTokens(u) - } - - usage := event["usage"].(map[string]any) + usage, ok := event["usage"].(map[string]any) + require.True(t, ok) + reconcileCachedTokens(usage) assert.Equal(t, float64(15), usage["cache_read_input_tokens"]) } @@ -183,11 +181,9 @@ func TestStreamingReconcile_MessageDelta_NativeClaude(t *testing.T) { var event map[string]any require.NoError(t, json.Unmarshal([]byte(eventJSON), &event)) - if u, ok := event["usage"].(map[string]any); ok { - reconcileCachedTokens(u) - } - - usage := event["usage"].(map[string]any) + usage, ok := event["usage"].(map[string]any) + require.True(t, ok) + reconcileCachedTokens(usage) _, hasCacheRead := usage["cache_read_input_tokens"] assert.False(t, hasCacheRead, "不应为原生 Claude 响应注入 cache_read_input_tokens") }