90 lines
2.1 KiB
Go
90 lines
2.1 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"
|
|
)
|
|
|
|
var _ ResponseWriter = &responseWriter{}
|
|
var _ http.ResponseWriter = &responseWriter{}
|
|
var _ http.ResponseWriter = 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)
|
|
}
|