From 07e299932389d761cb09ce66c1bb18d98e829b81 Mon Sep 17 00:00:00 2001 From: caoxiaozhu Date: Sat, 25 Jul 2026 18:19:37 +0800 Subject: [PATCH] =?UTF-8?q?fix(data-process):=20=E4=BF=AE=E5=A4=8D?= =?UTF-8?q?=E4=B8=89=E6=95=B0=E6=8D=AE=E9=9B=86=E5=8F=91=E5=B8=83=E5=8F=82?= =?UTF-8?q?=E6=95=B0=E9=94=99=E8=AF=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/app/modules/data_process/store.py | 7 +++++-- backend/tests/test_data_process_store.py | 9 ++++++++- 2 files changed, 13 insertions(+), 3 deletions(-) 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"):