diff --git a/studio/backend/tests/test_desktop_auth.py b/studio/backend/tests/test_desktop_auth.py index af0a4b940f..12cf071141 100644 --- a/studio/backend/tests/test_desktop_auth.py +++ b/studio/backend/tests/test_desktop_auth.py @@ -423,6 +423,8 @@ def test_health_response_reports_desktop_capability_fields(monkeypatch): llama_module.router = APIRouter() prompts_module = ModuleType("routes.prompts") prompts_module.router = APIRouter() + preview_module = ModuleType("routes.preview") + preview_module.router = APIRouter() for name, router in { "auth_router": APIRouter(), @@ -449,6 +451,7 @@ def test_health_response_reports_desktop_capability_fields(monkeypatch): monkeypatch.setitem(sys.modules, "routes.settings", settings_module) monkeypatch.setitem(sys.modules, "routes.llama", llama_module) monkeypatch.setitem(sys.modules, "routes.prompts", prompts_module) + monkeypatch.setitem(sys.modules, "routes.preview", preview_module) import studio.backend.main as backend_main