def test_default_invalid_token_callback(self): with self.app.test_request_context(): m = JWTManager(self.app) err = "Test error" result = m._invalid_token_callback(err) status_code, data = self._parse_callback_result(result) self.assertEqual(status_code, 422) self.assertEqual(data, {'msg': err})
def test_custom_invalid_token_callback(self): with self.app.test_request_context(): m = JWTManager(self.app) err = "Test error" @m.invalid_token_loader def custom_invalid_token(err): return jsonify({"err": err}), 200 result = m._invalid_token_callback(err) status_code, data = self._parse_callback_result(result) self.assertEqual(status_code, 200) self.assertEqual(data, {'err': err})