test.py 18.2 KB
Newer Older
P
plum-lihui 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
#!/usr/bin/python
###################################################################
#           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
#
###################################################################
# install pip
# pip install src/connector/python/

# -*- coding: utf-8 -*-
import sys
import getopt
import subprocess
import time
wafwerar's avatar
wafwerar 已提交
20 21
import base64
import json
wafwerar's avatar
wafwerar 已提交
22
import platform
wafwerar's avatar
wafwerar 已提交
23
import socket
wafwerar's avatar
wafwerar 已提交
24
import threading
C
cpwu 已提交
25 26

import toml
P
plum-lihui 已提交
27 28 29 30
sys.path.append("../pytest")
from util.log import *
from util.dnodes import *
from util.cases import *
haoranc's avatar
haoranc 已提交
31
from util.cluster import *
C
cpwu 已提交
32
from util.taosadapter import *
P
plum-lihui 已提交
33 34

import taos
C
cpwu 已提交
35
import taosrest
P
plum-lihui 已提交
36

wafwerar's avatar
wafwerar 已提交
37 38
def checkRunTimeError():
    import win32gui
wafwerar's avatar
wafwerar 已提交
39
    timeCount = 0
wafwerar's avatar
wafwerar 已提交
40 41
    while 1:
        time.sleep(1)
wafwerar's avatar
wafwerar 已提交
42
        timeCount = timeCount + 1
wafwerar's avatar
wafwerar 已提交
43
        print("checkRunTimeError",timeCount)
wafwerar's avatar
wafwerar 已提交
44
        if (timeCount>600):
wafwerar's avatar
wafwerar 已提交
45
            print("stop the test.")
wafwerar's avatar
wafwerar 已提交
46 47 48
            os.system("TASKKILL /F /IM taosd.exe")
            os.system("TASKKILL /F /IM taos.exe")
            os.system("TASKKILL /F /IM tmq_sim.exe")
wafwerar's avatar
wafwerar 已提交
49
            os.system("TASKKILL /F /IM mintty.exe")
wafwerar's avatar
wafwerar 已提交
50
            os.system("TASKKILL /F /IM python.exe")
wafwerar's avatar
wafwerar 已提交
51
            quit(0)
wafwerar's avatar
wafwerar 已提交
52 53 54
        hwnd = win32gui.FindWindow(None, "Microsoft Visual C++ Runtime Library")
        if hwnd:
            os.system("TASKKILL /F /IM taosd.exe")
P
plum-lihui 已提交
55 56

if __name__ == "__main__":
C
cpwu 已提交
57

P
plum-lihui 已提交
58 59 60 61 62
    fileName = "all"
    deployPath = ""
    masterIp = ""
    testCluster = False
    valgrind = 0
63
    killValgrind = 1
P
plum-lihui 已提交
64 65 66
    logSql = True
    stop = 0
    restart = False
67
    dnodeNums = 1
haoranc's avatar
haoranc 已提交
68
    mnodeNums = 0
wafwerar's avatar
wafwerar 已提交
69
    updateCfgDict = {}
C
cpwu 已提交
70
    adapter_cfg_dict = {}
wafwerar's avatar
wafwerar 已提交
71
    execCmd = ""
haoranc's avatar
haoranc 已提交
72
    queryPolicy = 1
haoranc's avatar
haoranc 已提交
73
    createDnodeNums = 1
C
cpwu 已提交
74
    restful = False
C
cpwu 已提交
75 76
    opts, args = getopt.gnu_getopt(sys.argv[1:], 'f:p:m:l:scghrd:k:e:N:M:Q:C:RD:', [
        'file=', 'path=', 'master', 'logSql', 'stop', 'cluster', 'valgrind', 'help', 'restart', 'updateCfgDict', 'killv', 'execCmd','dnodeNums','mnodeNums','queryPolicy','createDnodeNums','restful','adaptercfgupdate'])
P
plum-lihui 已提交
77 78 79 80 81 82 83 84 85 86 87 88
    for key, value in opts:
        if key in ['-h', '--help']:
            tdLog.printNoPrefix(
                'A collection of test cases written using Python')
            tdLog.printNoPrefix('-f Name of test case file written by Python')
            tdLog.printNoPrefix('-p Deploy Path for Simulator')
            tdLog.printNoPrefix('-m Master Ip for Simulator')
            tdLog.printNoPrefix('-l <True:False> logSql Flag')
            tdLog.printNoPrefix('-s stop All dnodes')
            tdLog.printNoPrefix('-c Test Cluster Flag')
            tdLog.printNoPrefix('-g valgrind Test Flag')
            tdLog.printNoPrefix('-r taosd restart test')
