chore: fix old module refs

tests/integration/conftest.py

@@ -30,7 +30,7 @@
 from argilla.datasets.__init__ import configure_dataset
 from argilla.server.database import get_async_db
 from argilla.server.models import User, UserRole, Workspace
-from argilla.server.server import app
+from argilla.server.app import app
 from argilla.server.settings import settings
 from argilla.utils import telemetry
 from argilla.utils.telemetry import TelemetryClient
@@ -132,7 +132,7 @@ async def override_get_async_db():
         session = TestSession()
         yield session
 
-    mocker.patch("argilla.server.server._get_db_wrapper", wraps=contextlib.asynccontextmanager(override_get_async_db))
+    mocker.patch("argilla.server.app._get_db_wrapper", wraps=contextlib.asynccontextmanager(override_get_async_db))
 
     app.dependency_overrides[get_async_db] = override_get_async_db
 

tests/unit/server/conftest.py

@@ -86,7 +86,7 @@ async def override_get_async_db():
     async def override_get_search_engine():
         yield mock_search_engine
 
-    mocker.patch("argilla.server.server._get_db_wrapper", wraps=contextlib.asynccontextmanager(override_get_async_db))
+    mocker.patch("argilla.server.app._get_db_wrapper", wraps=contextlib.asynccontextmanager(override_get_async_db))
 
     app.dependency_overrides[get_async_db] = override_get_async_db
     app.dependency_overrides[get_search_engine] = override_get_search_engine