add user wait and ac uplload

Author: nammn

docker/mongodb-kubernetes-tests/kubetester/mongotester.py

@@ -9,17 +9,16 @@
 from typing import Callable, Dict, List, Optional
 
 import pymongo
+from kubetester import kubetester
+from kubetester.kubetester import KubernetesTester
+from kubetester.mongodb_user import MongoDBUser
+from kubetester.phase import Phase
 from opentelemetry import trace
 from pycognito import Cognito
 from pymongo.auth_oidc import OIDCCallback, OIDCCallbackContext, OIDCCallbackResult
 from pymongo.errors import OperationFailure, PyMongoError, ServerSelectionTimeoutError
 from pytest import fail
 
-from kubetester import kubetester
-from kubetester.kubetester import KubernetesTester
-from kubetester.mongodb_user import MongoDBUser
-from kubetester.phase import Phase
-
 TEST_DB = "test-db"
 TEST_COLLECTION = "test-collection"
 
@@ -88,18 +87,15 @@ def _wait_for_mongodbuser_reconciliation() -> None:
     """
     try:
         # Import inside function to avoid circular imports
-        from tests.conftest import get_central_cluster_client
         import kubernetes.client as client
-        
+        from tests.conftest import get_central_cluster_client
+
         namespace = KubernetesTester.get_namespace()
         api_client = client.CustomObjectsApi(api_client=get_central_cluster_client())
 
         try:
             mongodb_users = api_client.list_namespaced_custom_object(
-                group="mongodb.com",
-                version="v1",
-                namespace=namespace,
-                plural="mongodbusers"
+                group="mongodb.com", version="v1", namespace=namespace, plural="mongodbusers"
             )
 
             all_users = []
@@ -112,11 +108,15 @@ def _wait_for_mongodbuser_reconciliation() -> None:
             if not all_users:
                 return
 
-            logging.info(f"Found {len(all_users)} MongoDBUser resource(s) in namespace '{namespace}', waiting for all to reach Updated phase...")
+            logging.info(
+                f"Found {len(all_users)} MongoDBUser resource(s) in namespace '{namespace}', waiting for all to reach Updated phase..."
+            )
 
             for user_name, username in all_users:
                 try:
-                    logging.info(f"Waiting for MongoDBUser '{user_name}' (username: {username}) to reach Updated phase...")
+                    logging.info(
+                        f"Waiting for MongoDBUser '{user_name}' (username: {username}) to reach Updated phase..."
+                    )
 
                     user = MongoDBUser(name=user_name, namespace=namespace)
                     user.assert_reaches_phase(Phase.Updated, timeout=300)
@@ -263,7 +263,7 @@ def assert_scram_sha_authentication(
                 if remaining_attempts >= attempts * 0.7:  # First ~30% of attempts
                     delay = 10  # Longer delay for initial propagation
                 else:
-                    delay = 5   # Standard delay for normal retries
+                    delay = 5  # Standard delay for normal retries
 
                 time.sleep(delay)
 

docker/mongodb-kubernetes-tests/tests/conftest.py

@@ -1692,7 +1692,9 @@ def pytest_sessionfinish(session, exitstatus):
                             if not automation_config:
                                 continue
 
-                            with open(f"/tmp/diagnostics/{project_id}-automation-config.json", "w", encoding="utf-8") as f:
+                            with open(
+                                f"/tmp/diagnostics/{project_id}-automation-config.json", "w", encoding="utf-8"
+                            ) as f:
                                 json.dump(automation_config, f, ensure_ascii=False, indent=4)
 
                             logging.info(f"Saved automation config for project {project_id}")