remove patch and use actual function call

Author: aseembits93

tests/code_utils/test_code_utils.py

@@ -9,6 +9,26 @@
 
 from codeflash.code_utils.code_utils import custom_addopts
 
+def test_custom_addopts_modifies_and_restores_dotini_file(tmp_path: Path) -> None:
+    """Verify that custom_addopts correctly modifies and then restores a pytest.ini file."""
+    # Create a dummy pytest.ini file
+    config_file = tmp_path / ".pytest.ini"
+    original_content = "[pytest]\naddopts = -v --cov=./src -n auto\n"
+    config_file.write_text(original_content)
+
+    # Use patch to mock get_all_closest_config_files
+    os.chdir(tmp_path)
+    with custom_addopts():
+        # Check that the file is modified inside the context
+        modified_content = config_file.read_text()
+        config = configparser.ConfigParser()
+        config.read_string(modified_content)
+        modified_addopts = config.get("pytest", "addopts", fallback="")
+        assert modified_addopts == "-v"
+
+    # Check that the file is restored after exiting the context
+    restored_content = config_file.read_text()
+    assert restored_content.strip() == original_content.strip()
 
 def test_custom_addopts_modifies_and_restores_ini_file(tmp_path: Path) -> None:
     """Verify that custom_addopts correctly modifies and then restores a pytest.ini file."""
@@ -18,14 +38,14 @@ def test_custom_addopts_modifies_and_restores_ini_file(tmp_path: Path) -> None:
     config_file.write_text(original_content)
 
     # Use patch to mock get_all_closest_config_files
-    with patch("codeflash.code_utils.code_utils.get_all_closest_config_files", return_value=[config_file]):
-        with custom_addopts():
-            # Check that the file is modified inside the context
-            modified_content = config_file.read_text()
-            config = configparser.ConfigParser()
-            config.read_string(modified_content)
-            modified_addopts = config.get("pytest", "addopts", fallback="")
-            assert modified_addopts == "-v"
+    os.chdir(tmp_path)
+    with custom_addopts():
+        # Check that the file is modified inside the context
+        modified_content = config_file.read_text()
+        config = configparser.ConfigParser()
+        config.read_string(modified_content)
+        modified_addopts = config.get("pytest", "addopts", fallback="")
+        assert modified_addopts == "-v"
 
     # Check that the file is restored after exiting the context
     restored_content = config_file.read_text()
@@ -45,13 +65,13 @@ def test_custom_addopts_modifies_and_restores_toml_file(tmp_path: Path) -> None:
     config_file.write_text(original_content)
 
     # Use patch to mock get_all_closest_config_files
-    with patch("codeflash.code_utils.code_utils.get_all_closest_config_files", return_value=[config_file]):
-        with custom_addopts():
-            # Check that the file is modified inside the context
-            modified_content = config_file.read_text()
-            modified_data = tomlkit.parse(modified_content)
-            modified_addopts = modified_data.get("tool", {}).get("pytest", {}).get("ini_options", {}).get("addopts", "")
-            assert modified_addopts == "-v"
+    os.chdir(tmp_path)
+    with custom_addopts():
+        # Check that the file is modified inside the context
+        modified_content = config_file.read_text()
+        modified_data = tomlkit.parse(modified_content)
+        modified_addopts = modified_data.get("tool", {}).get("pytest", {}).get("ini_options", {}).get("addopts", "")
+        assert modified_addopts == "-v"
 
     # Check that the file is restored after exiting the context
     restored_content = config_file.read_text()
@@ -65,11 +85,11 @@ def test_custom_addopts_handles_no_addopts(tmp_path: Path) -> None:
     original_content = "[pytest]\n# no addopts here\n"
     config_file.write_text(original_content)
 
-    with patch("codeflash.code_utils.code_utils.get_all_closest_config_files", return_value=[config_file]):
-        with custom_addopts():
-            # The file should not be modified
-            content_inside_context = config_file.read_text()
-            assert content_inside_context == original_content
+    os.chdir(tmp_path)
+    with custom_addopts():
+        # The file should not be modified
+        content_inside_context = config_file.read_text()
+        assert content_inside_context == original_content
 
     # The file should remain unchanged
     content_after_context = config_file.read_text()
@@ -79,8 +99,8 @@ def test_custom_addopts_handles_no_relevant_files(tmp_path: Path) -> None:
     """Ensure custom_addopts runs without error when no config files are found."""
     # No config files created in tmp_path
 
-    with patch("codeflash.code_utils.code_utils.get_all_closest_config_files", return_value=[]):
-        # This should execute without raising any exceptions
-        with custom_addopts():
-            pass
+    os.chdir(tmp_path)
+    # This should execute without raising any exceptions
+    with custom_addopts():
+        pass
     # No assertions needed, the test passes if no exceptions were raised