opensource/sub2api
3
0
Fork 0
mirror of https://github.com/Wei-Shaw/sub2api.git synced 2026-09-24 16:05:44 +08:00
Code Issues 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
Download Patch File Download Diff File Expand all files Collapse all files
backend/internal/service/gemini_messages_compat_service_test.go
+9 -3
View File
@@ -101,11 +101,17 @@ func TestGeminiForwardAsChatCompletions_OAuthRoutesToGeminiAndReturnsChatFormat(
require.NoError(t, json.Unmarshal(rec.Body.Bytes(), &got))
require.Equal(t, "chat.completion", got["object"])
require.Equal(t, "gemini-2.5-flash", got["model"])
choices := got["choices"].([]any)
message := choices[0].(map[string]any)["message"].(map[string]any)
choices, ok := got["choices"].([]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, "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(3), usage["completion_tokens"])
require.Equal(t, float64(10), usage["total_tokens"])