diff --git a/context_test.go b/context_test.go index d66e4eb..169768b 100644 --- a/context_test.go +++ b/context_test.go @@ -7,11 +7,11 @@ package gin import ( "bytes" "errors" - "fmt" "html/template" "mime/multipart" "net/http" "net/http/httptest" + "strings" "testing" "time" @@ -352,7 +352,7 @@ func TestContextRenderSSE(t *testing.T) { "bar": "foo", }) - assert.Equal(t, fmt.Sprintf("%v", w.Body), "event:float\ndata:1.5\n\nid:123\ndata:text\n\nevent:chat\ndata:{\"bar\":\"foo\",\"foo\":\"bar\"}\n\n") + assert.Equal(t, strings.Replace(w.Body.String(), " ", "", -1), strings.Replace("event:float\ndata:1.5\n\nid:123\ndata:text\n\nevent:chat\ndata:{\"bar\":\"foo\",\"foo\":\"bar\"}\n\n", " ", "", -1)) } func TestContextRenderFile(t *testing.T) { diff --git a/middleware_test.go b/middleware_test.go index 61d27c9..3101d52 100644 --- a/middleware_test.go +++ b/middleware_test.go @@ -6,6 +6,7 @@ package gin import ( "errors" + "strings" "testing" @@ -245,11 +246,5 @@ func TestMiddlewareWrite(t *testing.T) { w := performRequest(router, "GET", "/") assert.Equal(t, w.Code, 400) - assert.Equal(t, w.Body.String(), `hola -bar{"foo":"bar"} -{"foo":"bar"} -event:test -data:message - -`) + assert.Equal(t, strings.Replace(w.Body.String(), " ", "", -1), strings.Replace("hola\nbar{\"foo\":\"bar\"}\n{\"foo\":\"bar\"}\nevent:test\ndata:message\n\n", " ", "", -1)) }