From 265f2a858252c69f7a258e7ec018c25c767347b1 Mon Sep 17 00:00:00 2001
From: plum-lihui <huili@taosdata.com>
Date: Thu, 12 May 2022 16:36:08 +0800
Subject: [PATCH] test: add get rows api in test frame

---
 tests/pytest/util/sql.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/pytest/util/sql.py b/tests/pytest/util/sql.py
index 19a5d3ecc1..adbab04e07 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))
-- 
GitLab