Fixup setup_users in celery rate limit integration test

galaxyproject · May 13, 2024 · dfe9937 · dfe9937
1 parent d649b90
commit dfe9937
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/lib/tool_shed/managers/api_keys.py b/lib/tool_shed/managers/api_keys.py
@@ -1,5 +1,8 @@
 from galaxy.managers.api_keys import ApiKeyManager
-from tool_shed.webapp.model import APIKeys, User
+from tool_shed.webapp.model import (
+    APIKeys,
+    User,
+)
 
 
 class ToolShedApiKeyManager(ApiKeyManager[APIKeys, User]):

diff --git a/test/integration/test_celery_user_rate_limit.py b/test/integration/test_celery_user_rate_limit.py
@@ -49,8 +49,10 @@ def setup_users(dburl: str, num_users: int = 2):
                 user_ids_to_add = set(expected_user_ids).difference(found_user_ids)
                 for user_id in user_ids_to_add:
                     conn.execute(
-                        text("insert into galaxy_user(id, active, email, password) values (:id, :active, :email, :pw)"),
-                        [{"id": user_id, "active": True, "email": "e", "pw": "p"}],
+                        text(
+                            "insert into galaxy_user(id, active, email, password, username) values (:id, :active, :email, :pw)"
+                        ),
+                        [{"id": user_id, "active": True, "email": "e", "pw": "p", "username": f"u{user_id}"}],
                     )