Method handle

aibridge/internal/integrationtest/mockupstream.go:146–189

(w http.ResponseWriter, r *http.Request)

}

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

	body, err := io.ReadAll(r.Body)
	defer r.Body.Close()
	require.NoError(ms.t, err)

	ms.mu.Lock()
	ms.requests = append(ms.requests, receivedRequest{
		Method: r.Method,
		Path:   r.URL.Path,
		Header: r.Header.Clone(),
		Body:   append([]byte(nil), body...),
	})
	ms.mu.Unlock()

	validateRequest(ms.t, call, r.URL.Path, body)

	resp := ms.responseForCall(call)
	if resp.OnRequest != nil {
		resp.OnRequest(r, body)
	}

	if isStreaming(body, r.URL.Path) {
		require.NotEmpty(ms.t, resp.Streaming, "response #%d: Streaming body is empty (fixture missing streaming response?)", call+1)
		if isRawHTTPResponse(resp.Streaming) {
			ms.writeRawHTTPResponse(w, r, resp.Streaming)
			return
		}
		ms.writeSSE(w, resp.Streaming)
		return
	}

	require.NotEmpty(ms.t, resp.Blocking, "response #%d: Blocking body is empty (fixture missing non-streaming response?)", call+1)
	if isRawHTTPResponse(resp.Blocking) {
		ms.writeRawHTTPResponse(w, r, resp.Blocking)
		return
	}

	status := cmp.Or(ms.StatusCode, http.StatusOK)
	w.Header().Set("Content-Type", "application/json")
	w.WriteHeader(status)
	_, _ = w.Write(resp.Blocking)
}

func (ms *mockUpstream) responseForCall(call int) upstreamResponse {
	if call >= len(ms.responses) {