Function newMockUpstream

aibridge/internal/integrationtest/mockupstream.go:117–144

newMockUpstream creates a started httptest.Server that replays fixture responses. Responses are returned in order: first call → first response. The test fails if the number of requests doesn't match the number of responses (when AllowOverflow is not set, default). srv := newMockUpstream(ctx, t, ne

(ctx context.Context, t *testing.T, responses ...upstreamResponse)

//	srv := newMockUpstream(ctx, t, newFixtureResponse(fix))                        // simple
//	srv := newMockUpstream(ctx, t, newFixtureResponse(fix), newFixtureToolResponse(fix)) // multi-turn
func newMockUpstream(ctx context.Context, t *testing.T, responses ...upstreamResponse) *mockUpstream {
	t.Helper()
	require.NotEmpty(t, responses, "at least one upstreamResponse required")

	ms := &mockUpstream{
		t:         t,
		responses: responses,
	}

	srv := httptest.NewUnstartedServer(http.HandlerFunc(ms.handle))
	srv.Config.BaseContext = func(_ net.Listener) context.Context { return ctx }
	srv.Start()

	t.Cleanup(func() {
		srv.Close()

		// Verify the number of requests matches expectations.
		calls := int(ms.Calls.Load())
		if ms.AllowOverflow {
			require.LessOrEqual(t, len(ms.responses), calls, "too few requests, got: %v, want at least: %v", calls, len(ms.responses))
		} else {
			require.Equal(t, len(ms.responses), calls, "unexpected number of requests, got: %v, want: %v", calls, len(ms.responses))
		}
	})

	ms.Server = srv
	return ms
}

func (ms *mockUpstream) handle(w http.ResponseWriter, r *http.Request) {
	call := int(ms.Calls.Add(1) - 1)