From bbdda1fc5d24fac8126cb0a6333ff2304532a0f3 Mon Sep 17 00:00:00 2001
From: duwenxin <duwenxin@google.com>
Date: Tue, 10 Feb 2026 00:36:32 -0500
Subject: [PATCH] clickhouse

---
 tests/clickhouse/clickhouse_integration_test.go | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/tests/clickhouse/clickhouse_integration_test.go b/tests/clickhouse/clickhouse_integration_test.go
index b05f727283..6b0ae8d961 100644
--- a/tests/clickhouse/clickhouse_integration_test.go
+++ b/tests/clickhouse/clickhouse_integration_test.go
@@ -571,15 +571,15 @@ func TestClickHouseExecuteSQLTool(t *testing.T) {
 			resultSliceLen: 0,
 		},
 		{
-			name:           "MissingSQL",
-			sql:            "",
-			resultSliceLen: 1,
+			name:       "MissingSQL",
+			sql:        "",
+			isAgentErr: true,
 		},
 
 		{
-			name:           "SQLInjectionAttempt",
-			sql:            "SELECT 1; DROP TABLE system.users; SELECT 2",
-			resultSliceLen: 1,
+			name:       "SQLInjectionAttempt",
+			sql:        "SELECT 1; DROP TABLE system.users; SELECT 2",
+			isAgentErr: true,
 		},
 	}
 	for _, tc := range tcs {