def test_pre_populate_with_no_resources(self, enabled_drivers):
fake_scheduler = mock.Mock()
fake_scheduler.handle_message = mock.Mock()
fake_driver = fakes.fake_driver()
fake_driver.pre_populate_hook.return_value = []
enabled_drivers.return_value = [fake_driver]
populate._pre_populate_workers(fake_scheduler)
self.assertFalse(fake_scheduler.handle_message.called)
def test_pre_populate_with_no_resources(self, enabled_drivers):
fake_scheduler = mock.Mock()
fake_scheduler.handle_message = mock.Mock()
fake_driver = fakes.fake_driver()
fake_driver.pre_populate_hook.return_value = []
enabled_drivers.return_value = [fake_driver]
populate._pre_populate_workers(fake_scheduler)
self.assertFalse(fake_scheduler.handle_message.called)
def test_pre_populate_with_resources(self, enabled_drivers):
fake_scheduler = mock.Mock()
fake_scheduler.handle_message = mock.Mock()
fake_driver = fakes.fake_driver()
fake_resources = [
Resource(
id='fake_resource_%s' % i,
tenant_id='fake_tenant_%s' % i,
driver=fake_driver.RESOURCE_NAME,
) for i in range(2)
]
fake_driver.pre_populate_hook.return_value = fake_resources
enabled_drivers.return_value = [fake_driver]
populate._pre_populate_workers(fake_scheduler)
for res in fake_resources:
e = event.Event(resource=res, crud=event.POLL, body={})
call = mock.call(res.tenant_id, e)
self.assertIn(call, fake_scheduler.handle_message.call_args_list)
def test_pre_populate_with_resources(self, enabled_drivers):
fake_scheduler = mock.Mock()
fake_scheduler.handle_message = mock.Mock()
fake_driver = fakes.fake_driver()
fake_resources = [
Resource(
id='fake_resource_%s' % i,
tenant_id='fake_tenant_%s' % i,
driver=fake_driver.RESOURCE_NAME,
) for i in range(2)
]
fake_driver.pre_populate_hook.return_value = fake_resources
enabled_drivers.return_value = [fake_driver]
populate._pre_populate_workers(fake_scheduler)
for res in fake_resources:
e = event.Event(resource=res, crud=event.POLL, body={})
call = mock.call(res.tenant_id, e)
self.assertIn(call, fake_scheduler.handle_message.call_args_list)
