From ad2af696954127ffb0bb96bf770c354bc6aff4ea Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Tue, 28 Jul 2026 20:14:45 +0000 Subject: [PATCH] Validate table names in test row counter --- backend/tests/integration_tests/sql.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/backend/tests/integration_tests/sql.py b/backend/tests/integration_tests/sql.py index 8d726b78..89e326b8 100644 --- a/backend/tests/integration_tests/sql.py +++ b/backend/tests/integration_tests/sql.py @@ -31,8 +31,12 @@ from bracket.utils.id_types import TeamId from tests.integration_tests.mocks import get_mock_token, get_mock_user from tests.integration_tests.models import AuthContext +ROW_COUNT_TABLES = {"courts", "matches", "players", "rounds", "stage_items", "stages", "teams"} + async def assert_row_count_and_clear(table_name: str, expected_rows: int) -> None: + if table_name not in ROW_COUNT_TABLES: + raise ValueError(f"Unsupported table for row count assertion: {table_name}") rows = await database.fetch_val(query=f"SELECT COUNT(*) FROM {table_name}") assert rows == expected_rows await database.execute(query=f"DELETE FROM {table_name}")