fixed issue preventing unit tests from completing.

Author: travis-bauer

tests/conftest.py

@@ -26,6 +26,30 @@
 )
 
 
+def pytest_sessionfinish(session, exitstatus):
+    """Clean up async engine after all tests complete."""
+    import asyncio
+    # Dispose of the async engine to close all connections and threads
+    try:
+        # Try to dispose using the current event loop if available
+        try:
+            loop = asyncio.get_event_loop()
+            if not loop.is_closed() and not loop.is_running():
+                loop.run_until_complete(test_engine.dispose())
+            else:
+                # Create a new event loop for cleanup
+                asyncio.run(test_engine.dispose())
+        except RuntimeError:
+            # No event loop exists, create one
+            asyncio.run(test_engine.dispose())
+    except Exception:
+        # Fallback: try to close the underlying sync engine
+        try:
+            test_engine.sync_engine.dispose(close=True)
+        except Exception:
+            pass
+
+
 @pytest.fixture(scope="function")
 async def async_db_session() -> AsyncGenerator[AsyncSession, None]:
     """Create a fresh database for each test."""

tests/test_main_coverage.py

@@ -116,17 +116,16 @@ def test_get_user_preferences_empty(authenticated_client):
     assert data["preferences"] == {}
 
 
-def test_get_user_preferences_with_data(authenticated_client, async_db_session, test_user):
+async def test_get_user_preferences_with_data(authenticated_client, async_db_session, test_user):
     """Test getting user preferences with saved data."""
     import json
 
     # Set preferences on user
     test_user.preferences = json.dumps({"theme": "dark", "fontSize": 14})
     async_db_session.add(test_user)
 
-    # Use sync over async to commit
-    import asyncio
-    asyncio.get_event_loop().run_until_complete(async_db_session.commit())
+    # Commit the changes
+    await async_db_session.commit()
 
     response = authenticated_client.get("/user/preferences")
     assert response.status_code == 200
@@ -413,9 +412,10 @@ def test_create_snapshot_cleanup_old(authenticated_client):
     )
 
     # Create 12 snapshots
+    import time
     for i in range(12):
         authenticated_client.post("/documents/snapshot/cleanup.json")
-        asyncio.sleep(0.01)  # Small delay to ensure different timestamps
+        time.sleep(0.01)  # Small delay to ensure different timestamps
 
     # List snapshots
     response = authenticated_client.get("/documents/snapshots/cleanup.json")