From 1eb792a8f8e732a12cbd2d270e1ed2a2c78210dc Mon Sep 17 00:00:00 2001
From: tomchon <haoran920c@163.com>
Date: Tue, 8 Mar 2022 12:27:28 +0800
Subject: [PATCH] [TD-13526]<test>: add  checkClusterStatus.py

---
 .../manualTest/TD-5114/checkClusterStatus.py  | 111 ++++++++++++++++++
 1 file changed, 111 insertions(+)
 create mode 100644 tests/pytest/manualTest/TD-5114/checkClusterStatus.py

diff --git a/tests/pytest/manualTest/TD-5114/checkClusterStatus.py b/tests/pytest/manualTest/TD-5114/checkClusterStatus.py
new file mode 100644
index 0000000000..7acb03013b
--- /dev/null
+++ b/tests/pytest/manualTest/TD-5114/checkClusterStatus.py
@@ -0,0 +1,111 @@
+###################################################################
+#           Copyright (c) 2016 by TAOS Technologies, Inc.
+#                     All rights reserved.
+#
+#  This file is proprietary and confidential to TAOS Technologies.
+#  No part of this file may be reproduced, stored, transmitted,
+#  disclosed or used in any form or by any means other than as
+#  expressly provided by the written permission from Jianhui Tao
+#
+###################################################################
+
+# -*- coding: utf-8 -*-
+
+import os
+import sys
+sys.path.insert(0, os.getcwd())
+from util.log import *
+from util.sql import *
+from util.dnodes import *
+import taos
+import threading
+import subprocess
+import datetime
+from random import choice
+ 
+class TwoClients:
+    def initConnection(self):
+        self.host = sys.argv[1]
+        self.user = "root"
+        self.password = "taosdata"
+        self.config = "/etc/taos/"     
+        self.port =6030 
+        self.rowNum = 10
+        self.ts = 1537146000000  
+
+    def run(self):
+        
+        # new taos client
+        conn1 = taos.connect(host=self.host, user=self.user, password=self.password, config=self.config )
+        cur1 = conn1.cursor()
+        print(cur1)
+        tdSql.init(cur1, True)
+
+        # insert data with  python connector , if you want to use this case ,cancel note.
+        
+        # check cluster status
+        tdSql.query("show dnodes") 
+        print(tdSql.queryRows)
+        for i in range(tdSql.queryRows):
+            for j in range(1,tdSql.queryRows+1):
+                if (tdSql.queryResult[i][1] == "compat_container%d:6030" %j):
+                    tdSql.checkData(i,4,"ready")
+
+        tdSql.query("show mnodes") 
+        tdSql.checkRows(3)
+        roles = "master slave"
+        for i in range(tdSql.queryRows):
+            if (tdSql.queryResult[i][2] in roles ):
+                ep = tdSql.queryResult[i][1]
+                role = tdSql.queryResult[i][2]
+                print(" the role of %s is %s " %(ep,role))
+            else:
+                print("cluster is not ready")
+
+        version = sys.argv[2]
+        tdSql.query("show variables") 
+        tdSql.checkData(6,1,"%s" % version) 
+
+
+
+        # for x in range(10):
+        dataType= [ "tinyint", "smallint",  "int", "bigint", "float", "double", "bool", " binary(20)", "nchar(20)", "tinyint unsigned", "smallint unsigned", "int unsigned",  "bigint unsigned"] 
+        tdSql.execute("drop database if exists db1") 
+        tdSql.execute("create database db1 keep 3650  replica 2 ")
+        tdSql.execute("use db1")
+        tdSql.execute('''create table test(ts timestamp, col0 tinyint, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, 
+                    col7 bool, col8 binary(20), col9 nchar(20), col10 tinyint unsigned, col11 smallint unsigned, col12 int unsigned, col13 bigint unsigned) tags(loc nchar(3000), tag1 int)''')
+        print(datetime.datetime.now())
+        rowNum1= 20
+        for i in range(rowNum1):
+            tdSql.execute("alter table test add column col%d  %s ;" %( i+14, choice(dataType))     ) 
+        rowNum2= 20
+        for i in range(rowNum2):
+            tdSql.execute("alter table test drop column col%d   ;" %( i+14)     )    
+        self.rowNum3 = 50
+        self.rowNum4 = 100
+        self.ts = 1537146000000
+        for j in range(self.rowNum4):
+            tdSql.execute("create table test%d using test tags('beijing%d', 10)" % (j,j) )
+            for i in range(self.rowNum3):
+                tdSql.execute("insert into test%d values(%d, %d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" 
+                        % (j, self.ts + i*1000, i + 1, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1))
+        print(datetime.datetime.now())
+        # check data correct 
+        tdSql.execute("show databases")
+        tdSql.execute("use db1")
+        tdSql.query("select count (tbname) from test")
+        tdSql.checkData(0, 0, 100) 
+        tdSql.query("select count (*) from test")
+        tdSql.checkData(0, 0, 5000) 
+
+
+        # delete useless file
+        testcaseFilename = os.path.split(__file__)[-1]
+        os.system("rm -rf ./insert_res.txt")
+        os.system("rm -rf manualTest/TD-5114/%s.sql" % testcaseFilename )       
+   
+clients = TwoClients()
+clients.initConnection()
+# clients.getBuildPath()
+clients.run()
\ No newline at end of file
-- 
GitLab