merged the jwt tests

gbin · gbin · commit 3997bccd2806 · 2013-01-04T14:41:34.000-08:00
diff --git a/tests/test_jwt.py b/tests/test_jwt.py
@@ -1,3 +1,4 @@
+import time
 from twilio import jwt
 import sys
 if sys.version_info < (2, 7):
@@ -7,7 +8,7 @@
 
 from twilio.util import TwilioCapability
 
-class TokenTest(unittest.TestCase):
+class JwtTest(unittest.TestCase):
 
     def assertIn(self, foo, bar, msg=None):
         """backport for 2.6"""
@@ -80,3 +81,43 @@ def test_decode(self):
         self.assertIn(outgoing_uri, scope)
         self.assertIn(incoming_uri, scope)
         self.assertIn(event_uri, scope)
+
+    def setUp(self):
+        self.payload = {"iss": "jeff", "exp": int(time.time()), "claim": "insanity"}
+
+    def test_encode_decode(self):
+        secret = 'secret'
+        jwt_message = jwt.encode(self.payload, secret)
+        decoded_payload = jwt.decode(jwt_message, secret)
+        self.assertEqual(decoded_payload, self.payload)
+
+    def test_bad_secret(self):
+        right_secret = 'foo'
+        bad_secret = 'bar'
+        jwt_message = jwt.encode(self.payload, right_secret)
+        self.assertRaises(jwt.DecodeError, jwt.decode, jwt_message, bad_secret)
+
+    def test_decodes_valid_jwt(self):
+        example_payload = {"hello": "world"}
+        example_secret = "secret"
+        example_jwt = "eyJhbGciOiAiSFMyNTYiLCAidHlwIjogIkpXVCJ9.eyJoZWxsbyI6ICJ3b3JsZCJ9.tvagLDLoaiJKxOKqpBXSEGy7SYSifZhjntgm9ctpyj8"
+        decoded_payload = jwt.decode(example_jwt, example_secret)
+        self.assertEqual(decoded_payload, example_payload)
+
+    def test_allow_skip_verification(self):
+        right_secret = 'foo'
+        bad_secret = 'bar'
+        jwt_message = jwt.encode(self.payload, right_secret)
+        decoded_payload = jwt.decode(jwt_message, verify=False)
+        self.assertEqual(decoded_payload, self.payload)
+
+    def test_no_secret(self):
+        right_secret = 'foo'
+        bad_secret = 'bar'
+        jwt_message = jwt.encode(self.payload, right_secret)
+        self.assertRaises(jwt.DecodeError, jwt.decode, jwt_message)
+
+    def test_invalid_crypto_alg(self):
+        self.assertRaises(NotImplementedError, jwt.encode, self.payload, "secret", "HS1024")
+
+
diff --git a/tests/test_validation.py b/tests/test_validation.py
@@ -52,7 +52,7 @@ def test_validation(self):
         self.assertTrue(validator.validate(uri, params, expected))
 
     @unittest.skip("utf 8 support still a work in progress")
-    def test_international_sms(self):
+    def te_st_international_sms(self):
 
         token = os.environ["TWILIO_AUTH_TOKEN"]
         validator = RequestValidator(token)
