def setUp(self):
VirtualMachineSM._cassandra = mock.MagicMock()
VirtualMachineSM._cassandra.object_read = test_utils.vm_db_read
VirtualMachineInterfaceSM._cassandra = mock.MagicMock()
VirtualMachineInterfaceSM._cassandra.object_read = test_utils.vmi_db_read
InstanceIpSM._cassandra = mock.MagicMock()
InstanceIpSM._cassandra.object_read = test_utils.iip_db_read
InterfaceRouteTableSM._cassandra = mock.MagicMock()
InterfaceRouteTableSM._cassandra.object_read = test_utils.irt_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_network_create = test_utils.vn_create
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 = 'default-availability-zone'
self.log_mock = mock.MagicMock()
self.vm_manager = VirtualMachineManager(
db=self.mocked_db,
logger=self.log_mock,
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 vn_create(vn_obj):
vn_obj.uuid = (':').join(vn_obj.fq_name)
vn = {}
vn['uuid'] = vn_obj.uuid
vn['fq_name'] = vn_obj.fq_name
VirtualNetworkSM.locate(vn_obj.uuid, vn)
return
def vmi_create(vmi_obj):
vmi_obj.uuid = 'fake-vmi-uuid'
return
self.mocked_vnc = mock.MagicMock()
self.mocked_vnc.fq_name_to_id = get_vn_id
self.mocked_vnc.virtual_network_create = vn_create
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.vm_manager = VirtualMachineManager(
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)