wafwerar's avatar
wafwerar 已提交
89
            tdLog.printNoPrefix('-d update cfg dict, base64 json str')
90
            tdLog.printNoPrefix('-k not kill valgrind processer')
wafwerar's avatar
wafwerar 已提交
91
            tdLog.printNoPrefix('-e eval str to run')
haoranc's avatar
haoranc 已提交
92
            tdLog.printNoPrefix('-N start dnodes numbers in clusters')
haoranc's avatar
haoranc 已提交
93
            tdLog.printNoPrefix('-M create mnode numbers in clusters')
haoranc's avatar
haoranc 已提交
94
            tdLog.printNoPrefix('-Q set queryPolicy in one dnode')
haoranc's avatar
haoranc 已提交
95
            tdLog.printNoPrefix('-C create Dnode Numbers in one cluster')
C
cpwu 已提交
96
            tdLog.printNoPrefix('-R restful realization form')
C
cpwu 已提交
97
            tdLog.printNoPrefix('-D taosadapter update cfg dict ')
C
cpwu 已提交
98

haoranc's avatar
haoranc 已提交
99

P
plum-lihui 已提交
100 101
            sys.exit(0)

C
cpwu 已提交
102
        if key in ['-r', '--restart']:
P
plum-lihui 已提交
103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131
            restart = True

        if key in ['-f', '--file']:
            fileName = value

        if key in ['-p', '--path']:
            deployPath = value

        if key in ['-m', '--master']:
            masterIp = value

        if key in ['-l', '--logSql']:
            if (value.upper() == "TRUE"):
                logSql = True
            elif (value.upper() == "FALSE"):
                logSql = False
            else:
                tdLog.printNoPrefix("logSql value %s is invalid" % logSql)
                sys.exit(0)

        if key in ['-c', '--cluster']:
            testCluster = True

        if key in ['-g', '--valgrind']:
            valgrind = 1

        if key in ['-s', '--stop']:
            stop = 1

wafwerar's avatar
wafwerar 已提交
132 133 134 135 136 137
        if key in ['-d', '--updateCfgDict']:
            try:
                updateCfgDict = eval(base64.b64decode(value.encode()).decode())
            except:
                print('updateCfgDict convert fail.')
                sys.exit(0)
wafwerar's avatar
wafwerar 已提交
138

139 140 141
        if key in ['-k', '--killValgrind']:
            killValgrind = 0

wafwerar's avatar
wafwerar 已提交
142 143 144 145
        if key in ['-e', '--execCmd']:
            try:
                execCmd = base64.b64decode(value.encode()).decode()
            except:
C
cpwu 已提交
146
                print('execCmd run fail.')
wafwerar's avatar
wafwerar 已提交
147 148
                sys.exit(0)

haoranc's avatar
haoranc 已提交
149 150 151
        if key in ['-N', '--dnodeNums']:
            dnodeNums = value

haoranc's avatar
haoranc 已提交
152 153 154
        if key in ['-M', '--mnodeNums']:
            mnodeNums = value

haoranc's avatar
haoranc 已提交
155 156 157
        if key in ['-Q', '--queryPolicy']:
            queryPolicy = value

haoranc's avatar
haoranc 已提交
158 159 160
        if key in ['-C', '--createDnodeNums']:
            createDnodeNums = value

C
cpwu 已提交
161 162 163
        if key in ['-R', '--restful']:
            restful = True

C
cpwu 已提交
164 165 166 167 168 169 170
        if key in ['-D', '--adaptercfgupdate']:
            try:
                adaptercfgupdate = eval(base64.b64decode(value.encode()).decode())
            except:
                print('adapter cfg update convert fail.')
                sys.exit(0)

wafwerar's avatar
wafwerar 已提交
171
    if not execCmd == "":
C
cpwu 已提交
172 173 174 175
        if restful:
            tAdapter.init(deployPath)
        else:
            tdDnodes.init(deployPath)
