Merge pull request #92 from drakkan/errcheck

Encode: check for errors while setting claim fields

jwtauth.go

@@ -122,7 +122,9 @@ func VerifyToken(ja *JWTAuth, tokenString string) (jwt.Token, error) {
 func (ja *JWTAuth) Encode(claims map[string]interface{}) (t jwt.Token, tokenString string, err error) {
 	t = jwt.New()
 	for k, v := range claims {
-		t.Set(k, v)
+		if err := t.Set(k, v); err != nil {
+			return nil, "", err
+		}
 	}
 	payload, err := ja.sign(t)
 	if err != nil {

jwtauth_test.go

@@ -279,6 +279,23 @@ func TestMore(t *testing.T) {
 	}
 }
 
+func TestEncodeClaims(t *testing.T) {
+	claims := map[string]interface{}{
+		"key1": "val1",
+		"key2": 2,
+		"key3": time.Now(),
+		"key4": []string{"1", "2"},
+	}
+	claims[jwt.JwtIDKey] = 1
+	if _, _, err := TokenAuthHS256.Encode(claims); err == nil {
+		t.Fatal("encoding invalid claims succeeded")
+	}
+	claims[jwt.JwtIDKey] = "123"
+	if _, _, err := TokenAuthHS256.Encode(claims); err != nil {
+		t.Fatalf("unexpected error encoding valid claims: %v", err)
+	}
+}
+
 //
 // Test helper functions
 //