Fix tests

contentway · Mar 22, 2020 · 4f5133c · 4f5133c
1 parent be31c2b
commit 4f5133c
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/consts.go b/consts.go
@@ -40,7 +40,7 @@ const (
 
 	HeaderAuthorization      = "Authorization"
 	HeaderContentType        = "Content-Type"
-	HeaderContentDisposition = "ContentDisposition"
+	HeaderContentDisposition = "Content-Disposition"
 
 	AbortMiddlewareIndex = math.MaxInt8 / 2
 )
diff --git a/context_test.go b/context_test.go
@@ -61,8 +61,8 @@ func TestDownloadResp(t *testing.T) {
 	a.ServeHTTP(w, r)
 	ass.Equal(200, w.Code)
 	ass.Equal("123", w.Body.String())
-	ass.Equal("application/octet-stream; charset=UTF-8", w.Header().Get("Content-Type"))
-	ass.Equal("attachment; filename=\"test.txt\"", w.Header().Get("Content-Disposition"))
+	ass.Equal("application/octet-stream; charset=UTF-8", w.Header().Get(HeaderContentType))
+	ass.Equal("attachment; filename=\"test.txt\"", w.Header().Get(HeaderContentDisposition))
 }
 
 func TestCData(t *testing.T) {

diff --git a/session_test.go b/session_test.go
@@ -28,7 +28,7 @@ func TestSession(t *testing.T) {
 		test2 := session.GetInt("test2", 0)
 
 		assert.Equal(t, "123", test1)
-		assert.Equal(t, 123, test2)
+		assert.Equal(t, int64(123), test2)
 		c.String(200, "")
 	})
 

diff --git a/sessions/manager_test.go b/sessions/manager_test.go
@@ -59,7 +59,7 @@ func (t *SessionManagerSuite) TestSession() {
 
 	t.False(session.IsNew)
 	t.Equal("abc", session.GetString("foo", ""))
-	t.Equal(123, session.GetInt("baz", 0))
+	t.Equal(int64(123), session.GetInt("baz", 0))
 	t.Equal([]string{"1", "2", "3"}, session.GetStrings("test", nil))
 	t.Equal([]int64{1, 2, 3}, session.GetInts("test2", nil))
 	t.Equal(true, session.GetBool("bool", false))