54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package server
|
|
|
|
import (
|
|
"io"
|
|
"log/slog"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type nonFlusherRecorder struct {
|
|
recorder *httptest.ResponseRecorder
|
|
writeHeaderCalls int
|
|
}
|
|
|
|
func newNonFlusherRecorder() *nonFlusherRecorder {
|
|
return &nonFlusherRecorder{recorder: httptest.NewRecorder()}
|
|
}
|
|
|
|
func (w *nonFlusherRecorder) Header() http.Header {
|
|
return w.recorder.Header()
|
|
}
|
|
|
|
func (w *nonFlusherRecorder) Write(b []byte) (int, error) {
|
|
return w.recorder.Write(b)
|
|
}
|
|
|
|
func (w *nonFlusherRecorder) WriteHeader(statusCode int) {
|
|
w.writeHeaderCalls++
|
|
w.recorder.WriteHeader(statusCode)
|
|
}
|
|
|
|
func (w *nonFlusherRecorder) StatusCode() int {
|
|
return w.recorder.Code
|
|
}
|
|
|
|
func (w *nonFlusherRecorder) BodyString() string {
|
|
return w.recorder.Body.String()
|
|
}
|
|
|
|
func TestHandleStreamingResponseWithoutFlusherWritesSingleErrorHeader(t *testing.T) {
|
|
s := New(nil, nil, slog.New(slog.NewTextHandler(io.Discard, nil)))
|
|
req := httptest.NewRequest(http.MethodPost, "/v1/responses", nil)
|
|
w := newNonFlusherRecorder()
|
|
|
|
s.handleStreamingResponse(w, req, nil, nil, nil, nil, nil)
|
|
|
|
assert.Equal(t, 1, w.writeHeaderCalls)
|
|
assert.Equal(t, http.StatusInternalServerError, w.StatusCode())
|
|
assert.Contains(t, w.BodyString(), "streaming not supported")
|
|
}
|