xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

test:add concurrency test cases for executing the taos -k command line.

This commit is contained in:
chenhaoran 2024-08-05 11:01:32 +08:00
parent dd5b547404
commit 1a871e0292
1 changed files with 7 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tests/system-test/0-others/taosShell.py
View File

@ -111,7 +111,7 @@ class TDTestCase:
updatecfgDict["fqdn"] = hostname updatecfgDict["fqdn"] = hostname
print ("===================: ", updatecfgDict) print ("===================: ", updatecfgDict)
taos_output = []
def init(self, conn, logSql, replicaVar=1): def init(self, conn, logSql, replicaVar=1):
self.replicaVar = int(replicaVar) self.replicaVar = int(replicaVar)
tdLog.debug(f"start to excute {__file__}") tdLog.debug(f"start to excute {__file__}")
@ -119,7 +119,6 @@ class TDTestCase:
def getBuildPath(self): def getBuildPath(self):
selfPath = os.path.dirname(os.path.realpath(__file__)) selfPath = os.path.dirname(os.path.realpath(__file__))
if ("community" in selfPath): if ("community" in selfPath):
projPath = selfPath[:selfPath.find("community")] projPath = selfPath[:selfPath.find("community")]
else: else:
@ -133,21 +132,17 @@ class TDTestCase:
break break
return buildPath return buildPath
def run_command(self, commands): def run_command(self, commands):
self.taos_output = []
count = 0 count = 0
while count < 2: while count < 2:
print(f"count: {count}") # print(f"count: {count}")
value = subprocess.getoutput(f"nohup {commands} &") value = subprocess.getoutput(f"nohup {commands} &")
print(f"value: {value}") # print(f"value: {value}")
self.taos_output.append(value) self.taos_output.append(value)
count += 1 count += 1
return self.taos_output
def taos_thread_repeat_k(self, run_command, commands, threads_num=10, output=[]): def taos_thread_repeat_k(self, run_command, commands, threads_num=10, output=[]):
threads = [] threads = []
taos_output = self.taos_output taos_output = self.taos_output
threads_num = 20
for id in range(threads_num): for id in range(threads_num):
#threads.append(Process(target=cloud_consumer, args=(id,))) #threads.append(Process(target=cloud_consumer, args=(id,)))
threads.append(threading.Thread(target=run_command, args=(commands,))) threads.append(threading.Thread(target=run_command, args=(commands,)))
@ -155,11 +150,9 @@ class TDTestCase:
tr.start() tr.start()
for tr in threads: for tr in threads:
tr.join() tr.join()
for value in taos_output: for value in taos_output:
if "crash" in value: if "crash" in value:
print(f"command: {commands} crash") tdLog.exit(f"command: {commands} crash")
exit(1)
def run(self): # sourcery skip: extract-duplicate-method, remove-redundant-fstring def run(self): # sourcery skip: extract-duplicate-method, remove-redundant-fstring
tdSql.prepare() tdSql.prepare()
@ -479,7 +472,7 @@ class TDTestCase:
commands = f"taos -k -c {cfgPath}" commands = f"taos -k -c {cfgPath}"
output = self.run_command(commands) output = self.run_command(commands)
os.sys os.sys
self.taos_thread_repeat_k(self.run_command, commands, 10, output) self.taos_thread_repeat_k(self.run_command, commands, 100, output)
# os.system("python 0-others/repeat_taos_k.py") # os.system("python 0-others/repeat_taos_k.py")
def stop(self): def stop(self):