diff --git a/tests/pytest/util/sql.py b/tests/pytest/util/sql.py
index 19a5d3ecc1e314cbbb52f7c5249afd62c3303928..adbab04e07e31617b3f0aaf7abaa2f2a3c59b3dc 100644
--- a/tests/pytest/util/sql.py
+++ b/tests/pytest/util/sql.py
@@ -150,6 +150,9 @@ class TDSql:
             raise Exception(repr(e))
         return (self.queryRows, timeout)
 
+    def getRows(self):
+        return self.queryRows
+
     def checkRows(self, expectRows):
         if self.queryRows == expectRows:
             tdLog.info("sql:%s, queryRows:%d == expect:%d" % (self.sql, self.queryRows, expectRows))