Fixed test for content disposition

Signed-off-by: Vishal Rana <[email protected]>
marlncpe · Aug 24, 2017 · cec7629 · cec7629
1 parent 4df3551
commit cec7629
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/context_test.go b/context_test.go
@@ -187,7 +187,7 @@ func TestContext(t *testing.T) {
 	err = c.Attachment("_fixture/images/walle.png", "walle.png")
 	if assert.NoError(t, err) {
 		assert.Equal(t, http.StatusOK, rec.Code)
-		assert.Equal(t, "attachment; filename=walle.png", rec.Header().Get(HeaderContentDisposition))
+		assert.Equal(t, "attachment; filename=\"walle.png\"", rec.Header().Get(HeaderContentDisposition))
 		assert.Equal(t, 219885, rec.Body.Len())
 	}
 
@@ -197,7 +197,7 @@ func TestContext(t *testing.T) {
 	err = c.Inline("_fixture/images/walle.png", "walle.png")
 	if assert.NoError(t, err) {
 		assert.Equal(t, http.StatusOK, rec.Code)
-		assert.Equal(t, "inline; filename=walle.png", rec.Header().Get(HeaderContentDisposition))
+		assert.Equal(t, "inline; filename=\"walle.png\"", rec.Header().Get(HeaderContentDisposition))
 		assert.Equal(t, 219885, rec.Body.Len())
 	}