update to test app

openclimatefix · Sep 27, 2024 · f11ac42 · f11ac42
1 parent f7b928c
commit f11ac42
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/database-cleanup/tests/test_app.py b/database-cleanup/tests/test_app.py
@@ -54,11 +54,9 @@ def _run_cli(func, args: list[str]):
 @pytest.fixture
 def site(session):
     # Create a new site (this way we know it won't have any forecasts yet).
-    client = str(uuid.uuid4())
-    session.add(client)
-    session.commit()
-
     site = SiteSQL( ml_id=hash(uuid.uuid4()) % 2147483647)
+    session.add(site)
+    session.commit()
     return site