Don’t depend on old private PyJWT API

base64url_decode wasn’t meant to be public API and has since been moved.
jpadilla · Feb 18, 2015 · 84f4157 · 84f4157
1 parent 744d90b
commit 84f4157
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -1,5 +1,5 @@
 import json
-from jwt import base64url_decode
+import base64
 
 from django.contrib.auth import get_user_model
 from django.test import TestCase
@@ -8,6 +8,15 @@
 User = get_user_model()
 
 
+def base64url_decode(input):
+    rem = len(input) % 4
+
+    if rem > 0:
+        input += b'=' * (4 - rem)
+
+    return base64.urlsafe_b64decode(input)
+
+
 class UtilsTests(TestCase):
     def setUp(self):
         self.username = 'jpueblo'