def test_default_expired_token_callback(self):
with self.app.test_request_context():
m = JWTManager(self.app)
result = m._expired_token_callback()
status_code, data = self._parse_callback_result(result)
self.assertEqual(status_code, 401)
self.assertEqual(data, {'msg': 'Token has expired'})
def test_custom_expired_token_callback(self):
with self.app.test_request_context():
m = JWTManager(self.app)
@m.expired_token_loader
def custom_expired_token():
return jsonify({"res": "TOKEN IS EXPIRED FOOL"}), 422
result = m._expired_token_callback()
status_code, data = self._parse_callback_result(result)
self.assertEqual(status_code, 422)
self.assertEqual(data, {'res': 'TOKEN IS EXPIRED FOOL'})
