Function NewOpenAI

coderd/x/chatd/chattest/openai.go:192–211

NewOpenAI creates a new OpenAI test server with a handler function. The handler is called for each request and should return either a streaming response (via channel) or a non-streaming response. Returns the base URL of the server.

(t testing.TB, handler OpenAIHandler)

// response (via channel) or a non-streaming response.
// Returns the base URL of the server.
func NewOpenAI(t testing.TB, handler OpenAIHandler) string {
	t.Helper()

	s := &openAIServer{
		t:       t,
		handler: handler,
	}

	mux := http.NewServeMux()
	mux.HandleFunc("POST /chat/completions", s.handleChatCompletions)
	mux.HandleFunc("POST /responses", s.handleResponses)

	s.server = httptest.NewServer(mux)

	t.Cleanup(func() {
		s.server.Close()
	})

	return s.server.URL
}

func (s *openAIServer) handleChatCompletions(w http.ResponseWriter, r *http.Request) {
	body, err := io.ReadAll(r.Body)