116 lines
2.8 KiB
Go
116 lines
2.8 KiB
Go
|
// Copyright 2014 Manu Martinez-Almeida. All rights reserved.
|
||
|
|
// 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"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TODO
|
||
|
|
// func (w *responseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
|
||
|
|
// func (w *responseWriter) CloseNotify() <-chan bool {
|
||
|
|
// func (w *responseWriter) Flush() {
|
||
|
|
|
||
|
|
var _ ResponseWriter = &responseWriter{}
|
||
|
|
var _ http.ResponseWriter = &responseWriter{}
|
||
|
|
var _ http.ResponseWriter = ResponseWriter(&responseWriter{})
|
||
|
|
var _ http.Hijacker = ResponseWriter(&responseWriter{})
|
||
|
|
var _ http.Flusher = ResponseWriter(&responseWriter{})
|
||
|
|
var _ http.CloseNotifier = ResponseWriter(&responseWriter{})
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
SetMode(TestMode)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestResponseWriterReset(t *testing.T) {
|
||
|
|
testWritter := httptest.NewRecorder()
|
||
|
|
writer := &responseWriter{}
|
||
|
|
var w ResponseWriter = writer
|
||
|
|
|
||
|
|
writer.reset(testWritter)
|
||
|
|
assert.Equal(t, writer.size, -1)
|
||
|
|
assert.Equal(t, writer.status, 200)
|
||
|
|
assert.Equal(t, writer.ResponseWriter, testWritter)
|
||
|
|
assert.Equal(t, w.Size(), -1)
|
||
|
|
assert.Equal(t, w.Status(), 200)
|
||
|
|
assert.False(t, w.Written())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestResponseWriterWriteHeader(t *testing.T) {
|
||
|
|
testWritter := httptest.NewRecorder()
|
||
|
|
writer := &responseWriter{}
|
||
|
|
writer.reset(testWritter)
|
||
|
|
w := ResponseWriter(writer)
|
||
|
|
|
||
|
|
w.WriteHeader(300)
|
||
|
|
assert.False(t, w.Written())
|
||
|
|
assert.Equal(t, w.Status(), 300)
|
||
|
|
assert.NotEqual(t, testWritter.Code, 300)
|
||
|
|
|
||
|
|
w.WriteHeader(-1)
|
||
|
|
assert.Equal(t, w.Status(), 300)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestResponseWriterWriteHeadersNow(t *testing.T) {
|
||
|
|
testWritter := httptest.NewRecorder()
|
||
|
|
writer := &responseWriter{}
|
||
|
|
writer.reset(testWritter)
|
||
|
|
w := ResponseWriter(writer)
|
||
|
|
|
||
|
|
w.WriteHeader(300)
|
||
|
|
w.WriteHeaderNow()
|
||
|
|
|
||
|
|
assert.True(t, w.Written())
|
||
|
|
assert.Equal(t, w.Size(), 0)
|
||
|
|
assert.Equal(t, testWritter.Code, 300)
|
||
|
|
|
||
|
|
writer.size = 10
|
||
|
|
w.WriteHeaderNow()
|
||
|
|
assert.Equal(t, w.Size(), 10)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestResponseWriterWrite(t *testing.T) {
|
||
|
|
testWritter := httptest.NewRecorder()
|
||
|
|
writer := &responseWriter{}
|
||
|
|
writer.reset(testWritter)
|
||
|
|
w := ResponseWriter(writer)
|
||
|
|
|
||
|
|
n, err := w.Write([]byte("hola"))
|
||
|
|
assert.Equal(t, n, 4)
|
||
|
|
assert.Equal(t, w.Size(), 4)
|
||
|
|
assert.Equal(t, w.Status(), 200)
|
||
|
|
assert.Equal(t, testWritter.Code, 200)
|
||
|
|
assert.Equal(t, testWritter.Body.String(), "hola")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
n, err = w.Write([]byte(" adios"))
|
||
|
|
assert.Equal(t, n, 6)
|
||
|
|
assert.Equal(t, w.Size(), 10)
|
||
|
|
assert.Equal(t, testWritter.Body.String(), "hola adios")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestResponseWriterHijack(t *testing.T) {
|
||
|
|
testWritter := httptest.NewRecorder()
|
||
|
|
writer := &responseWriter{}
|
||
|
|
writer.reset(testWritter)
|
||
|
|
w := ResponseWriter(writer)
|
||
|
|
|
||
|
|
assert.Panics(t, func() {
|
||
|
|
w.Hijack()
|
||
|
|
})
|
||
|
|
assert.True(t, w.Written())
|
||
|
|
|
||
|
|
assert.Panics(t, func() {
|
||
|
|
w.CloseNotify()
|
||
|
|
})
|
||
|
|
|
||
|
|
w.Flush()
|
||
|
|
}
|