Update test_metrics_collector_events.py

Author: v1r3n

tests/unit/event/test_metrics_collector_events.py

@@ -45,8 +45,8 @@ def test_on_poll_completed(self):
             )
             self.collector.on_poll_completed(event)
 
-            # Duration should be converted from ms to seconds
-            mock_record.assert_called_once_with("test_task", 0.25)
+            # Duration should be converted from ms to seconds, status added
+            mock_record.assert_called_once_with("test_task", 0.25, status="SUCCESS")
 
     def test_on_poll_failure(self):
         """Test on_poll_failure event handler"""
@@ -87,8 +87,8 @@ def test_on_task_execution_completed(self):
             )
             self.collector.on_task_execution_completed(event)
 
-            # Duration should be converted from ms to seconds
-            mock_time.assert_called_once_with("test_task", 0.5)
+            # Duration should be converted from ms to seconds, status added
+            mock_time.assert_called_once_with("test_task", 0.5, status="SUCCESS")
             mock_size.assert_called_once_with("test_task", 1024)
 
     def test_on_task_execution_completed_no_output_size(self):
@@ -106,7 +106,7 @@ def test_on_task_execution_completed_no_output_size(self):
             )
             self.collector.on_task_execution_completed(event)
 
-            mock_time.assert_called_once_with("test_task", 0.5)
+            mock_time.assert_called_once_with("test_task", 0.5, status="SUCCESS")
             # Should not record size if None
             mock_size.assert_not_called()