wafwerar's avatar
wafwerar 已提交
176
        print(execCmd)
wafwerar's avatar
wafwerar 已提交
177 178 179
        exec(execCmd)
        quit()

P
plum-lihui 已提交
180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207
    if (stop != 0):
        if (valgrind == 0):
            toBeKilled = "taosd"
        else:
            toBeKilled = "valgrind.bin"

        killCmd = "ps -ef|grep -w %s| grep -v grep | awk '{print $2}' | xargs kill -TERM > /dev/null 2>&1" % toBeKilled

        psCmd = "ps -ef|grep -w %s| grep -v grep | awk '{print $2}'" % toBeKilled
        processID = subprocess.check_output(psCmd, shell=True)

        while(processID):
            os.system(killCmd)
            time.sleep(1)
            processID = subprocess.check_output(psCmd, shell=True)

        for port in range(6030, 6041):
            usePortPID = "lsof -i tcp:%d | grep LISTEn | awk '{print $2}'" % port
            processID = subprocess.check_output(usePortPID, shell=True)

            if processID:
                killCmd = "kill -TERM %s" % processID
                os.system(killCmd)
            fuserCmd = "fuser -k -n tcp %d" % port
            os.system(fuserCmd)
        if valgrind:
            time.sleep(2)

C
cpwu 已提交
208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232
        if restful:
            toBeKilled = "taosadapter"

            killCmd = "ps -ef|grep -w %s| grep -v grep | awk '{print $2}' | xargs kill -TERM > /dev/null 2>&1" % toBeKilled

            psCmd = "ps -ef|grep -w %s| grep -v grep | awk '{print $2}'" % toBeKilled
            processID = subprocess.check_output(psCmd, shell=True)

            while(processID):
                os.system(killCmd)
                time.sleep(1)
                processID = subprocess.check_output(psCmd, shell=True)

            for port in range(6030, 6041):
                usePortPID = "lsof -i tcp:%d | grep LISTEn | awk '{print $2}'" % port
                processID = subprocess.check_output(usePortPID, shell=True)

                if processID:
                    killCmd = "kill -TERM %s" % processID
                    os.system(killCmd)
                fuserCmd = "fuser -k -n tcp %d" % port
                os.system(fuserCmd)

            tdLog.info('stop taosadapter')

P
plum-lihui 已提交
233
        tdLog.info('stop All dnodes')
C
cpwu 已提交
234

P
plum-lihui 已提交
235
    if masterIp == "":
wafwerar's avatar
wafwerar 已提交
236
        host = socket.gethostname()
P
plum-lihui 已提交
237
    else:
wafwerar's avatar
wafwerar 已提交
238 239 240 241 242
        try:
            config = eval(masterIp)
            host = config["host"]
        except Exception as r:
            host = masterIp
P
plum-lihui 已提交
243 244

    tdLog.info("Procedures for tdengine deployed in %s" % (host))
wafwerar's avatar
wafwerar 已提交
245
    if platform.system().lower() == 'windows':
wafwerar's avatar
wafwerar 已提交
246
        fileName = fileName.replace("/", os.sep)
wafwerar's avatar
wafwerar 已提交
247
        if (masterIp == "" and not fileName == "0-others\\udf_create.py"):
wafwerar's avatar
wafwerar 已提交
248
            threading.Thread(target=checkRunTimeError,daemon=True).start()
P
plum-lihui 已提交
249
        tdLog.info("Procedures for testing self-deployment")
wafwerar's avatar
wafwerar 已提交
250
        tdDnodes.init(deployPath, masterIp)
wafwerar's avatar
wafwerar 已提交
251 252 253 254 255 256
        tdDnodes.setTestCluster(testCluster)
        tdDnodes.setValgrind(valgrind)
        tdDnodes.stopAll()
        key_word = 'tdCases.addWindows'
        is_test_framework = 0
        try:
wafwerar's avatar
wafwerar 已提交
257
            if key_word in open(fileName, encoding='UTF-8').read():
wafwerar's avatar
wafwerar 已提交
258
                is_test_framework = 1
wafwerar's avatar
wafwerar 已提交
259 260
        except Exception as r:
            print(r)
wafwerar's avatar
wafwerar 已提交
261
        updateCfgDictStr = ''
