Fix some unit tests

Author: vimalloc

tests/test_blacklist.py

@@ -1,7 +1,6 @@
 import unittest
 import json
 
-import simplekv.memory
 from flask import Flask, jsonify, request
 from flask_jwt_extended.utils import get_jwt_identity, get_jti
 
@@ -17,7 +16,6 @@ def setUp(self):
         self.app = Flask(__name__)
         self.app.secret_key = 'super=secret'
         self.app.config['JWT_BLACKLIST_ENABLED'] = True
-        self.app.config['JWT_BLACKLIST_STORE'] = simplekv.memory.DictStore()
         self.jwt_manager = JWTManager(self.app)
         self.client = self.app.test_client()
         self.blacklist = set()

tests/test_config.py

@@ -85,7 +85,6 @@ def test_override_configs(self):
         self.app.config['JWT_ALGORITHM'] = 'HS512'
 
         self.app.config['JWT_BLACKLIST_ENABLED'] = True
-        self.app.config['JWT_BLACKLIST_STORE'] = sample_store
         self.app.config['JWT_BLACKLIST_TOKEN_CHECKS'] = 'refresh'
 
         self.app.secret_key = 'banana'

tests/test_jwt_encode_decode.py

@@ -36,7 +36,7 @@ def test_encode_access_token(self):
             identity = 'user1'
             token = encode_access_token(identity, secret, algorithm, token_expire_delta,
                                         fresh=True, user_claims=user_claims, csrf=False)
-            data = jwt.decode(token, secret, algorithm=algorithm)
+            data = jwt.decode(token, secret, algorithms=[algorithm])
             self.assertIn('exp', data)
             self.assertIn('iat', data)
             self.assertIn('nbf', data)
@@ -60,7 +60,7 @@ def test_encode_access_token(self):
             identity = 12345  # identity can be anything json serializable
             token = encode_access_token(identity, secret, algorithm, token_expire_delta,
                                         fresh=False, user_claims=user_claims, csrf=True)
-            data = jwt.decode(token, secret, algorithm=algorithm)
+            data = jwt.decode(token, secret, algorithms=[algorithm])
             self.assertIn('exp', data)
             self.assertIn('iat', data)
             self.assertIn('nbf', data)
@@ -105,7 +105,7 @@ def test_encode_refresh_token(self):
             identity = 'user1'
             token = encode_refresh_token(identity, secret, algorithm,
                                          token_expire_delta, csrf=False)
-            data = jwt.decode(token, secret, algorithm=algorithm)
+            data = jwt.decode(token, secret, algorithms=[algorithm])
             self.assertIn('exp', data)
             self.assertIn('iat', data)
             self.assertIn('nbf', data)
@@ -125,7 +125,7 @@ def test_encode_refresh_token(self):
             identity = 12345  # identity can be anything json serializable
             token = encode_refresh_token(identity, secret, algorithm,
                                          token_expire_delta, csrf=True)
-            data = jwt.decode(token, secret, algorithm=algorithm)
+            data = jwt.decode(token, secret, algorithms=[algorithm])
             self.assertIn('exp', data)
             self.assertIn('iat', data)
             self.assertIn('nbf', data)