Function fakeOpenAIUpstream

backend/go/cloud-proxy/provider_openai_test.go:19–31

fakeOpenAIUpstream returns an httptest.Server that decodes the inbound request as an openAIRequest, calls handler with it, and writes the handler's reply as the response.

(t *testing.T, handler func(req openAIRequest) (status int, body string, contentType string))

// inbound request as an openAIRequest, calls handler with it, and
// writes the handler's reply as the response.
func fakeOpenAIUpstream(t *testing.T, handler func(req openAIRequest) (status int, body string, contentType string)) (*httptest.Server, *openAIRequest) {
	t.Helper()
	var captured openAIRequest
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		raw, _ := io.ReadAll(r.Body)
		_ = json.Unmarshal(raw, &captured)
		status, body, ct := handler(captured)
		w.Header().Set("Content-Type", ct)
		w.WriteHeader(status)
		_, _ = io.WriteString(w, body)
	}))
	return srv, &captured
}

func newTranslateCloudProxy(t *testing.T, upstreamURL string) *CloudProxy {
	t.Helper()