From befaee12a5889b930ca622144f131d5ad5473c90 Mon Sep 17 00:00:00 2001
From: Alex Duan <417921451@qq.com>
Date: Mon, 9 Dec 2024 16:57:04 +0800
Subject: [PATCH] case: test passed ok

---
 tests/parallel_test/cases.task                |  1 +
 .../eco-system/manager/schema_change.py       | 46 ++++++++++++++++---
 2 files changed, 41 insertions(+), 6 deletions(-)

diff --git a/tests/parallel_test/cases.task b/tests/parallel_test/cases.task
index a635ae680a..47a6b418f6 100644
--- a/tests/parallel_test/cases.task
+++ b/tests/parallel_test/cases.task
@@ -1082,6 +1082,7 @@
 ,,n,system-test,python3 ./test.py -f 5-taos-tools/taosbenchmark/insertMix.py -N 3
 ,,n,system-test,python3 ./test.py -f 5-taos-tools/taosbenchmark/stt.py -N 3
 ,,n,system-test,python3 ./test.py -f eco-system/meta/database/keep_time_offset.py
+,,y,system-test,./pytest.sh python3 ./test.py -f eco-system/manager/schema_change.py -N 3 -M 3
 
 #tsim test
 ,,y,script,./test.sh -f tsim/query/timeline.sim
diff --git a/tests/system-test/eco-system/manager/schema_change.py b/tests/system-test/eco-system/manager/schema_change.py
index 400d2b100b..8eef773d48 100644
--- a/tests/system-test/eco-system/manager/schema_change.py
+++ b/tests/system-test/eco-system/manager/schema_change.py
@@ -133,8 +133,17 @@ class TDTestCase:
         if len(tags) < 10:
             return 
 
-        sel_cols = random.sample(columns, random.randint(2,int(len(columns)/10)))
-        sel_tags = random.sample(tags, random.randint(1,int(len(tags)/10)))
+        # cmax
+        cmax = int(len(columns)/10) 
+        if cmax <=2 :
+            cmax = 3
+        sel_cols = random.sample(columns, random.randint(2, cmax))
+        
+        # tmax
+        tmax = int(len(tags)/10)
+        if tmax <=1 :
+            tmax = 2
+        sel_tags = random.sample(tags,    random.randint(1, tmax))
 
         field_cols = ",".join(sel_cols)
         field_tags = ",".join(sel_tags)
@@ -217,18 +226,43 @@ class TDTestCase:
             #time.sleep(0.3)
 
 
+    # create db and stb
+    def create(self, db, stb, cols, tags):
+        # create db
+        sql = f"create database {db} ;"
+        tdSql.execute(sql)
+
+        # switch db
+        tdSql.execute(f"use {db};")
+
+        # cols
+        sql1 = ""
+        for k, v in cols.items():
+            sql1 += f",{k} {v}"
+        # tags
+        sql2 = ""
+        for k, v in tags.items():
+            if sql2 == "":
+                sql2 =  f"{k} {v}"
+            else:
+                sql2 += f",{k} {v}"
+
+        # create stb
+        sql = f"create table {db}.{stb}(ts timestamp {sql1}) tags({sql2})"
+        tdSql.execute(sql)
+
     # run
     def run(self):
         # seed
         random.seed(int(time.time()))
         self.dbname = "schema_change"
 
-        # switch db
-        tdSql.execute(f"use {self.dbname};")
+        # create db
+        self.create(self.dbname, "meters", self.column_dict, self.tag_dict)
+
 
         # change meters
-        self.change_schema(1000000)
-
+        self.change_schema(1000)
         
         
     def stop(self):