C
cpwu 已提交
262
        # adapter_cfg_dict_str = ''
wafwerar's avatar
wafwerar 已提交
263 264 265 266 267
        if is_test_framework:
            moduleName = fileName.replace(".py", "").replace(os.sep, ".")
            uModule = importlib.import_module(moduleName)
            try:
                ucase = uModule.TDTestCase()
wafwerar's avatar
wafwerar 已提交
268
                if ((json.dumps(updateCfgDict) == '{}') and hasattr(ucase, 'updatecfgDict')):
wafwerar's avatar
wafwerar 已提交
269 270
                    updateCfgDict = ucase.updatecfgDict
                    updateCfgDictStr = "-d %s"%base64.b64encode(json.dumps(updateCfgDict).encode()).decode()
C
cpwu 已提交
271 272 273
                if ((json.dumps(adapter_cfg_dict) == '{}') and hasattr(ucase, 'taosadapter_cfg_dict')):
                    adapter_cfg_dict = ucase.taosadapter_cfg_dict
                    # adapter_cfg_dict_str = f"-D {base64.b64encode(toml.dumps(adapter_cfg_dict).encode()).decode()}"
wafwerar's avatar
wafwerar 已提交
274 275
            except Exception as r:
                print(r)
wafwerar's avatar
wafwerar 已提交
276 277
        else:
            pass
C
cpwu 已提交
278 279 280 281
        if restful:
            tAdapter.init(deployPath, masterIp)
            tAdapter.stop(force_kill=True)

wafwerar's avatar
wafwerar 已提交
282 283 284 285
        if dnodeNums == 1 :
            tdDnodes.deploy(1,updateCfgDict)
            tdDnodes.start(1)
            tdCases.logSql(logSql)
C
cpwu 已提交
286 287 288 289
            if restful:
                tAdapter.deploy(adapter_cfg_dict)
                tAdapter.start()

wafwerar's avatar
wafwerar 已提交
290 291
            if queryPolicy != 1:
                queryPolicy=int(queryPolicy)
C
cpwu 已提交
292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308
                if restful:
                    conn = taosrest.connect(url=f"http://{host}:6041")
                else:
                    conn = taos.connect(host,config=tdDnodes.getSimCfgPath())

                cursor = conn.cursor()
                cursor.execute("create qnode on dnode 1")
                cursor.execute(f'alter local "queryPolicy" "{queryPolicy}"')
                cursor.execute("show local variables")
                res = cursor.fetchall()
                for i in range(cursor.rowcount):
                    if res[i][0] == "queryPolicy" :
                        if int(res[i][1]) == int(queryPolicy):
                            tdLog.success(f'alter queryPolicy to {queryPolicy} successfully')
                        else:
                            tdLog.debug(res)
                            tdLog.exit(f"alter queryPolicy to  {queryPolicy} failed")
wafwerar's avatar
wafwerar 已提交
309 310 311 312 313 314 315 316 317 318 319 320 321
        else :
            tdLog.debug("create an cluster  with %s nodes and make %s dnode as independent mnode"%(dnodeNums,mnodeNums))
            dnodeslist = cluster.configure_cluster(dnodeNums=dnodeNums,mnodeNums=mnodeNums)
            tdDnodes = ClusterDnodes(dnodeslist)
            tdDnodes.init(deployPath, masterIp)
            tdDnodes.setTestCluster(testCluster)
            tdDnodes.setValgrind(valgrind)
            tdDnodes.stopAll()
            for dnode in tdDnodes.dnodes:
                tdDnodes.deploy(dnode.index,{})
            for dnode in tdDnodes.dnodes:
                tdDnodes.starttaosd(dnode.index)
            tdCases.logSql(logSql)
C
cpwu 已提交
322 323 324 325 326

            if restful:
                tAdapter.deploy(adapter_cfg_dict)
                tAdapter.start()

C
cpwu 已提交
327
            if not restful:
C
cpwu 已提交
328
                conn = taos.connect(host,config=tdDnodes.getSimCfgPath())
C
cpwu 已提交
329 330
            else:
                conn = taosrest.connect(url=f"http://{host}:6041")
C
cpwu 已提交
331
            tdLog.info(tdDnodes.getSimCfgPath(),host)
