Function TestSessionIDTracking

internal/integrationtest/bridge_test.go:832–917

(t *testing.T)

}

func TestSessionIDTracking(t *testing.T) {
	t.Parallel()

	testCases := []struct {
		name              string
		fixture           []byte
		header            http.Header
		metadataSessionID string
		expectedClient    aibridge.Client
		expectSessionID   string
	}{
		// Session in header.
		{
			name:            "mux",
			fixture:         fixtures.AntSimple,
			expectedClient:  aibridge.ClientMux,
			expectSessionID: "mux-workspace-321",
			header: http.Header{
				"User-Agent":         []string{"mux/1.0.0"},
				"X-Mux-Workspace-Id": []string{"mux-workspace-321"},
			},
		},
		// Session in body.
		{
			name:            "claude_code",
			fixture:         fixtures.AntSimple,
			expectedClient:  aibridge.ClientClaudeCode,
			expectSessionID: "f47ac10b-58cc-4372-a567-0e02b2c3d479",
			header: http.Header{
				"User-Agent": []string{"claude-cli/2.0.67 (external, cli)"},
			},
			metadataSessionID: "user_abc123_account_456_session_f47ac10b-58cc-4372-a567-0e02b2c3d479",
		},
		// No session.
		{
			name:           "zed",
			fixture:        fixtures.AntSimple,
			expectedClient: aibridge.ClientZed,
			header: http.Header{
				"User-Agent": []string{"Zed/0.219.4+stable.119.abc123 (macos; aarch64)"},
			},
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			ctx, cancel := context.WithTimeout(t.Context(), testutil.WaitLong)
			t.Cleanup(cancel)

			fix := fixtures.Parse(t, tc.fixture)
			upstream := newMockUpstream(ctx, t, newFixtureResponse(fix))
			bridgeServer := newBridgeTestServer(ctx, t, upstream.URL, withProvider(config.ProviderAnthropic))

			reqBody := fix.Request()
			if tc.metadataSessionID != "" {
				var err error