checkf or ac change instead

Author: nammn

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

@@ -133,21 +133,12 @@ def test_user_cannot_authenticate_with_incorrect_password(self, replica_set: Mon
 @mark.e2e_replica_set_scram_sha_256_user_connectivity
 class TestCanChangePassword(KubernetesTester):
     def test_user_can_authenticate_with_new_password(self, namespace: str, replica_set: MongoDB):
-        initial_tester = replica_set.get_automation_config_tester()
-        initial_version = initial_tester.automation_config.get("version", 0)
+        ac_version = replica_set.get_automation_config_tester().automation_config["version"]
 
         new_password = "my-new-password7"
         update_secret(namespace, PASSWORD_SECRET_NAME, {"password": new_password})
 
-        def ac_updated() -> bool:
-            try:
-                tester = replica_set.get_automation_config_tester()
-                current_version = tester.automation_config.get("version", 0)
-                return current_version > initial_version
-            except Exception:
-                return False
-
-        wait_until(ac_updated, timeout=600)
+        wait_until(lambda: replica_set.get_automation_config_tester().reached_version(ac_version + 1), timeout=800)
 
         replica_set.tester().assert_scram_sha_authentication(
             password=new_password,