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