From 318293edf182c21f5837b6571dc165c53b6aa9fe Mon Sep 17 00:00:00 2001
From: Alex Duan <417921451@qq.com>
Date: Mon, 15 Jan 2024 14:14:25 +0800
Subject: [PATCH] fix: countAlways return is 0

---
 tests/army/community/cluster/snapshot.py | 4 ++--
 tests/army/frame/caseBase.py             | 5 +++--
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/tests/army/community/cluster/snapshot.py b/tests/army/community/cluster/snapshot.py
index 8543101300..8d15530061 100644
--- a/tests/army/community/cluster/snapshot.py
+++ b/tests/army/community/cluster/snapshot.py
@@ -72,8 +72,8 @@ class TDTestCase(TBase):
 
         # check count always return value
         sql = f"select count(*) from {self.db}.ta"
-        tdSql.waitedQuery(sql, 0, 2)        
-
+        tdSql.query()
+        tdSql.checkRows(0) # countAlwaysReturnValue is false
 
     # run
     def run(self):
diff --git a/tests/army/frame/caseBase.py b/tests/army/frame/caseBase.py
index ca5539f75b..aca2b524da 100644
--- a/tests/army/frame/caseBase.py
+++ b/tests/army/frame/caseBase.py
@@ -132,8 +132,9 @@ class TBase:
         tdSql.checkAgg(sql, self.childtable_count)
 
         # check step
-        sql = f"select count(*) from (select diff(ts) as dif from {self.stb} partition by tbname) where dif != {self.timestamp_step}"
-        tdSql.checkAgg(sql, 0)
+        sql = f"select * from (select diff(ts) as dif from {self.stb} partition by tbname) where dif != {self.timestamp_step}"
+        tdSql.query(sql)
+        tdSql.checkRows(0)
 
     # save agg result
     def snapshotAgg(self):