Add timeouts to wait functions

Author: lucian-tosa

docker/mongodb-kubernetes-tests/tests/authentication/mongodb_custom_roles.py

@@ -177,9 +177,9 @@ def test_changing_role(
     mongodb_role["spec"]["roles"][0]["role"] = "readWrite"
     mongodb_role.update()
 
-    wait_until(lambda: replica_set.get_automation_config_tester().reached_version(rs_version + 1))
-    wait_until(lambda: sharded_cluster.get_automation_config_tester().reached_version(sc_version + 1))
-    wait_until(lambda: mc_replica_set.get_automation_config_tester().reached_version(mcrs_version + 1))
+    wait_until(lambda: replica_set.get_automation_config_tester().reached_version(rs_version + 1), timeout=120)
+    wait_until(lambda: sharded_cluster.get_automation_config_tester().reached_version(sc_version + 1), timeout=120)
+    wait_until(lambda: mc_replica_set.get_automation_config_tester().reached_version(mcrs_version + 1), timeout=120)
 
     replica_set.get_automation_config_tester().assert_expected_role(
         role_index=0, expected_value=mongodb_role.get_role()
@@ -200,7 +200,7 @@ def test_removing_role_from_replica_set(replica_set: MongoDB):
     replica_set.update()
 
     replica_set.assert_reaches_phase(Phase.Running)
-    wait_until(lambda: replica_set.get_automation_config_tester().reached_version(rs_version + 1))
+    wait_until(lambda: replica_set.get_automation_config_tester().reached_version(rs_version + 1), timeout=120)
     replica_set.get_automation_config_tester().assert_has_expected_number_of_roles(expected_roles=0)
 
 
@@ -225,7 +225,7 @@ def test_remove_role_from_sharded_cluster(sharded_cluster: MongoDB, mongodb_role
     sharded_cluster.update()
 
     sharded_cluster.assert_reaches_phase(Phase.Running)
-    wait_until(lambda: sharded_cluster.get_automation_config_tester().reached_version(sc_version + 1))
+    wait_until(lambda: sharded_cluster.get_automation_config_tester().reached_version(sc_version + 1), timeout=120)
     sharded_cluster.get_automation_config_tester().assert_has_expected_number_of_roles(expected_roles=0)
 
     # Resource should still exist since MCRS is still referencing it
@@ -243,7 +243,7 @@ def test_remove_role_from_mc_replica_set(mc_replica_set: MongoDBMulti, mongodb_r
     mc_replica_set.update()
 
     mc_replica_set.assert_reaches_phase(Phase.Running)
-    wait_until(lambda: mc_replica_set.get_automation_config_tester().reached_version(mcrs_version + 1))
+    wait_until(lambda: mc_replica_set.get_automation_config_tester().reached_version(mcrs_version + 1), timeout=120)
     mc_replica_set.get_automation_config_tester().assert_has_expected_number_of_roles(expected_roles=0)
 
     # No resources are referencing this role, should be gone