more e2e tests

Author: evgeniy-scherbina

fixtures/fixtures.go

@@ -172,6 +172,12 @@ var (
 
 	//go:embed bedrock/simple.resp.bin
 	BedrockSimpleResp []byte
+
+	//go:embed bedrock/single_builtin_tool.req.json
+	BedrockSingleBuiltinToolReq []byte
+
+	//go:embed bedrock/single_builtin_tool.resp.bin
+	BedrockSingleBuiltinToolResp []byte
 )
 
 // BedrockFixture holds a request/response pair for a Bedrock test case.

internal/integrationtest/bridge_test.go

@@ -2186,3 +2186,59 @@ func TestNativeBedrockSimple(t *testing.T) {
 	// Verify interception lifecycle.
 	bridgeServer.Recorder.VerifyAllInterceptionsEnded(t)
 }
+
+func TestNativeBedrockSingleBuiltinTool(t *testing.T) {
+	t.Parallel()
+
+	const bedrockModel = "us.anthropic.claude-sonnet-4-5-20250929-v1:0"
+
+	ctx, cancel := context.WithTimeout(t.Context(), testutil.WaitLong)
+	t.Cleanup(cancel)
+
+	fix := fixtures.BedrockFixture{
+		Request:  fixtures.BedrockSingleBuiltinToolReq,
+		Response: fixtures.BedrockSingleBuiltinToolResp,
+	}
+
+	upstream := newMockUpstream(ctx, t, upstreamResponse{
+		Streaming: fix.Response,
+	})
+
+	bridgeServer := newBridgeTestServer(ctx, t, upstream.URL)
+
+	path := "/bedrock/model/" + bedrockModel + "/invoke-with-response-stream"
+	resp, err := bridgeServer.makeRequest(t, http.MethodPost, path, fix.Request)
+	require.NoError(t, err)
+	defer resp.Body.Close()
+	require.Equal(t, http.StatusOK, resp.StatusCode)
+
+	// Verify non-empty response.
+	bodyBytes, err := io.ReadAll(resp.Body)
+	require.NoError(t, err)
+	assert.NotEmpty(t, bodyBytes)
+
+	// Verify token usage: message_start (input=583, output=1) + message_delta (output=56).
+	tokenUsages := bridgeServer.Recorder.RecordedTokenUsages()
+	require.Len(t, tokenUsages, 2)
+	assert.EqualValues(t, 583, bridgeServer.Recorder.TotalInputTokens(), "input tokens")
+	assert.EqualValues(t, 57, bridgeServer.Recorder.TotalOutputTokens(), "output tokens (1 + 56)")
+
+	// Verify tool usage.
+	toolUsages := bridgeServer.Recorder.RecordedToolUsages()
+	require.Len(t, toolUsages, 1)
+	assert.Equal(t, "Read", toolUsages[0].Tool)
+	assert.Equal(t, "toolu_bdrk_01K3c5nvHfTPbUTdKyt2XdiD", toolUsages[0].ToolCallID)
+	require.IsType(t, json.RawMessage{}, toolUsages[0].Args)
+	var args map[string]any
+	require.NoError(t, json.Unmarshal(toolUsages[0].Args.(json.RawMessage), &args))
+	require.Contains(t, args, "file_path")
+	assert.Equal(t, "/tmp/foo", args["file_path"])
+
+	// Verify prompt.
+	promptUsages := bridgeServer.Recorder.RecordedPromptUsages()
+	require.Len(t, promptUsages, 1)
+	assert.Equal(t, "read the /tmp/foo file", promptUsages[0].Prompt)
+
+	// Verify interception lifecycle.
+	bridgeServer.Recorder.VerifyAllInterceptionsEnded(t)
+}