haoranc's avatar
haoranc 已提交
332 333 334 335 336
            if createDnodeNums == 1:
                createDnodeNums=dnodeNums
            else:
                createDnodeNums=createDnodeNums
            cluster.create_dnode(conn,createDnodeNums)
wafwerar's avatar
wafwerar 已提交
337 338 339 340 341 342 343 344
            try:
                if cluster.check_dnode(conn) :
                    print("check dnode ready")
            except Exception as r:
                print(r)
        if ucase is not None and hasattr(ucase, 'noConn') and ucase.noConn == True:
            conn = None
        else:
C
cpwu 已提交
345
            if not restful:
C
cpwu 已提交
346
                conn = taos.connect(host="%s"%(host), config=tdDnodes.sim.getCfgDir())
C
cpwu 已提交
347 348
            else:
                conn = taosrest.connect(url=f"http://{host}:6041")
wafwerar's avatar
wafwerar 已提交
349 350 351 352
        if is_test_framework:
            tdCases.runOneWindows(conn, fileName)
        else:
            tdCases.runAllWindows(conn)
P
plum-lihui 已提交
353
    else:
354
        tdDnodes.setKillValgrind(killValgrind)
wafwerar's avatar
wafwerar 已提交
355
        tdDnodes.init(deployPath, masterIp)
P
plum-lihui 已提交
356 357 358 359 360 361 362 363 364 365 366 367 368 369 370
        tdDnodes.setTestCluster(testCluster)
        tdDnodes.setValgrind(valgrind)
        tdDnodes.stopAll()
        is_test_framework = 0
        key_word = 'tdCases.addLinux'
        try:
            if key_word in open(fileName).read():
                is_test_framework = 1
        except:
            pass
        if is_test_framework:
            moduleName = fileName.replace(".py", "").replace("/", ".")
            uModule = importlib.import_module(moduleName)
            try:
                ucase = uModule.TDTestCase()
wafwerar's avatar
wafwerar 已提交
371 372
                if (json.dumps(updateCfgDict) == '{}'):
                    updateCfgDict = ucase.updatecfgDict
C
cpwu 已提交
373 374
                if (json.dumps(adapter_cfg_dict) == '{}'):
                    adapter_cfg_dict = ucase.taosadapter_cfg_dict
wafwerar's avatar
wafwerar 已提交
375 376
            except:
                pass
C
cpwu 已提交
377 378 379 380 381

        if restful:
            tAdapter.init(deployPath, masterIp)
            tAdapter.stop(force_kill=True)

haoranc's avatar
haoranc 已提交
382 383 384 385
        if dnodeNums == 1 :
            tdDnodes.deploy(1,updateCfgDict)
            tdDnodes.start(1)
            tdCases.logSql(logSql)
C
cpwu 已提交
386 387 388 389 390

            if restful:
                tAdapter.deploy(adapter_cfg_dict)
                tAdapter.start()

haoranc's avatar
haoranc 已提交
391 392
            if queryPolicy != 1:
                queryPolicy=int(queryPolicy)
C
cpwu 已提交
393
                if not restful:
C
cpwu 已提交
394
                    conn = taos.connect(host,config=tdDnodes.getSimCfgPath())
C
cpwu 已提交
395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421
                else:
                    conn = taosrest.connect(url=f"http://{host}:6041")
                # tdSql.init(conn.cursor())
                # tdSql.execute("create qnode on dnode 1")
                # tdSql.execute('alter local "queryPolicy" "%d"'%queryPolicy)
                # tdSql.query("show local variables;")
                # for i in range(tdSql.queryRows):
                #     if tdSql.queryResult[i][0] == "queryPolicy" :
                #         if int(tdSql.queryResult[i][1]) == int(queryPolicy):
                #             tdLog.success('alter queryPolicy to %d successfully'%queryPolicy)
                #         else :
                #             tdLog.debug(tdSql.queryResult)
                #             tdLog.exit("alter queryPolicy to  %d failed"%queryPolicy)

                cursor = conn.cursor()
                cursor.execute("create qnode on dnode 1")
                cursor.execute(f'alter local "queryPolicy" "{queryPolicy}"')
                cursor.execute("show local variables")
                res = cursor.fetchall()
                for i in range(cursor.rowcount):
                    if res[i][0] == "queryPolicy" :
                        if int(res[i][1]) == int(queryPolicy):
                            tdLog.success(f'alter queryPolicy to {queryPolicy} successfully')
                        else:
                            tdLog.debug(res)
                            tdLog.exit(f"alter queryPolicy to  {queryPolicy} failed")

