Change DataDog scheduler_api_tests

spotify · Dec 17, 2018 · 7d9077f · 7d9077f
1 parent d387b99
commit 7d9077f
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/test/scheduler_api_test.py b/test/scheduler_api_test.py
@@ -2244,42 +2244,42 @@ def test_you_can_forgive_failures_twice(self):
         self.sch.forgive_failures(task_id='A')
         self.assertEqual(self.sch.get_work(worker=WORKER)['task_id'], 'A')
 
-    @mock.patch('luigi.metrics.MetricsCollector')
+    @mock.patch('luigi.contrib.NoMetricsCollector')
     def test_collector_metrics_on_task_started(self, MetricsCollector):
         from luigi.metrics import MetricsCollectors
 
-        s = Scheduler(metrics_collector=MetricsCollectors.default)
+        s = Scheduler(metrics_collector=MetricsCollectors.none)
         s.add_task(worker=WORKER, task_id='A', status=PENDING)
         s.get_work(worker=WORKER)
 
         task = s._state.get_task('A')
         MetricsCollector().handle_task_started.assert_called_once_with(task)
 
-    @mock.patch('luigi.metrics.MetricsCollector')
+    @mock.patch('luigi.metrics.NoMetricsCollector')
     def test_collector_metrics_on_task_disabled(self, MetricsCollector):
         from luigi.metrics import MetricsCollectors
 
-        s = Scheduler(metrics_collector=MetricsCollectors.default, retry_count=0)
+        s = Scheduler(metrics_collector=MetricsCollectors.none, retry_count=0)
         s.add_task(worker=WORKER, task_id='A', status=FAILED)
 
         task = s._state.get_task('A')
         MetricsCollector().handle_task_disabled.assert_called_once_with(task, s._config)
 
-    @mock.patch('luigi.metrics.MetricsCollector')
+    @mock.patch('luigi.metrics.NoMetricsCollector')
     def test_collector_metrics_on_task_failed(self, MetricsCollector):
         from luigi.metrics import MetricsCollectors
 
-        s = Scheduler(metrics_collector=MetricsCollectors.default)
+        s = Scheduler(metrics_collector=MetricsCollectors.none)
         s.add_task(worker=WORKER, task_id='A', status=FAILED)
 
         task = s._state.get_task('A')
         MetricsCollector().handle_task_failed.assert_called_once_with(task)
 
-    @mock.patch('luigi.metrics.MetricsCollector')
+    @mock.patch('luigi.metrics.NoMetricsCollector')
     def test_collector_metrics_on_task_done(self, MetricsCollector):
         from luigi.metrics import MetricsCollectors
 
-        s = Scheduler(metrics_collector=MetricsCollectors.default)
+        s = Scheduler(metrics_collector=MetricsCollectors.none)
         s.add_task(worker=WORKER, task_id='A', status=DONE)
 
         task = s._state.get_task('A')