From 18449f1995a85e14aaf49fc4f723775d44edd239 Mon Sep 17 00:00:00 2001 From: Adimanv Date: Tue, 24 Sep 2024 13:43:16 +0530 Subject: [PATCH] Client SQL removed --- database-cleanup/tests/conftest.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/database-cleanup/tests/conftest.py b/database-cleanup/tests/conftest.py index 9ec2f8e..6756139 100644 --- a/database-cleanup/tests/conftest.py +++ b/database-cleanup/tests/conftest.py @@ -2,7 +2,7 @@ import pytest import sqlalchemy as sa -from pvsite_datamodel.sqlmodels import Base, ClientSQL, SiteSQL +from pvsite_datamodel.sqlmodels import Base, SiteSQL from sqlalchemy.orm import Session from testcontainers.postgres import PostgresContainer @@ -29,11 +29,9 @@ def fill_db(engine): num_sites = 3 with Session(engine) as session: for i in range(num_clients): - client = ClientSQL(client_name=f"client-{i}") - session.add(client) session.commit() for j in range(num_sites): - session.add(SiteSQL(client_uuid=client.client_uuid, ml_id=j + num_sites * i)) + session.add(SiteSQL(ml_id=j + num_sites * i)) session.commit()