Improve testing

canonical · May 16, 2024 · 227d5ac · 227d5ac
1 parent 04dfccb
commit 227d5ac
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/tests/integration/tests/test_control_plane_taints.py b/tests/integration/tests/test_control_plane_taints.py
@@ -27,10 +27,12 @@ def test_control_plane_taints(instances: List[harness.Instance]):
     )
 
     k8s_instance.exec(["k8s", "bootstrap", "--file", "/root/config.yaml"])
-    while not (nodes := util.get_nodes(k8s_instance)):
+    retries = 10
+    while retries and not (nodes := util.get_nodes(k8s_instance)):
         LOG.info("Waiting for Nodes")
-        time.sleep(5)
-
+        time.sleep(3)
+        retries -= 1
+    assert len(nodes) == 1, "Should have found one node in 30 sec"
     assert all([
         t["effect"] == "NoSchedule"
         for t in nodes[0]["spec"]["taints"]

diff --git a/tests/integration/tests/test_util/util.py b/tests/integration/tests/test_util/util.py
@@ -246,7 +246,7 @@ def get_nodes(control_node: harness.Instance) -> List[Any]:
     node_list = json.loads(result.stdout.decode())
     assert node_list["kind"] == "List", "Should have found a list of nodes"
     return [node for node in node_list["items"]]
-    
+
 
 def ready_nodes(control_node: harness.Instance) -> List[Any]:
     """Get a list of the ready nodes.