From 7d9077f2dcbeebadcf54456886de6bcd51714a6e Mon Sep 17 00:00:00 2001 From: Charles-Andre Bouffard Date: Mon, 17 Dec 2018 14:19:32 -0500 Subject: [PATCH] Change DataDog scheduler_api_tests --- test/scheduler_api_test.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/test/scheduler_api_test.py b/test/scheduler_api_test.py index fa4c02f48b..4a464e38cf 100644 --- 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')