def test_default_revoked_token_callback(self):
with self.app.test_request_context():
m = JWTManager(self.app)
result = m._revoked_token_callback()
status_code, data = self._parse_callback_result(result)
self.assertEqual(status_code, 401)
self.assertEqual(data, {'msg': 'Token has been revoked'})
def test_custom_revoked_token_callback(self):
with self.app.test_request_context():
m = JWTManager(self.app)
@m.revoked_token_loader
def custom_revoken_token():
return jsonify({"err": "Nice knowing you!"}), 422
result = m._revoked_token_callback()
status_code, data = self._parse_callback_result(result)
self.assertEqual(status_code, 422)
self.assertEqual(data, {'err': 'Nice knowing you!'})