haoranc's avatar
haoranc 已提交
422
        else :
haoranc's avatar
haoranc 已提交
423 424
            tdLog.debug("create an cluster  with %s nodes and make %s dnode as independent mnode"%(dnodeNums,mnodeNums))
            dnodeslist = cluster.configure_cluster(dnodeNums=dnodeNums,mnodeNums=mnodeNums)
haoranc's avatar
haoranc 已提交
425 426 427 428 429 430 431 432 433 434
            tdDnodes = ClusterDnodes(dnodeslist)
            tdDnodes.init(deployPath, masterIp)
            tdDnodes.setTestCluster(testCluster)
            tdDnodes.setValgrind(valgrind)
            tdDnodes.stopAll()
            for dnode in tdDnodes.dnodes:
                tdDnodes.deploy(dnode.index,{})
            for dnode in tdDnodes.dnodes:
                tdDnodes.starttaosd(dnode.index)
            tdCases.logSql(logSql)
C
cpwu 已提交
435 436 437 438 439

            if restful:
                tAdapter.deploy(adapter_cfg_dict)
                tAdapter.start()

C
cpwu 已提交
440
            if not restful:
C
cpwu 已提交
441
                conn = taos.connect(host,config=tdDnodes.getSimCfgPath())
C
cpwu 已提交
442 443
            else:
                conn = taosrest.connect(url=f"http://{host}:6041")
haoranc's avatar
haoranc 已提交
444
            print(tdDnodes.getSimCfgPath(),host)
haoranc's avatar
haoranc 已提交
445 446 447 448 449
            if createDnodeNums == 1:
                createDnodeNums=dnodeNums
            else:
                createDnodeNums=createDnodeNums
            cluster.create_dnode(conn,createDnodeNums)
haoranc's avatar
haoranc 已提交
450 451 452 453 454
            try:
                if cluster.check_dnode(conn) :
                    print("check dnode ready")
            except Exception as r:
                print(r)
C
cpwu 已提交
455 456


P
plum-lihui 已提交
457 458 459 460 461 462 463 464
        if testCluster:
            tdLog.info("Procedures for testing cluster")
            if fileName == "all":
                tdCases.runAllCluster()
            else:
                tdCases.runOneCluster(fileName)
        else:
            tdLog.info("Procedures for testing self-deployment")
C
cpwu 已提交
465
            if not restful:
C
cpwu 已提交
466
                conn = taos.connect(host,config=tdDnodes.getSimCfgPath())
C
cpwu 已提交
467 468
            else:
                conn = taosrest.connect(url=f"http://{host}:6041")
C
cpwu 已提交
469

P
plum-lihui 已提交
470 471 472 473
            if fileName == "all":
                tdCases.runAllLinux(conn)
            else:
                tdCases.runOneLinux(conn, fileName)
C
cpwu 已提交
474

P
plum-lihui 已提交
475 476 477
        if restart:
            if fileName == "all":
                tdLog.info("not need to query ")
C
cpwu 已提交
478
            else:
P
plum-lihui 已提交
479 480 481 482
                sp = fileName.rsplit(".", 1)
                if len(sp) == 2 and sp[1] == "py":
                    tdDnodes.stopAll()
                    tdDnodes.start(1)
C
cpwu 已提交
483
                    time.sleep(1)
C
cpwu 已提交
484 485 486 487
                    if not restful:
                        conn = taos.connect( host, config=tdDnodes.getSimCfgPath())
                    else:
                        conn = taosrest.connect(url=f"http://{host}:6041")
P
plum-lihui 已提交
488 489 490 491 492
                    tdLog.info("Procedures for tdengine deployed in %s" % (host))
                    tdLog.info("query test after taosd restart")
                    tdCases.runOneLinux(conn, sp[0] + "_" + "restart.py")
                else:
                    tdLog.info("not need to query")
C
cpwu 已提交
493

wafwerar's avatar
wafwerar 已提交
494 495
    if conn is not None:
        conn.close()
T
tangfangzhi 已提交
496
    sys.exit(0)