Handle unicode strings for base64 decoding

manifoldco · Apr 19, 2017 · 18f1f24 · 18f1f24
1 parent a8f76cd
commit 18f1f24
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/manifoldco_signature/__init__.py b/manifoldco_signature/__init__.py
@@ -110,7 +110,7 @@ def _decode_base64(data):
     missing_padding = len(data) % 4
     if missing_padding != 0:
         data += '='* (4 - missing_padding)
-    return base64.urlsafe_b64decode(data)
+    return base64.urlsafe_b64decode(str(data))
 
 from ._version import get_versions
 __version__ = get_versions()['version']

diff --git a/tests/test_manifoldco_signature.py b/tests/test_manifoldco_signature.py
@@ -1,3 +1,4 @@
+from __future__ import unicode_literals
 from builtins import bytes
 
 import pytest