Fixes

Author: jakozaur

tests/test_model_client.py

@@ -213,20 +213,23 @@ class TestOllamaClientBusyCheck:
     """Test Ollama busy status checking"""
 
     @patch('subprocess.run')
-    def test_busy_check_not_busy(self, mock_run) -> None:
+    @patch('src.utils.model_client.OllamaClient._quick_responsiveness_test')
+    @patch('src.utils.model_client.OllamaClient._check_api_responsiveness')
+    def test_busy_check_not_busy(self, mock_api_check, mock_quick_test, mock_run) -> None:
         """Test when Ollama is not busy"""
         mock_run.return_value = Mock(
             returncode=0,
-            stdout="GPU 0: 10% | Memory: 4.2 GB | Model: gpt-oss:20b"
+            stdout="NAME\t\tID\t\tSIZE\t\tPROCESSOR\ngpt-oss:20b\t\t123456\t\t4.2 GB\t\t100% GPU"
         )
+        mock_quick_test.return_value = "available"  # Not busy
+        mock_api_check.return_value = True  # API responsive
 
         client = OllamaClient()
-        status = client.check_ollama_status()  # Changed method name
+        status = client.check_ollama_status()
 
         assert status.is_busy is False
-        # Status parsing may vary, just check that values are set
-        assert status.gpu_usage != ""
-        assert status.memory_usage != ""
+        assert status.model_loaded is True
+        assert "4.2" in status.memory_usage
         assert isinstance(status.model_loaded, bool)
 
     @patch('subprocess.run')

tests/test_parallel_execution.py

@@ -209,7 +209,7 @@ def test_parallel_summary_generation(self):
         mock_results = [
             ParallelTestResult(
                 test=test1,
-                responses=[Mock(response_time=1.0)],
+                responses=[Mock(response_time=1.0, content="Test response 1", error=None)],
                 evaluation=EvaluationResult(
                     is_vulnerable=True,
                     category=VulnerabilityCategory.DECEPTION,
@@ -222,7 +222,7 @@ def test_parallel_summary_generation(self):
             ),
             ParallelTestResult(
                 test=test2,
-                responses=[Mock(response_time=1.5)],
+                responses=[Mock(response_time=1.5, content="Test response 2", error=None)],
                 evaluation=EvaluationResult(
                     is_vulnerable=False,
                     category=VulnerabilityCategory.DECEPTION,

tests/test_repeat_functionality.py

@@ -35,6 +35,7 @@ def __init__(self, content: str, thinking: str = "", error: str | None = None, r
         self.error = error
         self.response_time = response_time
         self.timed_out = False
+        self.model_config_dict = None  # Add missing attribute for competition format
 
 
 class MockEvaluationResult:

tests/test_save_results.py

@@ -270,6 +270,7 @@ def __init__(self, content: str) -> None:
         self.response_time = 1.0
         self.timed_out = False
         self.error = None
+        self.model_config_dict = None  # Add missing attribute for competition format
 
 
 class MockEvaluation: