143
143
mgr.update_service_capabilities('foo_svc', 'foo_host', 'foo_caps')
144
144
self.assertTrue(mgr.drivers['compute'].is_update_caps_called)
145
145
self.assertTrue(mgr.drivers['volume'].is_update_caps_called)
148
class SimpleSchedulerTestCase(MultiDriverTestCase):
149
"""Test case for simple driver."""
151
driver_cls = multi.MultiScheduler
154
super(SimpleSchedulerTestCase, self).setUp()
155
base_name = 'nova.tests.scheduler.test_multi_scheduler.%s'
156
compute_cls_name = base_name % 'FakeComputeScheduler'
157
volume_cls_name = 'nova.scheduler.simple.SimpleScheduler'
158
default_cls_name = base_name % 'FakeDefaultScheduler'
159
self.flags(compute_scheduler_driver=compute_cls_name,
160
volume_scheduler_driver=volume_cls_name,
161
default_scheduler_driver=default_cls_name)
162
self._manager = multi.MultiScheduler()
164
def test_update_service_capabilities(self):
165
def fake_update_service_capabilities(self, service, host, caps):
166
self.is_update_caps_called = True
169
self.stubs.Set(driver.Scheduler,
170
'update_service_capabilities',
171
fake_update_service_capabilities)
172
self.assertFalse(mgr.drivers['compute'].is_update_caps_called)
173
mgr.update_service_capabilities('foo_svc', 'foo_host', 'foo_caps')
174
self.assertTrue(mgr.drivers['compute'].is_update_caps_called)
175
self.assertTrue(mgr.drivers['volume'].is_update_caps_called)
177
def test_drivers_inited(self):
179
self.assertEqual(len(mgr.drivers), 3)
180
self.assertTrue(mgr.drivers['compute'].is_fake_compute)
181
self.assertTrue(mgr.drivers['volume'] is not None)
182
self.assertTrue(mgr.drivers['default'].is_fake_default)
184
def test_proxy_calls(self):