def post_create_port(self, port):
if port.port_status == model.PortStatus.UP:
device_id = port.vm_id
_clu_mor, _clu_path, vds_name = self.\
_find_cluster_switch_for_vm(device_id)
pg_mor = network_util.get_portgroup_mor_by_name(
self.session,
vds_name,
port.network_uuid)
if pg_mor is None:
raise error_util.RunTimeError("Port group %s not created "
"on virtual switch %s"
% (port.network_uuid, vds_name))
vm_mor = resource_util.get_vm_mor_for_uuid(self.session,
device_id)
if vm_mor is None:
raise error_util.RunTimeError("Virtual machine %s with port "
" %s not created "
% (device_id, port.uuid))
(pg_key, port_key, swuuid) = self.\
_wait_for_port_update_on_vm(vm_mor, pg_mor)
if None not in (pg_key, port_key, swuuid):
# enable the port on virtual switch
network_util.enable_disable_port(self.session, swuuid,
pg_key, port_key, True)
def test_get_portgroup_mor_by_name_no_dvs(self):
dvs_name = "non_existent_dvs"
port_group_name = fake_vmware_api.Constants.PORTGROUP_NAME
with mock.patch.object(network_util, "get_dvs_mor_by_name",
return_value=None):
port_group = network_util.get_portgroup_mor_by_name(
self.session, dvs_name, port_group_name)
self.assertIsNone(port_group)
def test_get_portgroup_mor_by_name_not_found(self):
dvs_name = "test_dvs"
port_group_name = fake_vmware_api.Constants.PORTGROUP_NAME
dvs = fake_vmware_api.DataObject()
dvs_config = fake_vmware_api.DataObject()
port_group_mors = []
pg1 = fake_vmware_api.create_network()
pg1.set("summary.name", "pg1")
port_group_mors.append(pg1)
dvs_config.ManagedObjectReference = port_group_mors
with contextlib.nested(
mock.patch.object(vim_util,
'get_properties_for_a_collection_of_objects',
return_value=port_group_mors),
mock.patch.object(network_util, "get_dvs_mor_by_name",
return_value=dvs),
mock.patch.object(vim_util, "get_dynamic_property",
return_value=dvs_config)):
port_group = network_util.get_portgroup_mor_by_name(
self.session, dvs_name, port_group_name)
self.assertIsNone(port_group)
