gin/response_writer_test.go

153 lines
3.9 KiB
Go
Raw Normal View History

// Copyright 2014 Manu Martinez-Almeida. All rights reserved.
2015-04-08 12:24:49 +00:00
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.
package gin
import (
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/assert"
)
2015-04-09 10:15:02 +00:00
// TODO
// func (w *responseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
// func (w *responseWriter) CloseNotify() <-chan bool {
// func (w *responseWriter) Flush() {
var (
_ ResponseWriter = &responseWriter{}
_ http.ResponseWriter = &responseWriter{}
_ http.ResponseWriter = ResponseWriter(&responseWriter{})
_ http.Hijacker = ResponseWriter(&responseWriter{})
_ http.Flusher = ResponseWriter(&responseWriter{})
_ http.CloseNotifier = ResponseWriter(&responseWriter{})
)
2015-04-08 12:24:49 +00:00
func init() {
SetMode(TestMode)
}
func TestResponseWriterReset(t *testing.T) {
2019-09-30 06:22:12 +00:00
testWriter := httptest.NewRecorder()
2015-04-08 12:24:49 +00:00
writer := &responseWriter{}
var w ResponseWriter = writer
2019-09-30 06:22:12 +00:00
writer.reset(testWriter)
assert.Equal(t, -1, writer.size)
assert.Equal(t, http.StatusOK, writer.status)
2019-09-30 06:22:12 +00:00
assert.Equal(t, testWriter, writer.ResponseWriter)
assert.Equal(t, -1, w.Size())
assert.Equal(t, http.StatusOK, w.Status())
2015-04-08 12:24:49 +00:00
assert.False(t, w.Written())
}
func TestResponseWriterWriteHeader(t *testing.T) {
2019-09-30 06:22:12 +00:00
testWriter := httptest.NewRecorder()
2015-04-08 12:24:49 +00:00
writer := &responseWriter{}
2019-09-30 06:22:12 +00:00
writer.reset(testWriter)
2015-04-08 12:24:49 +00:00
w := ResponseWriter(writer)
w.WriteHeader(http.StatusMultipleChoices)
2015-04-08 12:24:49 +00:00
assert.False(t, w.Written())
assert.Equal(t, http.StatusMultipleChoices, w.Status())
2019-09-30 06:22:12 +00:00
assert.NotEqual(t, http.StatusMultipleChoices, testWriter.Code)
2015-04-08 12:24:49 +00:00
w.WriteHeader(-1)
assert.Equal(t, http.StatusMultipleChoices, w.Status())
2015-04-08 12:24:49 +00:00
}
func TestResponseWriterWriteHeadersNow(t *testing.T) {
2019-09-30 06:22:12 +00:00
testWriter := httptest.NewRecorder()
2015-04-08 12:24:49 +00:00
writer := &responseWriter{}
2019-09-30 06:22:12 +00:00
writer.reset(testWriter)
2015-04-08 12:24:49 +00:00
w := ResponseWriter(writer)
w.WriteHeader(http.StatusMultipleChoices)
2015-04-08 12:24:49 +00:00
w.WriteHeaderNow()
assert.True(t, w.Written())
assert.Equal(t, 0, w.Size())
2019-09-30 06:22:12 +00:00
assert.Equal(t, http.StatusMultipleChoices, testWriter.Code)
2015-04-08 12:24:49 +00:00
writer.size = 10
w.WriteHeaderNow()
assert.Equal(t, 10, w.Size())
2015-04-08 12:24:49 +00:00
}
func TestResponseWriterWrite(t *testing.T) {
2019-09-30 06:22:12 +00:00
testWriter := httptest.NewRecorder()
2015-04-08 12:24:49 +00:00
writer := &responseWriter{}
2019-09-30 06:22:12 +00:00
writer.reset(testWriter)
2015-04-08 12:24:49 +00:00
w := ResponseWriter(writer)
n, err := w.Write([]byte("hola"))
assert.Equal(t, 4, n)
assert.Equal(t, 4, w.Size())
assert.Equal(t, http.StatusOK, w.Status())
2019-09-30 06:22:12 +00:00
assert.Equal(t, http.StatusOK, testWriter.Code)
assert.Equal(t, "hola", testWriter.Body.String())
2015-04-08 12:24:49 +00:00
assert.NoError(t, err)
n, err = w.Write([]byte(" adios"))
assert.Equal(t, 6, n)
assert.Equal(t, 10, w.Size())
2019-09-30 06:22:12 +00:00
assert.Equal(t, "hola adios", testWriter.Body.String())
2015-04-08 12:24:49 +00:00
assert.NoError(t, err)
}
2015-05-19 00:29:32 +00:00
func TestResponseWriterHijack(t *testing.T) {
2019-09-30 06:22:12 +00:00
testWriter := httptest.NewRecorder()
2015-05-19 00:29:32 +00:00
writer := &responseWriter{}
2019-09-30 06:22:12 +00:00
writer.reset(testWriter)
2015-05-19 00:29:32 +00:00
w := ResponseWriter(writer)
assert.Panics(t, func() {
2019-01-18 01:32:53 +00:00
_, _, err := w.Hijack()
assert.NoError(t, err)
2015-05-19 00:29:32 +00:00
})
assert.True(t, w.Written())
assert.Panics(t, func() {
w.CloseNotify()
})
w.Flush()
}
func TestResponseWriterFlush(t *testing.T) {
testServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
writer := &responseWriter{}
writer.reset(w)
writer.WriteHeader(http.StatusInternalServerError)
writer.Flush()
}))
defer testServer.Close()
// should return 500
resp, err := http.Get(testServer.URL)
assert.NoError(t, err)
assert.Equal(t, http.StatusInternalServerError, resp.StatusCode)
}
func TestResponseWriterStatusCode(t *testing.T) {
testWriter := httptest.NewRecorder()
writer := &responseWriter{}
writer.reset(testWriter)
w := ResponseWriter(writer)
w.WriteHeader(http.StatusOK)
w.WriteHeaderNow()
assert.Equal(t, http.StatusOK, w.Status())
assert.True(t, w.Written())
w.WriteHeader(http.StatusUnauthorized)
// status must be 200 although we tried to change it
assert.Equal(t, http.StatusOK, w.Status())
}