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

test: add kill processer function

This commit is contained in:
plum-lihui 2023-07-05 18:32:03 +08:00
parent 91dea69534
commit 944b0d26ef
2 changed files with 34 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/system-test/7-tmq/tmqCommon.py
View File

@ -577,6 +577,20 @@ class TMQCom:
tdLog.info(tsql.queryResult) tdLog.info(tsql.queryResult)
tdLog.info("wait subscriptions exit for %d s"%wait_cnt) tdLog.info("wait subscriptions exit for %d s"%wait_cnt)
def killProcesser(self, processerName):
killCmd = (
"ps -ef|grep -w %s| grep -v grep | awk '{print $2}' | xargs kill -TERM > /dev/null 2>&1"
% processerName
)
psCmd = "ps -ef|grep -w %s| grep -v grep | awk '{print $2}'" % processerName
processID = subprocess.check_output(psCmd, shell=True)
while processID:
os.system(killCmd)
time.sleep(1)
processID = subprocess.check_output(psCmd, shell=True)
def close(self): def close(self):
self.cursor.close() self.cursor.close()

40
tests/system-test/7-tmq/tmqDropConsumer.py
View File

@ -235,28 +235,28 @@ class TDTestCase:
if (0 == existFlag): if (0 == existFlag):
groupIdList.append(groupId) groupIdList.append(groupId)
# # kill taosBenchmark # kill taosBenchmark
# os.system("pkill -9 taosBenchmark") tmqCom.killProcesser("taosBenchmark")
# # wait the status to "lost" # wait the status to "lost"
# while (1): while (1):
# exitFlag = 1 exitFlag = 1
# tdSql.query('show consumers;') tdSql.query('show consumers;')
# consumerNUm = tdSql.queryRows consumerNUm = tdSql.queryRows
# for i in range(consumerNUm): for i in range(consumerNUm):
# status = tdSql.getData(i,3) status = tdSql.getData(i,3)
# if (status != "lost"): if (status != "lost"):
# exitFlag = 0 exitFlag = 0
# time.sleep(2) time.sleep(2)
# break break
# if (1 == exitFlag): if (1 == exitFlag):
# break break
# # drop consumer groups # drop consumer groups
# for i in range(len(groupIdList)): for i in range(len(groupIdList)):
# for j in range(len(topicNameList)): for j in range(len(topicNameList)):
# sqlCmd = f"drop consumer group `%s` on %s"%(groupIdList[i], topicNameList[j]) sqlCmd = f"drop consumer group `%s` on %s"%(groupIdList[i], topicNameList[j])
# tdSql.execute(sqlCmd) tdSql.execute(sqlCmd)
tmqCom.g_end_insert_flag = 1 tmqCom.g_end_insert_flag = 1
tdLog.debug("notify sub-thread to stop insert data") tdLog.debug("notify sub-thread to stop insert data")