diff --git a/tests/pytest/stream/showStreamExecTimeisNull.py b/tests/pytest/stream/showStreamExecTimeisNull.py index 39b025901856c1de6c21a1889d97a4b599c53de4..0ffbdb204210ddd6d6b56ed4416241562bfe832b 100644 --- a/tests/pytest/stream/showStreamExecTimeisNull.py +++ b/tests/pytest/stream/showStreamExecTimeisNull.py @@ -67,7 +67,9 @@ class TDTestCase: self.queryRows = len(self.queryResult) self.queryCols = len(tdSql.cursor.description) # tdLog.info("sql: %s, try to retrieve %d rows,get %d rows" % (sql, expectRows, self.queryRows)) - if self.queryRows >= timeout: + if self.queryRows >= 1: + tdSql.query(sql) + tdSql.checkData(0, 5, None) return (self.queryRows, i) time.sleep(1) except Exception as e: @@ -75,6 +77,8 @@ class TDTestCase: else: tdLog.exit(f"sql: {sql} except raise {exception}, actually not") + tdSql.checkData(0, 5, None) + def run(self): tdSql.execute("drop database if exists dbcq")