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

test: adjust test case for alter table

This commit is contained in:
kailixu 2023-11-25 00:49:49 +08:00
parent 0b27cb487b
commit 9ceddd3f60
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
tests/system-test/1-insert/alter_stable.py
View File

@ -209,7 +209,16 @@ class TDTestCase:
def alterTableTask(self, i): def alterTableTask(self, i):
os.system(f'taos -f {self.fname}.{i};') os.system(f'taos -f {self.fname}.{i};')
def alterTableCheck(self, opt): def executeAlterTable(self, opt):
threads = []
for i in range(self.threadnum):
thread = threading.Thread(target=self.alterTableTask, args=(i,))
threads.append(thread)
thread.start()
for i in range(self.threadnum):
threads[i].join()
def checkAlterTable(self, opt):
if opt in ["stb_add_col", "stb_add_tag"]: if opt in ["stb_add_col", "stb_add_tag"]:
for i in range(self.stbnum): for i in range(self.stbnum):
tdSql.execute(f'desc {self.stbname}_{i}') tdSql.execute(f'desc {self.stbname}_{i}')
@ -220,23 +229,13 @@ class TDTestCase:
for i in range(self.ntbnum): for i in range(self.ntbnum):
tdSql.execute(f'desc {self.ntbname}_{i}') tdSql.execute(f'desc {self.ntbname}_{i}')
def executeAlterTableAndCheck(self, opt): def destroyAlterTable(self):
threads = []
for i in range(self.threadnum):
thread = threading.Thread(target=self.alterTableTask, args=(i,))
threads.append(thread)
thread.start()
for i in range(self.threadnum):
threads[i].join()
self.alterTableCheck(opt)
def destroyAlterTableEnv(self):
for i in range(self.threadnum): for i in range(self.threadnum):
if os.path.isfile(f'{self.fname}.{i}'): if os.path.isfile(f'{self.fname}.{i}'):
os.remove(f'{self.fname}.{i}') os.remove(f'{self.fname}.{i}')
def prepareAlterTableEnv(self, opt): def prepareAlterTable(self, opt):
self.destroyAlterTableEnv() self.destroyAlterTable()
lines = [f'use {self.dbname};\n'] lines = [f'use {self.dbname};\n']
if opt in ["stb_add_col", "stb_add_tag"]: if opt in ["stb_add_col", "stb_add_tag"]:
for i in range(self.stbnum): for i in range(self.stbnum):
@ -297,9 +296,10 @@ class TDTestCase:
for opt in alter_table_check_type: for opt in alter_table_check_type:
self.prepareAlterEnv() self.prepareAlterEnv()
self.prepareAlterTableEnv(opt) self.prepareAlterTable(opt)
self.executeAlterTableAndCheck(opt) self.executeAlterTable(opt)
self.destroyAlterTableEnv() self.checkAlterTable(opt)
self.destroyAlterTable()
self.destroyAlterEnv() self.destroyAlterEnv()
def run(self): def run(self):