From 8ba804b5f482a43fba72814e38b5c8b5847cfc3d Mon Sep 17 00:00:00 2001 From: salamer Date: Sat, 18 Aug 2018 12:07:28 +0800 Subject: [PATCH] fixed missed code fixed missed code bug replace 201 with http lib and update annotation --- context_test.go | 2 +- render/render_test.go | 7 +++++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/context_test.go b/context_test.go index 45db8799..c380db6c 100644 --- a/context_test.go +++ b/context_test.go @@ -976,7 +976,7 @@ func TestContextRenderProtoBuf(t *testing.T) { protoData, err := proto.Marshal(data) assert.NoError(t, err) - assert.Equal(t, 201, w.Code) + assert.Equal(t, http.StatusCreated, w.Code) assert.Equal(t, string(protoData[:]), w.Body.String()) assert.Equal(t, "application/x-protobuf", w.HeaderMap.Get("Content-Type")) } diff --git a/render/render_test.go b/render/render_test.go index 53ee9ae6..5f28240d 100755 --- a/render/render_test.go +++ b/render/render_test.go @@ -280,10 +280,13 @@ func TestRenderProtoBuf(t *testing.T) { (ProtoBuf{data}).WriteContentType(w) protoData, err := proto.Marshal(data) assert.NoError(t, err) - (YAML{data}).WriteContentType(w) - assert.Equal(t, "application/x-yaml; charset=utf-8", w.Header().Get("Content-Type")) + assert.Equal(t, "application/x-protobuf", w.Header().Get("Content-Type")) + err = (ProtoBuf{data}).Render(w) + + assert.NoError(t, err) assert.Equal(t, string(protoData[:]), w.Body.String()) + assert.Equal(t, "application/x-protobuf", w.Header().Get("Content-Type")) } func TestRenderXML(t *testing.T) {