def test_default_unauthorized_callback(self):
with self.app.test_request_context():
m = JWTManager(self.app)
result = m._unauthorized_callback("Missing Authorization Header")
status_code, data = self._parse_callback_result(result)
self.assertEqual(status_code, 401)
self.assertEqual(data, {'msg': 'Missing Authorization Header'})
def test_custom_unauthorized_callback(self):
with self.app.test_request_context():
m = JWTManager(self.app)
@m.unauthorized_loader
def custom_unauthorized(err_str):
return jsonify({"err": err_str}), 200
result = m._unauthorized_callback("GOTTA LOGIN FOOL")
status_code, data = self._parse_callback_result(result)
self.assertEqual(status_code, 200)
self.assertEqual(data, {'err': 'GOTTA LOGIN FOOL'})
