renamed

djeck1432 · Nov 25, 2024 · 5965bc2 · 5965bc2
1 parent 488b21b
commit 5965bc2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/web_app/tests/test_integration/__init__.py → web_app/test_integration/__init__.py b/web_app/tests/test_integration/__init__.py → web_app/test_integration/__init__.py
diff --git a/...s/test_integration/test_close_position.py → ...p/test_integration/test_close_position.py b/...s/test_integration/test_close_position.py → ...p/test_integration/test_close_position.py
@@ -4,10 +4,11 @@
 
 import pytest
 from web_app.contract_tools.mixins.dashboard import DashboardMixin
-from web_app.db.crud import PositionDBConnector, UserDBConnector
+from web_app.db.crud import PositionDBConnector, UserDBConnector, AirDropDBConnector
 from web_app.db.models import Status
 
 user_db = UserDBConnector()
+airdrop = AirDropDBConnector()
 position_db = PositionDBConnector()
 
 
@@ -98,6 +99,8 @@ def test_close_position(self, form_data: Dict[str, Any]) -> None:
         ), "Position status should be 'closed' after close operation"
 
         # Clean up - delete the position and user
+        user = position_db.get_user_by_wallet_id(wallet_id)
+        airdrop.delete_all_users_airdrop(user.id)
         position_db.delete_position(position)
         if not position_db.get_positions_by_wallet_id(wallet_id):
             position_db.delete_user_by_wallet_id(wallet_id)
diff --git a/.../test_integration/test_create_position.py → .../test_integration/test_create_position.py b/.../test_integration/test_create_position.py → .../test_integration/test_create_position.py