diff --git a/tests/manager/conftest.py b/tests/manager/conftest.py index 219a1a91a1..b919904749 100644 --- a/tests/manager/conftest.py +++ b/tests/manager/conftest.py @@ -67,10 +67,12 @@ vfolders, ) from ai.backend.manager.models.base import ( - load_table, + metadata, pgsql_connect_opts, populate_fixture, ) + +# from ai.backend.manager.models.container_registry import ContainerRegistry from ai.backend.manager.models.scaling_group import ScalingGroupOpts from ai.backend.manager.models.utils import connect_database from ai.backend.manager.registry import AgentRegistry @@ -415,6 +417,7 @@ def database_fixture(local_config, test_db, database): Populate the example data as fixtures to the database and delete them after use. """ + print("database_fixture!!!ewodkpw") db_addr = local_config["db"]["addr"] db_user = local_config["db"]["user"] db_pass = local_config["db"]["password"] @@ -464,7 +467,8 @@ async def clean_fixture(): await conn.execute((users.delete())) await conn.execute((scaling_groups.delete())) await conn.execute((domains.delete())) - await conn.execute(((await load_table(engine, "container_registries")).delete())) + # await conn.execute(((await load_table(engine, "container_registries")).delete())) + await conn.execute((metadata.tables.get("container_registries"))) finally: await engine.dispose()