From 4567f6582b5d61b39c715eef739dd32c43f73c80 Mon Sep 17 00:00:00 2001 From: feitianbubu Date: Wed, 24 Jun 2026 23:59:22 +0800 Subject: [PATCH] test(admin/usage): update sqlmock rows for cache breakdown columns --- .../repository/usage_log_repo_request_type_test.go | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/backend/internal/repository/usage_log_repo_request_type_test.go b/backend/internal/repository/usage_log_repo_request_type_test.go index 597c95971d..62ce892c5f 100644 --- a/backend/internal/repository/usage_log_repo_request_type_test.go +++ b/backend/internal/repository/usage_log_repo_request_type_test.go @@ -397,11 +397,13 @@ func TestUsageLogRepositoryGetStatsWithFiltersRequestTypePriority(t *testing.T) "total_input_tokens", "total_output_tokens", "total_cache_tokens", + "total_cache_creation_tokens", + "total_cache_read_tokens", "total_cost", "total_actual_cost", "total_account_cost", "avg_duration_ms", - }).AddRow(int64(1), int64(2), int64(3), int64(4), 1.2, 1.0, 1.2, 20.0)) + }).AddRow(int64(1), int64(2), int64(3), int64(4), int64(1), int64(3), 1.2, 1.0, 1.2, 20.0)) mock.ExpectQuery("SELECT COALESCE\\(NULLIF\\(TRIM\\(inbound_endpoint\\), ''\\), 'unknown'\\) AS endpoint"). WithArgs(sqlmock.AnyArg(), sqlmock.AnyArg(), requestType). WillReturnRows(sqlmock.NewRows([]string{"endpoint", "requests", "total_tokens", "cost", "actual_cost"})) @@ -489,9 +491,10 @@ func TestUsageLogRepositoryGetStatsWithFiltersAlwaysReturnsAccountCost(t *testin mock.ExpectQuery("FROM usage_logs"). WillReturnRows(sqlmock.NewRows([]string{ "total_requests", "total_input_tokens", "total_output_tokens", - "total_cache_tokens", "total_cost", "total_actual_cost", + "total_cache_tokens", "total_cache_creation_tokens", "total_cache_read_tokens", + "total_cost", "total_actual_cost", "total_account_cost", "avg_duration_ms", - }).AddRow(int64(50), int64(1000), int64(2000), int64(100), 15.0, 12.5, 11.0, 100.0)) + }).AddRow(int64(50), int64(1000), int64(2000), int64(100), int64(60), int64(40), 15.0, 12.5, 11.0, 100.0)) mock.ExpectQuery("SELECT COALESCE\\(NULLIF\\(TRIM\\(inbound_endpoint\\)"). WillReturnRows(sqlmock.NewRows([]string{"endpoint", "requests", "total_tokens", "cost", "actual_cost"})) mock.ExpectQuery("SELECT COALESCE\\(NULLIF\\(TRIM\\(upstream_endpoint\\)").