-
Notifications
You must be signed in to change notification settings - Fork 12
/
response_test.go
104 lines (85 loc) · 2.51 KB
/
response_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
package gogo
import (
"net/http"
"net/http/httptest"
"testing"
"github.com/golib/assert"
)
func Test_NewResponse(t *testing.T) {
it := assert.New(t)
recorder := httptest.NewRecorder()
response := NewResponse(recorder)
it.Implements((*Responser)(nil), response)
it.Equal(http.StatusOK, response.Status())
it.Equal(nonHeaderFlushed, response.Size())
}
func Test_ResponseWriteHeader(t *testing.T) {
it := assert.New(t)
recorder := httptest.NewRecorder()
response := NewResponse(recorder)
response.WriteHeader(http.StatusRequestTimeout)
it.Equal(http.StatusRequestTimeout, response.Status())
it.Equal(nonHeaderFlushed, response.Size())
}
func Test_ResponseFlushHeader(t *testing.T) {
it := assert.New(t)
recorder := httptest.NewRecorder()
response := NewResponse(recorder)
response.WriteHeader(http.StatusRequestTimeout)
response.FlushHeader()
it.Equal(http.StatusRequestTimeout, recorder.Code)
it.NotEqual(nonHeaderFlushed, response.Size())
// no effect after flushed headers
response.WriteHeader(http.StatusOK)
it.Equal(http.StatusOK, response.Status())
response.FlushHeader()
it.NotEqual(http.StatusOK, recorder.Code)
}
func Test_ResponseWrite(t *testing.T) {
it := assert.New(t)
recorder := httptest.NewRecorder()
hello := []byte("Hello,")
world := []byte("world!")
expected := []byte("Hello,world!")
response := NewResponse(recorder)
response.Write(hello)
response.Write(world)
it.True(response.HeaderFlushed())
it.Equal(len(expected), response.Size())
it.Equal(expected, recorder.Body.Bytes())
}
func Benchmark_ResponseWrite(b *testing.B) {
b.ReportAllocs()
b.ResetTimer()
hello := []byte("Hello,")
world := []byte("world!")
recorder := httptest.NewRecorder()
response := NewResponse(recorder)
for i := 0; i < b.N; i++ {
response.Write(hello)
response.Write(world)
}
}
func Test_ResponseHijack(t *testing.T) {
it := assert.New(t)
recorder := httptest.NewRecorder()
expected := []byte("Hello,world!")
response := NewResponse(recorder)
response.Write(expected)
it.True(response.HeaderFlushed())
it.Equal(recorder, response.(*Response).ResponseWriter)
it.Equal(len(expected), response.Size())
response.Hijack(httptest.NewRecorder())
it.False(response.HeaderFlushed())
it.NotEqual(recorder, response.(*Response).ResponseWriter)
it.Equal(nonHeaderFlushed, response.Size())
}
func Benchmark_ResponseHijack(b *testing.B) {
b.ReportAllocs()
b.ResetTimer()
recorder := httptest.NewRecorder()
response := NewResponse(recorder)
for i := 0; i < b.N; i++ {
response.Hijack(recorder)
}
}