mock here

Author: KRRT7

tests/test_languages/test_js_code_extractor.py

@@ -5,6 +5,7 @@
 
 import shutil
 from pathlib import Path
+from unittest.mock import MagicMock
 
 import pytest
 
@@ -1131,7 +1132,9 @@ def test_function_optimizer_workflow(self, cjs_project):
             pytest_cmd="jest",
         )
 
-        func_optimizer = FunctionOptimizer(function_to_optimize=func, test_cfg=test_config)
+        func_optimizer = FunctionOptimizer(
+            function_to_optimize=func, test_cfg=test_config, aiservice_client=MagicMock()
+        )
         result = func_optimizer.get_code_optimization_context()
 
         from codeflash.either import is_successful

tests/test_languages/test_multi_file_code_replacer.py

@@ -79,6 +79,7 @@
 """
 
 from pathlib import Path
+from unittest.mock import MagicMock
 
 from codeflash.discovery.functions_to_optimize import FunctionToOptimize
 from codeflash.languages.registry import get_language_support
@@ -127,7 +128,9 @@ def test_js_replcement() -> None:
             project_root_path=root_dir,
             pytest_cmd="jest",
         )
-        func_optimizer = FunctionOptimizer(function_to_optimize=func, test_cfg=test_config)
+        func_optimizer = FunctionOptimizer(
+            function_to_optimize=func, test_cfg=test_config, aiservice_client=MagicMock()
+        )
         result = func_optimizer.get_code_optimization_context()
         from codeflash.either import is_successful