def test_get_portgroup_mor_by_name(self): dvs_name = "test_dvs" port_group_name = fake_api.Constants.PORTGROUP_NAME dvs = fake_api.DataObject() dvs_config = fake_api.DataObject() port_group_mors = [] pg1 = fake_api.create_network() pg1.set("summary.name", "pg1") port_group_mors.append(pg1) pg2 = fake_api.create_network() pg2.set("summary.name", port_group_name) 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_group = network_util.get_portgroup_mor_by_name( self.session, dvs_name, port_group_name) self.assertEqual(port_group.value, pg2.value)
def test_get_portgroup_mor_by_name(self, mock_get_dy_pro, mock_get_dvs_mor, mock_get_pro): dvs_name = "test_dvs" port_group_name = fake_api.Constants.PORTGROUP_NAME dvs = fake_api.DataObject() dvs_config = fake_api.DataObject() port_group_mors = [] pg1 = fake_api.create_network() pg1.set("summary.name", "pg1") port_group_mors.append(pg1) pg2 = fake_api.create_network() pg2.set("summary.name", port_group_name) port_group_mors.append(pg2) dvs_config.ManagedObjectReference = port_group_mors mock_get_pro.return_value = port_group_mors mock_get_dvs_mor.return_value = dvs mock_get_dy_pro.return_value = dvs_config port_group = network_util.get_portgroup_mor_by_name( self.session, dvs_name, port_group_name) self.assertEqual(port_group.value, pg2.value)
def test_get_portgroup_mor_by_names_plus_cluster_id( self, mock_get_dy_pro, mock_get_dvs_mor, mock_get_pro): dvs_name = "test_dvs" network_id = fake_api.Constants.PORTGROUP_NAME cluster_id = "cluster1" dvs = fake_api.DataObject() dvs_config = fake_api.DataObject() port_group_mors = [] pg1 = fake_api.create_network() pg1.set("summary.name", "pg1") port_group_mors.append(pg1) pg2 = fake_api.create_network() pg2.set("summary.name", network_id + "-" + cluster_id) port_group_mors.append(pg2) dvs_config.ManagedObjectReference = port_group_mors mock_get_pro.return_value = port_group_mors mock_get_dvs_mor.return_value = dvs mock_get_dy_pro.return_value = dvs_config port_group = network_util.get_portgroup_mor_by_names( self.session, dvs_name, network_id, cluster_id) self.assertEqual(port_group, network_id + "-" + cluster_id)
def test_get_portgroup_mor_by_names_plus_cluster_id(self, mock_get_dy_pro, mock_get_dvs_mor, mock_get_pro): dvs_name = "test_dvs" network_id = fake_api.Constants.PORTGROUP_NAME cluster_id = "cluster1" dvs = fake_api.DataObject() dvs_config = fake_api.DataObject() port_group_mors = [] pg1 = fake_api.create_network() pg1.set("summary.name", "pg1") port_group_mors.append(pg1) pg2 = fake_api.create_network() pg2.set("summary.name", network_id + "-" + cluster_id) port_group_mors.append(pg2) dvs_config.ManagedObjectReference = port_group_mors mock_get_pro.return_value = port_group_mors mock_get_dvs_mor.return_value = dvs mock_get_dy_pro.return_value = dvs_config port_group = network_util.get_portgroup_mor_by_names( self.session, dvs_name, network_id, cluster_id) self.assertEqual(port_group, network_id + "-" + cluster_id)
def test_get_portgroup_mor_by_net_id_not_found( self, mock_get_dy_pro, mock_get_dvs_mor, mock_get_pro): dvs_name = "test_dvs" port_group_name = fake_api.Constants.PORTGROUP_NAME net_id = "net" dvs = fake_api.DataObject() dvs_config = fake_api.DataObject() port_group_mors = [] pg1 = fake_api.create_network() pg1.set("summary.name", "pg1") port_group_mors.append(pg1) dvs_config.ManagedObjectReference = port_group_mors mock_get_pro.return_value = port_group_mors mock_get_dvs_mor.return_value = dvs mock_get_dy_pro.return_value = dvs_config port_group = network_util.get_portgroup_mor_by_net_id( self.session, dvs_name, port_group_name, net_id) self.assertIsNone(port_group)
def test_get_portgroup_mor_by_net_id_not_found(self, mock_get_dy_pro, mock_get_dvs_mor, mock_get_pro): dvs_name = "test_dvs" port_group_name = fake_api.Constants.PORTGROUP_NAME net_id = "net" dvs = fake_api.DataObject() dvs_config = fake_api.DataObject() port_group_mors = [] pg1 = fake_api.create_network() pg1.set("summary.name", "pg1") port_group_mors.append(pg1) dvs_config.ManagedObjectReference = port_group_mors mock_get_pro.return_value = port_group_mors mock_get_dvs_mor.return_value = dvs mock_get_dy_pro.return_value = dvs_config port_group = network_util.get_portgroup_mor_by_net_id( self.session, dvs_name, port_group_name, net_id) self.assertIsNone(port_group)