zfc/sub2api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(gateway): check Gemini chat completion assertions

Browse Source
This commit is contained in:
wucm667 2026-05-14 15:33:10 +08:00
parent 041d138f76
commit f9d5ccdf24
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
backend/internal/service/gemini_messages_compat_service_test.go
View File

@ -101,11 +101,17 @@ func TestGeminiForwardAsChatCompletions_OAuthRoutesToGeminiAndReturnsChatFormat(
require.NoError(t, json.Unmarshal(rec.Body.Bytes(), &got)) require.NoError(t, json.Unmarshal(rec.Body.Bytes(), &got))
require.Equal(t, "chat.completion", got["object"]) require.Equal(t, "chat.completion", got["object"])
require.Equal(t, "gemini-2.5-flash", got["model"]) require.Equal(t, "gemini-2.5-flash", got["model"])
choices := got["choices"].([]any) choices, ok := got["choices"].([]any)
message := choices[0].(map[string]any)["message"].(map[string]any) require.True(t, ok)
require.NotEmpty(t, choices)
choice, ok := choices[0].(map[string]any)
require.True(t, ok)
message, ok := choice["message"].(map[string]any)
require.True(t, ok)
require.Equal(t, "assistant", message["role"]) require.Equal(t, "assistant", message["role"])
require.Equal(t, "hello from gemini", message["content"]) require.Equal(t, "hello from gemini", message["content"])
usage := got["usage"].(map[string]any) usage, ok := got["usage"].(map[string]any)
require.True(t, ok)
require.Equal(t, float64(7), usage["prompt_tokens"]) require.Equal(t, float64(7), usage["prompt_tokens"])
require.Equal(t, float64(3), usage["completion_tokens"]) require.Equal(t, float64(3), usage["completion_tokens"])
require.Equal(t, float64(10), usage["total_tokens"]) require.Equal(t, float64(10), usage["total_tokens"])