From 274e40b559e3d9b340c80e66be3fe6972961d4a2 Mon Sep 17 00:00:00 2001 From: kailixu Date: Thu, 21 Nov 2024 16:16:48 +0800 Subject: [PATCH] test: improve the robustness of alterConfig.py --- tests/army/alter/alterConfig.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/tests/army/alter/alterConfig.py b/tests/army/alter/alterConfig.py index ef8de0011b..6a22dd014f 100644 --- a/tests/army/alter/alterConfig.py +++ b/tests/army/alter/alterConfig.py @@ -113,6 +113,15 @@ class TDTestCase(TBase): if not result: raise Exception(f"key:{key} not found") + def checkRows(self, sql, nExpect, nRetry): + for i in range(nRetry): + res = tdSql.getResult(sql) + if len(res) == nExpect: + break + time.sleep(1) + if len(res) != nExpect: + raise Exception(f"rows:{len(res)} != {nExpect}") + def alterBypassFlag(self): """Add test case for altering bypassFlag(TD-32907) """ @@ -151,8 +160,7 @@ class TDTestCase(TBase): tdSql.query("select * from stb0") tdSql.checkRows(2) tdSql.execute("flush database db") - tdSql.query("select * from stb0") - tdSql.checkRows(0) + self.checkRows("select * from stb0", 0, 10) tdSql.execute("alter all dnodes 'bypassFlag 0'") self.checkKeyValue(tdSql.getResult("show local variables"), "bypassFlag", "0") self.checkKeyValue(tdSql.getResult("show dnode 1 variables like 'bypassFlag'"), "bypassFlag", "0", 1, 2) @@ -161,8 +169,9 @@ class TDTestCase(TBase): tdSql.query("select * from stb0") tdSql.checkRows(2) tdSql.execute("flush database db") - tdSql.query("select * from stb0") - tdSql.checkRows(2) + for i in range(5): + self.checkRows("select * from stb0", 2, 1) + time.sleep(1) # run def run(self):