From 78b5396f861393e4ecc00876689a88d81b333da4 Mon Sep 17 00:00:00 2001
From: factosea <285808407@qq.com>
Date: Tue, 24 Dec 2024 14:45:18 +0800
Subject: [PATCH] fix: test error

---
 tests/pytest/util/sql.py              | 7 +++----
 tests/system-test/0-others/udfTest.py | 2 +-
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/tests/pytest/util/sql.py b/tests/pytest/util/sql.py
index 46b7e1f795..bb7b8411f9 100644
--- a/tests/pytest/util/sql.py
+++ b/tests/pytest/util/sql.py
@@ -155,6 +155,9 @@ class TDSql:
 
         try:
             self.cursor.execute(sql)
+            self.queryResult = self.cursor.fetchall()
+            self.queryRows = len(self.queryResult)
+            self.queryCols = len(self.cursor.description)
         except BaseException as e:
             tdLog.info("err:%s" % (e))
             expectErrNotOccured = False
@@ -165,10 +168,6 @@ class TDSql:
         if expectErrNotOccured:
             tdLog.exit("%s(%d) failed: sql:%s, expect error not occured" % (caller.filename, caller.lineno, sql))
         else:
-            self.queryRows = 0
-            self.queryCols = 0
-            self.queryResult = None
-
             if fullMatched:
                 if expectedErrno != None:
                     expectedErrno_rest = expectedErrno & 0x0000ffff
diff --git a/tests/system-test/0-others/udfTest.py b/tests/system-test/0-others/udfTest.py
index 165ac5bc7d..8134327b41 100644
--- a/tests/system-test/0-others/udfTest.py
+++ b/tests/system-test/0-others/udfTest.py
@@ -740,7 +740,7 @@ class TDTestCase:
                 tdLog.info(f"create {func_name} functions success, using {self.libchange_udf_normal}")
                 break
                 
-        tdSql.query(f"select {func_name}(c8) from db.t1")
+        tdSql.error(f"select {func_name}(c8) from db.t1", TSDB_CODE_TSC_INTERNAL_ERROR)
         tdSql.execute(f"drop function {func_name}")
         tdSql.error(f"select {func_name}(num1) from db.tb", TSDB_CODE_MND_FUNC_NOT_EXIST)