def setUp(self): VirtualMachineSM._object_db = mock.MagicMock() VirtualMachineSM._object_db.object_read = test_utils.vm_db_read VirtualRouterSM._object_db = mock.MagicMock() VirtualRouterSM._object_db.object_read = test_utils.vr_db_read VirtualMachineInterfaceSM._object_db = mock.MagicMock() VirtualMachineInterfaceSM._object_db.object_read = test_utils.vmi_db_read InstanceIpSM._object_db = mock.MagicMock() InstanceIpSM._object_db.object_read = test_utils.iip_db_read self.mocked_vnc = mock.MagicMock() self.mocked_vnc.fq_name_to_id = test_utils.get_vn_id_for_fq_name self.mocked_vnc.virtual_machine_interface_create = test_utils.vmi_create self.mocked_vnc.instance_ip_create = test_utils.iip_create self.nova_mock = mock.MagicMock() self.mocked_db = mock.MagicMock() self.mocked_args = mock.MagicMock() self.mocked_args.availability_zone = None self.vrouter_manager = VRouterInstanceManager( db=self.mocked_db, logger=mock.MagicMock(), vnc_lib=self.mocked_vnc, vrouter_scheduler=mock.MagicMock(), nova_client=self.nova_mock, args=self.mocked_args, agent_manager=mock.MagicMock())
def setUp(self): def get_vn_id(obj_type, fq_name): if obj_type != 'virtual-network': return for vn in VirtualNetworkSM.values(): if vn.fq_name == fq_name: return vn.uuid raise NoIdError(fq_name) def vm_create(vm_obj): vm_obj.uuid = 'fake-vm-uuid' return def vmi_create(vmi_obj): vmi_obj.uuid = 'fake-vmi-uuid' return def vm_read(vm_id): class SI(object): def __init__(self, name, fq_name): self.name = name self.fq_name = fq_name vm_obj = {} vm_obj['uuid'] = 'fake-vm-uuid' vm_obj['fq_name'] = ['fake-vm-uuid'] fq_name = ['fake-domain', 'fake-project', 'fake-instance'] name = 'fake-instance' si = SI(name, fq_name) instance_name = self.vrouter_manager._get_instance_name(si, 0) vm_obj['display_name'] = instance_name + '__' + 'vrouter-instance' return True, [vm_obj] def vr_read(vm_id): vr_obj = {} vr_obj['uuid'] = 'fake-vr-uuid' vr_obj['fq_name'] = ['fake-vr-uuid'] return True, [vr_obj] VirtualMachineSM._cassandra = mock.MagicMock() VirtualMachineSM._cassandra._cassandra_virtual_machine_read = vm_read VirtualRouterSM._cassandra = mock.MagicMock() VirtualRouterSM._cassandra._cassandra_virtual_router_read = vr_read self.mocked_vnc = mock.MagicMock() self.mocked_vnc.fq_name_to_id = get_vn_id self.mocked_vnc.virtual_machine_interface_create = vmi_create self.nova_mock = mock.MagicMock() self.mocked_db = mock.MagicMock() self.mocked_args = mock.MagicMock() self.mocked_args.availability_zone = None self.vrouter_manager = VRouterInstanceManager( db=self.mocked_db, logger=mock.MagicMock(), vnc_lib=self.mocked_vnc, vrouter_scheduler=mock.MagicMock(), nova_client=self.nova_mock, args=self.mocked_args)
def setUp(self): mocked_vnc = mock.MagicMock() mocked_vm_ob = mock.MagicMock() mocked_vm_ob.get_virtual_router_back_refs\ .return_value = self.MOCKED_VR_BACK_REF mocked_vm_ob.uuid = self.VM_UUID mocked_vnc.virtual_machine_read.return_value = mocked_vm_ob self.mocked_vm_ob = mocked_vm_ob mocked_db = mock.MagicMock() mocked_db.get_vm_db_prefix.return_value = self.DB_PREFIX self.vrouter_manager = VRouterInstanceManager( db=mocked_db, logger=mock.MagicMock(), vnc_lib=mocked_vnc, vrouter_scheduler=mock.MagicMock())