diff --git a/backend/app/modules/data_process/store.py b/backend/app/modules/data_process/store.py index 6945597..09c0de2 100644 --- a/backend/app/modules/data_process/store.py +++ b/backend/app/modules/data_process/store.py @@ -1399,8 +1399,11 @@ class DataProcessStore: size, size_bytes, count, record_count, description, metadata, tenant_id, project_id, owner_id, created_by, create_time, created_at, updated_at) - VALUES (%s, %s, %s, %s, 'task', %s, %s, %s, %s, %s, %s, - %s, %s, %s, %s, %s, %s, %s, %s, %s, %s) + VALUES ( + %s, %s, %s, %s, 'task', %s, %s, + %s, %s, %s, %s, %s, %s, + %s, %s, %s, %s, %s, %s, %s + ) RETURNING * """, ( diff --git a/backend/tests/test_data_process_store.py b/backend/tests/test_data_process_store.py index eec8e4d..604ae21 100644 --- a/backend/tests/test_data_process_store.py +++ b/backend/tests/test_data_process_store.py @@ -1,8 +1,9 @@ from __future__ import annotations +from collections.abc import Iterator from contextlib import contextmanager from decimal import Decimal -from typing import Any, Iterator +from typing import Any import pytest @@ -35,6 +36,12 @@ class _PublishConnection: def execute(self, sql: str, params: Any = None) -> _Result: normalized = " ".join(sql.split()) + if params is not None: + placeholder_count = normalized.count("%s") + assert placeholder_count == len(params), ( + f"SQL placeholder count {placeholder_count} does not match " + f"parameter count {len(params)}" + ) if normalized.startswith("SELECT * FROM data_process_results"): return _Result(rows=self.results) if normalized.startswith("SELECT * FROM datasets WHERE source_task_id"):