Merge pull request redpanda-data#14181 from mmaslankaprv/fix-14047

tests: retry on timeout in partition balancer test
vbotbuildovich · Oct 17, 2023 · cb1a8ba · cb1a8ba
2 parents 7f2cb35 + bdab660
commit cb1a8ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/rptest/tests/partition_balancer_test.py b/tests/rptest/tests/partition_balancer_test.py
@@ -777,7 +777,7 @@ def test_maintenance_mode(self, kill_same_node):
         node_id = self.redpanda.idx(node)
 
         rpk = RpkTool(self.redpanda)
-        admin = Admin(self.redpanda)
+        admin = Admin(self.redpanda, retry_codes=[503, 504])
 
         rpk.cluster_maintenance_enable(node, wait=True)
         # the node should now report itself in maintenance mode