def test_get_devs_object(self): def _fake_obj_load_attr(foo, attrname): if attrname == 'pci_devices': self.load_attr_called = True foo.pci_devices = objects.PciDeviceList() inst = fakes.stub_instance(id='1') self.mox.StubOutWithMock(db, 'instance_get') db.instance_get(self.fake_context, '1', columns_to_join=[] ).AndReturn(inst) self.mox.ReplayAll() inst = objects.Instance.get_by_id(self.fake_context, '1', expected_attrs=[]) self.stubs.Set(objects.Instance, 'obj_load_attr', _fake_obj_load_attr) self.load_attr_called = False manager.get_instance_pci_devs(inst) self.assertEqual(self.load_attr_called, True)
def test_get_devs_object(self): def _fake_obj_load_attr(foo, attrname): if attrname == 'pci_devices': self.load_attr_called = True foo.pci_devices = objects.PciDeviceList() inst = fakes.stub_instance(id='1') self.mox.StubOutWithMock(db, 'instance_get') db.instance_get(self.fake_context, '1', columns_to_join=[]).AndReturn(inst) self.mox.ReplayAll() inst = objects.Instance.get_by_id(self.fake_context, '1', expected_attrs=[]) self.stubs.Set(objects.Instance, 'obj_load_attr', _fake_obj_load_attr) self.load_attr_called = False manager.get_instance_pci_devs(inst) self.assertEqual(self.load_attr_called, True)
def test_add_console(self): instance = self._create_instance() self.console.add_console(self.context, instance['id']) instance = db.instance_get(self.context, instance['id']) pool = db.console_pool_get_by_host_type(self.context, instance['host'], self.console.host, self.console.driver.console_type) console_instances = [con['instance_uuid'] for con in pool['consoles']] self.assertIn(instance['uuid'], console_instances) db.instance_destroy(self.context, instance['uuid'])
def test_get_test_instance(self): # get_test_instance's return value looks like an instance_ref. instance_ref = test_utils.get_test_instance() ctxt = test_utils.get_test_admin_context() db.instance_get(ctxt, instance_ref['id'])