From f80f4324b7db0d8c820c2cee2aaa272d96f17c27 Mon Sep 17 00:00:00 2001
From: Bomin Zhang <bmzhang@taosdata.com>
Date: Mon, 26 Oct 2020 18:47:33 +0800
Subject: [PATCH] TD-1287: fix failed test case

---
 tests/pytest/stream/new.py | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/tests/pytest/stream/new.py b/tests/pytest/stream/new.py
index eac93dc0e6..12ec6d4507 100644
--- a/tests/pytest/stream/new.py
+++ b/tests/pytest/stream/new.py
@@ -26,7 +26,6 @@ class TDTestCase:
 
     def run(self):
         rowNum = 200
-        totalNum = 200
         tdSql.prepare()
 
         tdLog.info("=============== step1")
@@ -42,7 +41,9 @@ class TDTestCase:
         tdSql.execute("create table st as select count(*), count(tbcol), count(tbcol2) from mt interval(10s)")
 
         tdLog.info("=============== step3")
+        start = time.time()
         tdSql.waitedQuery("select * from st", 1, 120)
+        delay = int(time.time() - start) + 20
         v = tdSql.getData(0, 3)
         if v >= 51:
             tdLog.exit("value is %d, which is larger than 51" % v)
@@ -54,11 +55,18 @@ class TDTestCase:
                 tdSql.execute("insert into tb%d values(now + %ds, %d, %d)" % (i, j, j, j))
 
         tdLog.info("=============== step5")
-        tdLog.sleep(40)
-        tdSql.waitedQuery("select * from st order by ts desc", 1, 120)
-        v = tdSql.getData(0, 3)
-        if v <= 51:
-            tdLog.exit("value is %d, which is smaller than 51" % v)
+        maxValue = 0
+        for i in range(delay):
+            time.sleep(1)
+            tdSql.query("select * from st order by ts desc")
+            v = tdSql.getData(0, 3)
+            if v > maxValue:
+                maxValue = v
+            if v > 51:
+                break
+
+        if maxValue <= 51:
+            tdLog.exit("value is %d, which is smaller than 51" % maxValue)
 
     def stop(self):
         tdSql.close()
-- 
GitLab