codemie-test-harness 0.1.146__py3-none-any.whl → 0.1.147__py3-none-any.whl

codemie_test_harness/tests/conftest.py

@@ -64,13 +64,15 @@ def pytest_configure(config):
     """
     from codemie_test_harness.cli.runner import resolve_tests_path_and_root
     from codemie_test_harness.tests.utils.aws_parameters_store import AwsParameterStore
+    from codemie_test_harness.tests.utils.env_utils import EnvManager
 
     # Resolve the root directory and .env file path
     _, root_dir = resolve_tests_path_and_root()
     env_file_path = Path(root_dir) / ".env"
 
     # Load initial .env file
-    load_dotenv(env_file_path)
+    if env_file_path.exists():
+        load_dotenv(env_file_path)
 
     # Check if AWS credentials are available for parameter store
     if os.getenv("AWS_ACCESS_KEY") and os.getenv("AWS_SECRET_KEY"):
@@ -85,12 +87,10 @@ def pytest_configure(config):
             f"/codemie/autotests/dotenv/{os.getenv('ENV')}"
         )
 
-        # Write the AWS parameters to .env file
-        with open(env_file_path, "w") as file:
-            file.write(dotenv)
-
-        # Reload .env file with AWS parameters
-        load_dotenv(env_file_path)
+        # Safely update .env file with new content
+        EnvManager.update_env_file_safely(
+            env_file_path=env_file_path, new_content=dotenv, clear_old_vars=True
+        )
 
 
 def pytest_unconfigure(config):

codemie_test_harness/tests/utils/env_utils.py

@@ -0,0 +1,125 @@
+"""
+Utility functions for managing environment variables and dotenv files with caching prevention.
+"""
+
+import os
+import time
+from pathlib import Path
+from dotenv import load_dotenv
+
+
+class EnvManager:
+    """
+    Manages environment variables with proper caching prevention.
+    """
+
+    @staticmethod
+    def clear_env_vars_from_file(env_file_path: Path) -> None:
+        """
+        Clear environment variables that exist in the specified .env file from os.environ.
+
+        Args:
+            env_file_path: Path to the .env file
+        """
+        if not env_file_path.exists():
+            return
+
+        with open(env_file_path, "r") as f:
+            content = f.read()
+
+        for line in content.splitlines():
+            line = line.strip()
+            if line and "=" in line and not line.startswith("#"):
+                key = line.split("=", 1)[0].strip()
+                os.environ.pop(key, None)
+
+    @staticmethod
+    def write_env_file_safely(env_file_path: Path, content: str) -> None:
+        """
+        Write content to .env file with proper flushing and sync.
+
+        Args:
+            env_file_path: Path to the .env file
+            content: Content to write
+        """
+        # Ensure parent directory exists
+        env_file_path.parent.mkdir(parents=True, exist_ok=True)
+
+        # Write with proper flushing
+        with open(env_file_path, "w", encoding="utf-8") as file:
+            file.write(content)
+            file.flush()
+            os.fsync(file.fileno())
+
+    @staticmethod
+    def load_env_with_retry(
+        env_file_path: Path,
+        max_retries: int = 3,
+        retry_delay: float = 0.1,
+        override: bool = True,
+    ) -> bool:
+        """
+        Load .env file with retry mechanism to handle caching issues.
+
+        Args:
+            env_file_path: Path to the .env file
+            max_retries: Maximum number of retry attempts
+            retry_delay: Delay between retries in seconds
+            override: Whether to override existing environment variables
+
+        Returns:
+            bool: True if successfully loaded, False otherwise
+        """
+        for attempt in range(max_retries):
+            try:
+                # Small delay to ensure file system consistency
+                if attempt > 0:
+                    time.sleep(retry_delay)
+
+                # Verify file exists and is readable
+                if not env_file_path.exists():
+                    raise FileNotFoundError(f".env file not found: {env_file_path}")
+
+                # Read file content to verify it's accessible
+                with open(env_file_path, "r") as f:
+                    content = f.read()
+
+                if not content.strip():
+                    raise ValueError("Empty .env file")
+
+                # Load environment variables
+                success = load_dotenv(env_file_path, override=override)
+
+                if success:
+                    return True
+
+            except Exception as e:
+                if attempt == max_retries - 1:
+                    raise Exception(
+                        f"Failed to load .env file after {max_retries} attempts: {e}"
+                    )
+                continue
+
+        return False
+
+    @staticmethod
+    def update_env_file_safely(
+        env_file_path: Path, new_content: str, clear_old_vars: bool = True
+    ) -> None:
+        """
+        Safely update .env file with new content, handling caching issues.
+
+        Args:
+            env_file_path: Path to the .env file
+            new_content: New content to write
+            clear_old_vars: Whether to clear old environment variables
+        """
+        # Clear old environment variables if requested
+        if clear_old_vars:
+            EnvManager.clear_env_vars_from_file(env_file_path)
+
+        # Write new content safely
+        EnvManager.write_env_file_safely(env_file_path, new_content)
+
+        # Load new environment variables with retry
+        EnvManager.load_env_with_retry(env_file_path, override=True)

{codemie_test_harness-0.1.146.dist-info → codemie_test_harness-0.1.147.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: codemie-test-harness
-Version: 0.1.146
+Version: 0.1.147
 Summary: Autotest for CodeMie backend and UI
 Author: Anton Yeromin
 Author-email: anton_yeromin@epam.com
@@ -13,7 +13,7 @@ Requires-Dist: aws-assume-role-lib (>=2.10.0,<3.0.0)
 Requires-Dist: boto3 (>=1.39.8,<2.0.0)
 Requires-Dist: click (>=8.1.7,<9.0.0)
 Requires-Dist: codemie-plugins (>=0.1.123,<0.2.0)
-Requires-Dist: codemie-sdk-python (==0.1.146)
+Requires-Dist: codemie-sdk-python (==0.1.147)
 Requires-Dist: pytest (>=8.4.1,<9.0.0)
 Requires-Dist: pytest-playwright (>=0.7.0,<0.8.0)
 Requires-Dist: pytest-reportportal (>=5.5.2,<6.0.0)

{codemie_test_harness-0.1.146.dist-info → codemie_test_harness-0.1.147.dist-info}/RECORD

@@ -55,7 +55,7 @@ codemie_test_harness/tests/assistant/tools/servicenow/__init__.py,sha256=47DEQpj
 codemie_test_harness/tests/assistant/tools/servicenow/test_servicenow_tools.py,sha256=JV_n6COaCsd0_rxQBJvvLZfPdo8BsCbG8Ti-UpzFvfQ,644
 codemie_test_harness/tests/assistant/tools/vcs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/assistant/tools/vcs/test_assistant_with_vcs_tools.py,sha256=5-QybLWmYSqHG4Sqr-IN-vNcStV4skSwCJQN44Nfats,922
-codemie_test_harness/tests/conftest.py,sha256=pS_7Zh2QLUMgC2wepEFx-ut1-UzOGZy_WIn9rb7z2D4,28557
+codemie_test_harness/tests/conftest.py,sha256=QPqNG47RE7tqzqMW21VMlEeijY2q9PYQNwbTauDkTmY,28638
 codemie_test_harness/tests/conversations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/conversations/test_conversations_endpoints.py,sha256=SQraRVrjjE4mYTs4EuGzMrO96DuOuKTiYQ4ZRBhHx70,3913
 codemie_test_harness/tests/e2e/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -217,6 +217,7 @@ codemie_test_harness/tests/utils/client_factory.py,sha256=Yyg2iXe7g7fIfIUbOH8z_8
 codemie_test_harness/tests/utils/constants.py,sha256=lb6mgLjt0GM85Khs18rL1jnE9ILx8L811ykNhpF6IDA,1111
 codemie_test_harness/tests/utils/conversation_utils.py,sha256=SWj6TBWOQoX5Yh6Wk63yHQFveRXgK1mpLb3PUKAa57A,648
 codemie_test_harness/tests/utils/datasource_utils.py,sha256=-_HZfW_UufSUVkSUCVq4jMlq9MCcKtMhzD6iBtadzmk,12502
+codemie_test_harness/tests/utils/env_utils.py,sha256=9tyVgxKfYqdtSoo9dRTScOZWjAUm82_65JjaKggcwCg,3999
 codemie_test_harness/tests/utils/file_utils.py,sha256=hY-kwnyzvtd1BQif8r5NhvRTGfpKLmQKyRsq1Tuflhg,585
 codemie_test_harness/tests/utils/gitbud_utils.py,sha256=UJ3RbhPSjHQSdos6S6zTR9iZULrBDJXoXq9cbjFH7bo,7829
 codemie_test_harness/tests/utils/http_utils.py,sha256=wjhttibzzNhleKzWgWC01Q0Y5sV9scu-Ski-qgJPd-Q,4179
@@ -326,7 +327,7 @@ codemie_test_harness/tests/workflow/virtual_assistant_tools/servicenow/__init__.
 codemie_test_harness/tests/workflow/virtual_assistant_tools/servicenow/test_workflow_with_servicenow_tools.py,sha256=Cxe5Yuy9JE_IEsnQOThpVIZQqpfLuHfDkF6JwLngDc8,890
 codemie_test_harness/tests/workflow/virtual_assistant_tools/vcs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/workflow/virtual_assistant_tools/vcs/test_workflow_with_vcs_tools.py,sha256=tm0NvUf_UtzLPTYJWykYpsNoxoVs-Eh80guDmRwImwg,1106
-codemie_test_harness-0.1.146.dist-info/METADATA,sha256=C9bfCCtoQp9HHKSUzlo4GdgfZpMtU1OQPG_yf7GHQxs,8998
-codemie_test_harness-0.1.146.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-codemie_test_harness-0.1.146.dist-info/entry_points.txt,sha256=n98t-EOM5M1mnMl_j2X4siyeO9zr0WD9a5LF7JyElIM,73
-codemie_test_harness-0.1.146.dist-info/RECORD,,
+codemie_test_harness-0.1.147.dist-info/METADATA,sha256=p7qB5ZcTZmcvbvhjgT0nucOvVLqFQRZV54T2v0GCy68,8998
+codemie_test_harness-0.1.147.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+codemie_test_harness-0.1.147.dist-info/entry_points.txt,sha256=n98t-EOM5M1mnMl_j2X4siyeO9zr0WD9a5LF7JyElIM,73
+codemie_test_harness-0.1.147.dist-info/RECORD,,