test(backend): 修复 usage 类型断言未检查

This commit is contained in:
yangjianbo
2026-02-06 09:54:29 +08:00
parent 98671a73f4
commit ee01f80dc1

View File

@@ -163,11 +163,9 @@ func TestStreamingReconcile_MessageDelta(t *testing.T) {
require.Equal(t, "message_delta", eventType) require.Equal(t, "message_delta", eventType)
// 模拟 processSSEEvent 中的 reconcile 逻辑 // 模拟 processSSEEvent 中的 reconcile 逻辑
if u, ok := event["usage"].(map[string]any); ok { usage, ok := event["usage"].(map[string]any)
reconcileCachedTokens(u) require.True(t, ok)
} reconcileCachedTokens(usage)
usage := event["usage"].(map[string]any)
assert.Equal(t, float64(15), usage["cache_read_input_tokens"]) 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 var event map[string]any
require.NoError(t, json.Unmarshal([]byte(eventJSON), &event)) require.NoError(t, json.Unmarshal([]byte(eventJSON), &event))
if u, ok := event["usage"].(map[string]any); ok { usage, ok := event["usage"].(map[string]any)
reconcileCachedTokens(u) require.True(t, ok)
} reconcileCachedTokens(usage)
usage := event["usage"].(map[string]any)
_, hasCacheRead := usage["cache_read_input_tokens"] _, hasCacheRead := usage["cache_read_input_tokens"]
assert.False(t, hasCacheRead, "不应为原生 Claude 响应注入 cache_read_input_tokens") assert.False(t, hasCacheRead, "不应为原生 Claude 响应注入 cache_read_input_tokens")
} }