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!'})