Merge pull request jpadilla#12 from Lothiraldan/master

Add a check in jwt.encode for data type.

Author: progrium

jwt/__init__.py

@@ -10,6 +10,7 @@
 from time import time
 from datetime import datetime
 from calendar import timegm
+from collections import Mapping
 
 try:
     import json
@@ -70,6 +71,11 @@ def header(jwt):
 def encode(payload, key, algorithm='HS256'):
     segments = []
 
+    # Check that we get a mapping
+    if not isinstance(payload, Mapping):
+        raise TypeError("Expecting a mapping object, as json web token only"
+                        "support json objects.")
+
     # Header
     header = {"typ": "JWT", "alg": algorithm}
     segments.append(base64url_encode(json.dumps(header)))

tests/test_jwt.py

@@ -22,6 +22,14 @@ def test_encode_decode(self):
         decoded_payload = jwt.decode(jwt_message, secret)
         self.assertEqual(decoded_payload, self.payload)
 
+    def test_encode_bad_type(self):
+
+        types = ['string', tuple(), list(), 42, set()]
+
+        for t in types:
+            with self.assertRaises(TypeError):
+                jwt.encode(t, 'secret')
+
     def test_encode_expiration_datetime(self):
         secret = "secret"
         current_datetime = datetime.utcnow()