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