Fix some tests missing client fixture

langflow-ai · Oct 19, 2024 · 5456192 · 5456192
1 parent f96f2ea
commit 5456192
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 1 deletion.
diff --git a/...nents/models/test_ChatOllama_component.py → ...nents/models/test_chatollama_component.py b/...nents/models/test_ChatOllama_component.py → ...nents/models/test_chatollama_component.py
diff --git a/src/backend/tests/unit/test_cli.py b/src/backend/tests/unit/test_cli.py
@@ -25,7 +25,7 @@ def test_components_path(runner, default_settings, tmp_path):
     assert str(temp_dir) in settings_service.settings.components_path
 
 
-@pytest.mark.usefixtures("session")
+@pytest.mark.usefixtures("client")
 def test_superuser(runner):
     result = runner.invoke(app, ["superuser"], input="admin\nadmin\n")
     assert result.exit_code == 0, result.stdout

diff --git a/src/backend/tests/unit/test_database.py b/src/backend/tests/unit/test_database.py
@@ -513,6 +513,7 @@ async def test_load_flows(client: TestClient):
     assert response.json()["folder_id"] is not None
 
 
+@pytest.mark.usefixtures("client")
 def test_sqlite_pragmas():
     db_service = get_db_service()