From e1b0d4a555b383232241a39c83d728886d0ac60d Mon Sep 17 00:00:00 2001
From: Alex Duan <417921451@qq.com>
Date: Sat, 20 Jan 2024 14:26:34 +0800
Subject: [PATCH] fix: tweak delete max value

---
 tests/army/community/query/query_basic.py | 9 ++++++++-
 tests/army/frame/sql.py                   | 6 ++++++
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/tests/army/community/query/query_basic.py b/tests/army/community/query/query_basic.py
index eb332b8c71..575655c2b1 100644
--- a/tests/army/community/query/query_basic.py
+++ b/tests/army/community/query/query_basic.py
@@ -172,7 +172,14 @@ class TDTestCase(TBase):
             realMax = self.queryMax(colname)
             if realMax != expectMax:
                 tdLog.exit(f"Max value not expect. expect:{expectMax} real:{realMax}")
-            sql = f"delete from d0 where ui={expectMax}"
+
+            # query ts list
+            sql = f"select ts from d0 where ui={expectMax}"
+            tdSql.query(sql)
+            tss = tdSql.getColData(0)
+            strts = ",".join(tss)
+            # delete
+            sql = f"delete from d0 where ts in ({strts})"
             tdSql.execute(sql)
             expectMax -= 1
 
diff --git a/tests/army/frame/sql.py b/tests/army/frame/sql.py
index 8b7538321d..83f1243167 100644
--- a/tests/army/frame/sql.py
+++ b/tests/army/frame/sql.py
@@ -223,6 +223,12 @@ class TDSql:
     def getData(self, row, col):
         self.checkRowCol(row, col)
         return self.res[row][col]
+    
+    def getColData(self, col):
+        colDatas = []
+        for i in range(self.queryRows):
+            colDatas.append(self.res[i][col])
+        return colDatas
 
     def getResult(self, sql):
         self.sql = sql