def test_get_portgroup_and_datacenter_id_by_name(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() folder_mor = fake_vmware_api.create_folder() port_group_mors = [] pg1 = fake_vmware_api.create_network() pg1.set("summary.name", "pg1") pg1.set("parent", folder_mor) port_group_mors.append(pg1) pg2 = fake_vmware_api.create_network() pg2.set("summary.name", port_group_name) pg2.set("parent", folder_mor) port_group_mors.append(pg2) 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_groupid, datacenter_id = network_util.\ get_portgroup_and_datacenter_id_by_name(self.session, dvs_name, port_group_name ) self.assertEqual(port_groupid, pg2.value) self.assertEqual(datacenter_id, datacenter_id)
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)