def test_mock_store_with_scheduler(self): """ SchedulerService is not created with mock store """ self.config['mock'] = True set_config_data(self.config) setup_scheduler(self.parent, self.store, self.kz_client) self.assertFalse(self.store.set_scheduler_buckets.called) self.assertFalse(self.scheduler_service.called)
def test_success(self): """ `SchedulerService` is configured with config values and set as parent to passed `MultiService` """ setup_scheduler(self.parent, self.store, self.kz_client) buckets = range(1, 11) self.store.set_scheduler_buckets.assert_called_once_with(buckets) self.scheduler_service.assert_called_once_with( 100, 10, self.store, self.kz_client, '/part_path', 15, buckets) self.scheduler_service.return_value.setServiceParent.assert_called_once_with(self.parent)
def test_mock_store_with_scheduler(self): """ SchedulerService is not created with mock store """ self.config['mock'] = True set_config_data(self.config) self.assertIs( setup_scheduler(self.parent, "disp", self.store, self.kz_client), None) self.assertFalse(self.store.set_scheduler_buckets.called)
def test_success(self): """ `SchedulerService` is configured with config values and set as parent to passed `MultiService` """ svc = setup_scheduler(self.parent, "disp", self.store, self.kz_client) buckets = range(1, 11) self.store.set_scheduler_buckets.assert_called_once_with(buckets) self.assertEqual(self.parent.services, [svc]) self.assertEqual(svc.store, self.store) self.assertEqual(svc.partitioner.buckets, buckets) self.assertEqual(svc.partitioner.kz_client, self.kz_client) self.assertEqual(svc.partitioner.partitioner_path, '/part_path') self.assertEqual(svc.dispatcher, "disp")