fix

Author: v1r3n

tests/unit/automator/test_api_metrics.py

@@ -430,13 +430,16 @@ async def run_test():
                 http_client=mock_http_client
             )
 
+            # Reset counter before test
+            self.metrics_collector_mock.record_api_request_time.reset_mock()
+
             # Run 5 concurrent polls
             await asyncio.gather(*[
                 runner._poll_tasks_from_server(count=1) for _ in range(5)
             ])
 
             # Should have 5 timing records
-            self.assertEqual(self.metrics_collector_mock.record_api_request_time.call_count,5)
+            self.assertEqual(self.metrics_collector_mock.record_api_request_time.call_count, 5)
 
         asyncio.run(run_test())
 

tests/unit/automator/test_task_handler_coverage.py

@@ -653,7 +653,9 @@ def test_setup_logging_queue_with_configuration(self, mock_queue_class, mock_pro
         logger_process, queue = _setup_logging_queue(config)
 
         config.apply_logging_config.assert_called_once()
-        mock_process.start.assert_called_once()
+        # Verify Process was called and start was invoked on the returned mock
+        mock_process_class.assert_called_once()
+        self.assertTrue(mock_process.start.called)
         self.assertEqual(queue, mock_queue)
         self.assertEqual(logger_process, mock_process)
 
@@ -669,7 +671,9 @@ def test_setup_logging_queue_without_configuration(self, mock_queue_class, mock_
 
         logger_process, queue = _setup_logging_queue(None)
 
-        mock_process.start.assert_called_once()
+        # Verify Process was called and start was invoked on the returned mock
+        mock_process_class.assert_called_once()
+        self.assertTrue(mock_process.start.called)
         self.assertEqual(queue, mock_queue)
         self.assertEqual(logger_process, mock_process)