Fixed #183 AttributeError: 'NoneType' object has no attribute 'rsplit'

The issue also occurs when payload is int raising:
    AttributeError: 'int' object has no attribute 'rsplit'
Test for None and int payload added

2 files changed, 21 insertions, 0 deletions

diff --git a/jwt/api_jws.py b/jwt/api_jws.py
index 3e79d41..557812e 100644
--- a/jwt/api_jws.py
+++ b/jwt/api_jws.py
@@ -138,6 +138,9 @@ class PyJWS(object):
         try:
             signing_input, crypto_segment = jwt.rsplit(b'.', 1)
             header_segment, payload_segment = signing_input.split(b'.', 1)
+        except AttributeError:
+            raise DecodeError('Invalid payload type. It should be a {} not {}'.format(
+                text_type, type(jwt)))
         except ValueError:
             raise DecodeError('Not enough segments')
 
diff --git a/tests/test_api_jws.py b/tests/test_api_jws.py
index 2079bef..c08d3e4 100644
--- a/tests/test_api_jws.py
+++ b/tests/test_api_jws.py
@@ -122,6 +122,24 @@ class TestJWS:
         exception = context.value
         assert str(exception) == 'Not enough segments'
 
+    def test_decode_invalid_payload_type_is_none(self, jws):
+        example_jws = None
+        example_secret = 'secret'
+
+        with pytest.raises(DecodeError) as context:
+            jws.decode(example_jws, example_secret)
+
+        assert 'Invalid payload type' in str(context.value)
+
+    def test_decode_invalid_payload_type_is_int(self, jws):
+        example_jws = 123
+        example_secret = 'secret'
+
+        with pytest.raises(DecodeError) as context:
+            jws.decode(example_jws, example_secret)
+
+        assert 'Invalid payload type' in str(context.value)
+
     def test_decode_with_non_mapping_header_throws_exception(self, jws):
         secret = 'secret'
         example_jws = ('MQ'  # == 1