homework-jianmu/tests/system-test/6-cluster/5dnode2mnode.py

from ssl import ALERT_DESCRIPTION_CERTIFICATE_UNOBTAINABLE
import taos
import sys
import time
import os 

from util.log import *
from util.sql import *
from util.cases import *
from util.dnodes import TDDnodes
from util.dnodes import TDDnode
from util.cluster import *

import time
import socket
import subprocess
sys.path.append("./6-cluster")

from clusterCommonCreate import *
from clusterCommonCheck import * 
 
class TDTestCase:
    def init(self,conn ,logSql):
        tdLog.debug(f"start to excute {__file__}")
        tdSql.init(conn.cursor())
        self.host = socket.gethostname()


    def getBuildPath(self):
        selfPath = os.path.dirname(os.path.realpath(__file__))

        if ("community" in selfPath):
            projPath = selfPath[:selfPath.find("community")]
        else:
            projPath = selfPath[:selfPath.find("tests")]

        for root, dirs, files in os.walk(projPath):
            if ("taosd" in files):
                rootRealPath = os.path.dirname(os.path.realpath(root))
                if ("packaging" not in rootRealPath):
                    buildPath = root[:len(root) - len("/build/bin")]
                    break
        return buildPath

    def five_dnode_two_mnode(self):
        tdSql.query("show dnodes;")
        tdSql.checkData(0,1,'%s:6030'%self.host)
        tdSql.checkData(4,1,'%s:6430'%self.host)
        tdSql.checkData(0,4,'ready')
        tdSql.checkData(4,4,'ready')
        tdSql.query("show mnodes;")       
        tdSql.checkData(0,1,'%s:6030'%self.host)
        tdSql.checkData(0,2,'leader')
        tdSql.checkData(0,3,'ready')


        # fisr add two mnodes;
        tdSql.error("create mnode on dnode 1;")
        tdSql.error("drop mnode on dnode 1;")

        tdSql.execute("create mnode on dnode 2")
        count=0
        while count < 10:
            time.sleep(1)
            tdSql.query("show mnodes;")
            tdSql.checkRows(2) 
            if  tdSql.queryResult[0][2]=='leader' :
                if  tdSql.queryResult[1][2]=='follower':
                    print("two mnodes is ready")
                    break
            count+=1
        else:
            print("two mnodes is not ready in 10s ")

        # fisrt check statut ready
                
        tdSql.checkData(0,1,'%s:6030'%self.host)
        tdSql.checkData(0,2,'leader')
        tdSql.checkData(0,3,'ready')
        tdSql.checkData(1,1,'%s:6130'%self.host)
        tdSql.checkData(1,2,'follower')
        tdSql.checkData(1,3,'ready')

        # fisrt add data : db\stable\childtable\general table

        tdSql.execute("drop database if exists db2")
        tdSql.execute("create database if not exists db2 replica 1 duration 300")
        tdSql.execute("use db2")
        tdSql.execute(
        '''create table stb1
        (ts timestamp, c1 int, c2 bigint, c3 smallint, c4 tinyint, c5 float, c6 double, c7 bool, c8 binary(16),c9 nchar(32), c10 timestamp)
        tags (t1 int)
        '''
        )
        tdSql.execute(
            '''
            create table t1
            (ts timestamp, c1 int, c2 bigint, c3 smallint, c4 tinyint, c5 float, c6 double, c7 bool, c8 binary(16),c9 nchar(32), c10 timestamp)
            '''
        )
        for i in range(4):
            tdSql.execute(f'create table ct{i+1} using stb1 tags ( {i+1} )')
        tdSql.error("create mnode on dnode 2")
        tdSql.query("show dnodes;")
        print(tdSql.queryResult)
        clusterComCheck.checkDnodes(5)
        # restart all taosd
        tdDnodes=cluster.dnodes
       
        # stop follower
        tdLog.info("stop follower")
        tdDnodes[1].stoptaosd()
        if cluster.checkConnectStatus(0) :
            print("cluster also work")

        # start follower
        tdLog.info("start follower")
        tdDnodes[1].starttaosd()
        if clusterComCheck.checkMnodeStatus(2) :
            print("both mnodes are ready")
        
        # stop leader
        tdLog.info("stop leader")
        tdDnodes[0].stoptaosd()
        try:
            cluster.checkConnectStatus(2)
            tdLog.notice(" The election  still  succeeds  when leader of both mnodes are killed ")
        except Exception:
            pass
        tdSql.error("create user user1 pass '123';")
        tdLog.info("start leader")
        tdDnodes[0].starttaosd()
        if clusterComCheck.checkMnodeStatus(2) :
            print("both mnodes are ready")

        # # fisrt drop follower of mnode
        # BUG 
        # tdSql.execute("drop mnode on dnode 2")
        # tdSql.query("show mnodes;")     
        # tdSql.checkRows(1)  
        # tdSql.checkData(0,1,'%s:6030'%self.host)
        # tdSql.checkData(0,2,'leader')
        # tdSql.checkData(0,3,'ready')
        
        # tdSql.execute("create mnode on dnode 2")
        # count=0
        # while count < 10:
        #     time.sleep(1)
        #     tdSql.query("show mnodes;")
        #     tdSql.checkRows(2) 
        #     if  tdSql.queryResult[0][2]=='leader' :
        #         if  tdSql.queryResult[1][2]=='follower':
        #             print("two mnodes is ready")
        #             break
        #     count+=1
        # else:
        #     print("two mnodes is not ready in 10s ")
        
        # tdSql.query("show mnodes;")     
        # tdSql.checkRows(2)         
        # tdSql.checkData(0,1,'%s:6030'%self.host)
        # tdSql.checkData(0,2,'leader')
        # tdSql.checkData(0,3,'ready')
        # tdSql.checkData(1,1,'%s:6130'%self.host)
        # tdSql.checkData(1,2,'follower')
        # tdSql.checkData(2,3,'ready')


    def getConnection(self, dnode):
        host = dnode.cfgDict["fqdn"]
        port = dnode.cfgDict["serverPort"]
        config_dir = dnode.cfgDir
        return taos.connect(host=host, port=int(port), config=config_dir)


    def run(self): 
        self.five_dnode_two_mnode()


    def stop(self):
        tdSql.close()
        tdLog.success(f"{__file__} successfully executed")

tdCases.addLinux(__file__, TDTestCase())
tdCases.addWindows(__file__, TDTestCase())