monoadmin/the-other-dude
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(ci): filter cleanup tables to only those that exist

Browse Source 
Tables like invites/user_tenants only exist on saas-tiers branch.
Query pg_tables to skip missing tables in TRUNCATE.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Jason Staack
2026-03-15 06:31:21 -05:00
parent 9085d90b93
commit 6393945505
1 changed files with 16 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
backend/tests/integration/conftest.py
View File

@@ -165,13 +165,24 @@ async def admin_session(admin_engine) -> AsyncGenerator[AsyncSession, None]:
Cleanup deletes all rows from test tables after the test. Cleanup deletes all rows from test tables after the test.
""" """
session = AsyncSession(admin_engine, expire_on_commit=False) session = AsyncSession(admin_engine, expire_on_commit=False)
# TRUNCATE CASCADE reliably removes all data regardless of FK order # TRUNCATE CASCADE reliably removes all data regardless of FK order.
tables_csv = ", ".join(_CLEANUP_TABLES) # Filter to only tables that exist (some are saas-tiers only).
existing = await session.execute(
text(
"SELECT tablename FROM pg_tables "
"WHERE schemaname = 'public' AND tablename = ANY(:names)"
),
{"names": _CLEANUP_TABLES},
)
existing_tables = [row[0] for row in existing.fetchall()]
if existing_tables:
tables_csv = ", ".join(existing_tables)
await session.execute(text(f"TRUNCATE {tables_csv} CASCADE")) await session.execute(text(f"TRUNCATE {tables_csv} CASCADE"))
await session.commit() await session.commit()
try: try:
yield session yield session
finally: finally:
if existing_tables:
await session.execute(text(f"TRUNCATE {tables_csv} CASCADE")) await session.execute(text(f"TRUNCATE {tables_csv} CASCADE"))
await session.commit() await session.commit()
await session.close() await session.close()