chore: try fix web app test flakiness (#3126)

Author: eakmanrq

tests/web/conftest.py

@@ -1,14 +1,22 @@
 from pathlib import Path
 
 import pytest
+from fastapi import FastAPI
 
 from sqlmesh.core.context import Context
-from web.server.main import api_console, app
+from web.server.console import api_console
 from web.server.settings import Settings, get_loaded_context, get_settings
 
 
 @pytest.fixture
-def project_tmp_path(tmp_path: Path):
+def web_app() -> FastAPI:
+    from web.server.main import create_app
+
+    return create_app()
+
+
+@pytest.fixture
+def project_tmp_path(web_app: FastAPI, tmp_path: Path):
     def get_settings_override() -> Settings:
         return Settings(project_path=tmp_path)
 
@@ -19,29 +27,29 @@ def get_settings_override() -> Settings:
     """
     )
 
-    app.dependency_overrides[get_settings] = get_settings_override
+    web_app.dependency_overrides[get_settings] = get_settings_override
     yield tmp_path
-    app.dependency_overrides = {}
+    web_app.dependency_overrides = {}
 
 
 @pytest.fixture
-def project_context(project_tmp_path: Path):
+def project_context(web_app: FastAPI, project_tmp_path: Path):
     context = Context(paths=project_tmp_path, console=api_console)
 
     def get_loaded_context_override() -> Context:
         return context
 
-    app.dependency_overrides[get_loaded_context] = get_loaded_context_override
+    web_app.dependency_overrides[get_loaded_context] = get_loaded_context_override
     yield context
-    app.dependency_overrides = {}
+    web_app.dependency_overrides = {}
 
 
 @pytest.fixture
-def web_sushi_context(sushi_context: Context):
+def web_sushi_context(web_app: FastAPI, sushi_context: Context):
     def get_context_override() -> Context:
         sushi_context.console = api_console
         return sushi_context
 
-    app.dependency_overrides[get_loaded_context] = get_context_override
+    web_app.dependency_overrides[get_loaded_context] = get_context_override
     yield sushi_context
-    app.dependency_overrides = {}
+    web_app.dependency_overrides = {}

tests/web/test_lineage.py

@@ -1,17 +1,17 @@
 from __future__ import annotations
 
 import pytest
+from fastapi import FastAPI
 from fastapi.testclient import TestClient
 
 from sqlmesh.core.context import Context
-from web.server.main import app
 
 pytestmark = pytest.mark.web
 
 
 @pytest.fixture
-def client() -> TestClient:
-    return TestClient(app)
+def client(web_app: FastAPI) -> TestClient:
+    return TestClient(web_app)
 
 
 def test_get_lineage(client: TestClient, web_sushi_context: Context) -> None: