This commit is contained in:
Dmitry Dorogin 2018-07-17 16:07:09 +03:00
parent dce9741650
commit 8e2c7309eb

View File

@ -1508,7 +1508,7 @@ func TestContextStream(t *testing.T) {
c, _ := CreateTestContext(w) c, _ := CreateTestContext(w)
stopStream := true stopStream := true
c.Stream(func (w io.Writer) bool { c.Stream(func(w io.Writer) bool {
defer func() { defer func() {
stopStream = false stopStream = false
}() }()
@ -1525,7 +1525,7 @@ func TestContextStreamWithClientGone(t *testing.T) {
w := CreateTestResponseRecorder() w := CreateTestResponseRecorder()
c, _ := CreateTestContext(w) c, _ := CreateTestContext(w)
c.Stream(func (writer io.Writer) bool { c.Stream(func(writer io.Writer) bool {
defer func() { defer func() {
w.closeClient() w.closeClient()
}() }()
@ -1536,4 +1536,4 @@ func TestContextStreamWithClientGone(t *testing.T) {
}) })
assert.Equal(t, "test", w.Body.String()) assert.Equal(t, "test", w.Body.String())
} }