From 971544570d6da41c6486d0e92f225bf28dfe2d7a Mon Sep 17 00:00:00 2001 From: wucm667 Date: Mon, 17 Aug 2026 17:11:33 +0800 Subject: [PATCH] test(antigravity): check tool config assertions --- backend/internal/service/antigravity_gateway_compat_test.go | 3 ++- .../internal/service/antigravity_gateway_service_test.go | 6 ++++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/backend/internal/service/antigravity_gateway_compat_test.go b/backend/internal/service/antigravity_gateway_compat_test.go index 3d2176a00..3df0b0081 100644 --- a/backend/internal/service/antigravity_gateway_compat_test.go +++ b/backend/internal/service/antigravity_gateway_compat_test.go @@ -254,7 +254,8 @@ func TestBuildAntigravityCompatGeminiBody_ConfiguresMixedToolInvocations(t *test var wrapped map[string]any require.NoError(t, json.Unmarshal(body, &wrapped)) - request := wrapped["request"].(map[string]any) + request, ok := wrapped["request"].(map[string]any) + require.True(t, ok) toolConfig, exists := request["toolConfig"].(map[string]any) if !tt.wantField { require.False(t, exists) diff --git a/backend/internal/service/antigravity_gateway_service_test.go b/backend/internal/service/antigravity_gateway_service_test.go index 1a1bfc998..d1c389c7a 100644 --- a/backend/internal/service/antigravity_gateway_service_test.go +++ b/backend/internal/service/antigravity_gateway_service_test.go @@ -370,8 +370,10 @@ func TestAntigravityGatewayService_ForwardGemini_PreservesServerSideToolInvocati var wrapped map[string]any require.NoError(t, json.Unmarshal(upstream.requestBodies[0], &wrapped)) - request := wrapped["request"].(map[string]any) - toolConfig := request["toolConfig"].(map[string]any) + request, ok := wrapped["request"].(map[string]any) + require.True(t, ok) + toolConfig, ok := request["toolConfig"].(map[string]any) + require.True(t, ok) require.Equal(t, true, toolConfig["includeServerSideToolInvocations"]) require.NotContains(t, toolConfig, "include_server_side_tool_invocations") }