generate the mock test context correctly

Author: mohammedahmed18

tests/test_code_utils.py

@@ -21,6 +21,7 @@
 )
 from codeflash.code_utils.concolic_utils import clean_concolic_tests
 from codeflash.code_utils.coverage_utils import extract_dependent_function, generate_candidates, prepare_coverage_files
+from codeflash.models.models import CodeStringsMarkdown
 
 
 @pytest.fixture
@@ -382,69 +383,76 @@ def mock_code_context():
 def test_extract_dependent_function_sync_and_async(mock_code_context):
     """Test extract_dependent_function with both sync and async functions."""
     # Test sync function extraction
-    mock_code_context.testgen_context_code = """
+    mock_code_context.testgen_context = CodeStringsMarkdown.parse_markdown_code("""```python:file.py
 def main_function():
     pass
 
 def helper_function():
     pass
-"""
+```
+""")
     assert extract_dependent_function("main_function", mock_code_context) == "helper_function"
 
     # Test async function extraction
-    mock_code_context.testgen_context_code = """
+    mock_code_context.testgen_context = CodeStringsMarkdown.parse_markdown_code("""```python:file.py
 def main_function():
     pass
 
 async def async_helper_function():
     pass
-"""
+```
+""")
+
     assert extract_dependent_function("main_function", mock_code_context) == "async_helper_function"
 
 
 def test_extract_dependent_function_edge_cases(mock_code_context):
     """Test extract_dependent_function edge cases."""
     # No dependent functions
-    mock_code_context.testgen_context_code = """
+    mock_code_context.testgen_context = CodeStringsMarkdown.parse_markdown_code("""```python:file.py
 def main_function():
     pass
-"""
+```
+""")
     assert extract_dependent_function("main_function", mock_code_context) is False
 
     # Multiple dependent functions
-    mock_code_context.testgen_context_code = """
+    mock_code_context.testgen_context = CodeStringsMarkdown.parse_markdown_code("""```python:file.py
 def main_function():
     pass
-
 def helper1():
     pass
 
 async def helper2():
     pass
-"""
+```
+""")
     assert extract_dependent_function("main_function", mock_code_context) is False
 
 
 def test_extract_dependent_function_mixed_scenarios(mock_code_context):
     """Test extract_dependent_function with mixed sync/async scenarios."""
     # Async main with sync helper
-    mock_code_context.testgen_context_code = """
+    mock_code_context.testgen_context = CodeStringsMarkdown.parse_markdown_code("""```python:file.py
 async def async_main():
     pass
 
 def sync_helper():
     pass
-"""
+```
+""")
     assert extract_dependent_function("async_main", mock_code_context) == "sync_helper"
 
     # Only async functions
-    mock_code_context.testgen_context_code = """
+    mock_code_context.testgen_context = CodeStringsMarkdown.parse_markdown_code("""```python:file.py
 async def async_main():
     pass
 
 async def async_helper():
     pass
-"""
+```
+""")
+
     assert extract_dependent_function("async_main", mock_code_context) == "async_helper"