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")
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")
