From 0e868735beb99b8f3d9c40bf47b0a4934895a701 Mon Sep 17 00:00:00 2001
From: wangjiaming0909 <604227650@qq.com>
Date: Wed, 10 Jul 2024 19:10:17 +0800
Subject: [PATCH] fill add test

---
 tests/system-test/2-query/fill.py | 74 +++++++++++++++++++++----------
 1 file changed, 51 insertions(+), 23 deletions(-)

diff --git a/tests/system-test/2-query/fill.py b/tests/system-test/2-query/fill.py
index 274e0710bd..8ea4622830 100644
--- a/tests/system-test/2-query/fill.py
+++ b/tests/system-test/2-query/fill.py
@@ -1,8 +1,10 @@
 import random
+from fabric2.runners import threading
 from pandas._libs import interval
 import taos
 import sys
 
+from util.common import TDCom
 from util.log import *
 from util.sql import *
 from util.cases import *
@@ -26,16 +28,20 @@ class TDTestCase:
                 ret.append((i,j))
         return ret
 
-    def check_fill_range(self, where_start, where_end, res, sql: str):
-        if len(res) == 0:
-            tdLog.debug(f'fill sql got no rows {sql}')
+    def check_fill_range(self, where_start, where_end, res_asc, res_desc, sql: str, interval):
+        if len(res_asc) != len(res_desc):
+            tdLog.exit(f"err, asc desc with different rows, asc: {len(res_asc)}, desc: {len(res_desc)} sql: {sql}")
+        if len(res_asc) == 0:
+            tdLog.info(f'from {where_start} to {where_end} no rows returned')
             return
-        if len(res) == 1:
-            tdLog.debug(f'fill sql got one row {sql}: {res}')
+        asc_first = res_asc[0]
+        asc_last = res_asc[-1]
+        desc_first = res_desc[0]
+        desc_last = res_desc[-1]
+        if asc_first[0] != desc_last[0] or asc_last[0] != desc_first[0]:
+            tdLog.exit(f'fill sql different row data {sql}: asc<{asc_first[0].timestamp()}, {asc_last[0].timestamp()}>, desc<{desc_last[0].timestamp()}, {desc_first[0].timestamp()}>')
         else:
-            first = res[0]
-            last = res[-1]
-            tdLog.debug(f'fill sql got   rows {sql}: {res}')
+            tdLog.info(f'from {where_start} to {where_end} same time returned asc<{asc_first[0].timestamp()}, {asc_last[0].timestamp()}>, desc<{desc_last[0].timestamp()}, {desc_first[0].timestamp()}> interval: {interval}')
 
     def generate_partition_by(self):
         val = random.random()
@@ -48,36 +54,58 @@ class TDTestCase:
 
     def generate_fill_interval(self)-> list[tuple]:
         ret = []
-        intervals = [1, 30, 60, 90, 120, 300, 3600]
+        intervals = [60, 90, 120, 300, 3600]
         for i in range(0, len(intervals)):
             for j in range(0, i+1):
                 ret.append((intervals[i], intervals[j]))
         return ret
 
-    def generate_fill_sql(self, where_start, where_end, fill_interval: tuple) -> str:
+    def generate_fill_sql(self, where_start, where_end, fill_interval: tuple):
         partition_by = self.generate_partition_by()
-        where = f'ts >= {where_start} and ts < {where_end}'
-        return f'select _wstart, _wend, count(*) from meters {where} {partition_by} interval({fill_interval[0]}s) sliding({fill_interval[1]}s) fill(NULL)'
+        where = f'where ts >= {where_start} and ts < {where_end}'
+        sql = f'select _wstart, _wend, count(*) from test.meters {where} {partition_by} interval({fill_interval[0]}s) sliding({fill_interval[1]}s) fill(NULL)'
+        sql_asc = sql + " order by _wstart asc"
+        sql_desc = sql + " order by _wstart desc"
+        return sql_asc, sql_desc
+
+    def fill_test_thread_routine(self, cli: TDSql, interval, data_start, data_end, step):
+        ranges = self.generate_fill_range(data_start, data_end, interval[0], step)
+        for range in ranges:
+            sql_asc, sql_desc = self.generate_fill_sql(range[0], range[1], interval)
+            cli.query(sql_asc, queryTimes=1)
+            asc_res = cli.queryResult
+            cli.query(sql_desc, queryTimes=1)
+            desc_res = cli.queryResult
+            self.check_fill_range(range[0], range[1], asc_res,desc_res , sql_asc, interval)
 
     def test_fill_range(self):
-        os.system('taosBenchmark -t 1000 -n 1000 -v 2 -S 32000 -y')
-        data_start = 1500000000
-        data_end = 1500031968
-        step = 100
+        os.system('taosBenchmark -t 10 -n 10000 -v 8 -S 32000 -y')
+        data_start = 1500000000000
+        data_end = 1500319968000
+        step = 2000000
+        tdCom = TDCom()
+        inses: list[TDSql] = []
+        threads: list[threading.Thread] = []
 
         fill_intervals: list[tuple] = self.generate_fill_interval()
-        fill_interval = random.choice(fill_intervals)
-        ranges = self.generate_fill_range(data_start, data_end, fill_interval[0], step)
-        range = random.choice(ranges)
-        sql = self.generate_fill_sql(range[0], range[1], fill_interval)
-        tdSql.query(sql)
-        res = tdSql.queryResult
-        self.check_fill_range(range[0], range[1], res, sql)
+        for interval in fill_intervals:
+            ins = tdCom.newTdSql()
+            t = threading.Thread(target=self.fill_test_thread_routine, args=(ins, interval, data_start, data_end, step))
+            t.start()
+            inses.append(ins)
+            threads.append(t)
+
+        for t in threads:
+            t.join()
+
+        for ins in inses:
+            ins.close()
 
         ## tdSql.execute('drop database test')
 
     def run(self):
         self.test_fill_range()
+        return
         dbname = "db"
         tbname = "tb"