def setUp(self):
super(TestDetach, self).setUp()
self.cmd = node_network.Detach(self.app, None)
self.node = utils.create_mock_object({
"uuid": "node_uuid_1",
"name": "node1",
"provision_state": "active"
})
self.neutron_port = utils.create_mock_object({
"id":
"neutron_port_uuid_1",
"network_id":
"network_uuid",
"name":
"node1",
"mac_address":
"bb:bb:bb:bb:bb:bb",
"fixed_ips": [{
"ip_address": "2.2.2.2"
}],
"trunk_details":
None
})
self.app.client_manager.baremetal.node.get.\
return_value = self.node
def setUp(self):
super(TestGetFullNetworkInfoFromPort, self).setUp()
self.network1 = test_utils.create_mock_object({
"id":
"network_uuid_1",
"name":
"test_network",
"provider_segmentation_id":
"777"
})
self.network2 = test_utils.create_mock_object({
"id":
"network_uuid_2",
"name":
"test_network_2",
"provider_segmentation_id":
"888"
})
self.subport1 = test_utils.create_mock_object({
"id":
"subport_uuid_1",
"name":
"test_subport_1",
"network_id":
"network_uuid_1",
"fixed_ips": [{
"ip_address": '11.22.33.44'
}]
})
self.subport2 = test_utils.create_mock_object({
"id":
"subport_uuid_2",
"name":
"test_subport_2",
"network_id":
"network_uuid_2",
"fixed_ips": [{
"ip_address": '55.66.77.88'
}]
})
self.neutron_client = mock.Mock()
def mock_network_get(network_uuid):
if network_uuid == "network_uuid_1":
return self.network1
elif network_uuid == "network_uuid_2":
return self.network2
return None
self.neutron_client.get_network.side_effect = mock_network_get
def mock_port_get(port_uuid):
if port_uuid == "subport_uuid_1":
return self.subport1
elif port_uuid == "subport_uuid_2":
return self.subport2
return None
self.neutron_client.get_port.side_effect = mock_port_get
def test_get_full_network_info_from_port(self):
port = test_utils.create_mock_object({
"id":
"port_uuid",
"name":
"test_port",
"network_id":
"network_uuid_1",
"fixed_ips": [{
"ip_address": '77.77.77.77'
}],
"trunk_details": {
"trunk_id":
"trunk_uuid",
"sub_ports": [
{
"segmentation_id": "777",
"port_id": "subport_uuid_1"
},
{
"segmentation_id": "888",
"port_id": "subport_uuid_2"
},
]
}
})
results = utils.get_full_network_info_from_port(
port, self.neutron_client)
self.assertEqual(([
'test_network (777)', 'test_network (777)', 'test_network_2 (888)'
], ['test_port', 'test_subport_1', 'test_subport_2'
], ['77.77.77.77', '11.22.33.44', '55.66.77.88']), results)
def setUp(self):
super(TestDelete, self).setUp()
self.cmd = trunk.Delete(self.app, None)
self.trunk = utils.create_mock_object({
"id":
"trunk_uuid",
"name":
"trunk",
"port_id":
"port_uuid_1",
"sub_ports": [{
"port_id": 'port_uuid_2',
"segmentation_id": '222',
"segmentation_type": 'vlan'
}, {
"port_id": 'port_uuid_3',
"segmentation_id": '333',
"segmentation_type": 'vlan'
}]
})
def mock_find_trunk(trunk_name):
if trunk_name == "trunk":
return self.trunk
return None
self.app.client_manager.network.find_trunk.\
side_effect = mock_find_trunk
self.app.client_manager.network.delete_trunk.\
return_value = None
self.app.client_manager.network.delete_port.\
return_value = None
def test_get_network_display_name_no_segmentation_id(self):
network = test_utils.create_mock_object({
"id": "network_uuid",
"name": "test_network"
})
results = utils.get_network_display_name(network)
self.assertEqual('test_network', results)
def test_get_network_info_from_port_no_fixed_ips(self):
port = test_utils.create_mock_object({
"id": "port_uuid",
"name": "test_port",
"network_id": "network_uuid",
"fixed_ips": None
})
results = utils.get_network_info_from_port(port, self.neutron_client)
self.assertEqual(('test_network (777)', ''), results)
def setUp(self):
super(TestGetNetworkInfoFromPort, self).setUp()
self.network = test_utils.create_mock_object({
"id":
"network_uuid",
"name":
"test_network",
"provider_segmentation_id":
"777"
})
self.neutron_client = mock.Mock()
self.neutron_client.get_network.return_value = self.network
def test_get_network_info_from_port(self):
port = test_utils.create_mock_object({
"id":
"port_uuid",
"name":
"test_port",
"network_id":
"network_uuid",
"fixed_ips": [{
"ip_address": '11.22.33.44'
}]
})
results = utils.get_network_info_from_port(port, self.neutron_client)
self.assertEqual(('test_network (777)', '11.22.33.44'), results)
def test_get_full_network_info_from_port_no_trunk(self):
port = test_utils.create_mock_object({
"id":
"port_uuid",
"name":
"test_port",
"network_id":
"network_uuid_1",
"fixed_ips": [{
"ip_address": '77.77.77.77'
}],
"trunk_details":
None
})
results = utils.get_full_network_info_from_port(
port, self.neutron_client)
self.assertEqual(
(['test_network (777)'], ['test_port'], ['77.77.77.77']), results)
def setUp(self):
super(TestList, self).setUp()
self.cmd = node_network.List(self.app, None)
self.port1 = utils.create_mock_object({
"uuid": "port_uuid_1",
"node_uuid": "node_uuid_1",
"address": "aa:aa:aa:aa:aa:aa",
"internal_info": {
'tenant_vif_port_id': 'neutron_port_uuid_1'
}
})
self.port2 = utils.create_mock_object({
"uuid": "port_uuid_2",
"node_uuid": "node_uuid_2",
"address": "bb:bb:bb:bb:bb:bb",
"internal_info": {}
})
self.port3 = utils.create_mock_object({
"uuid": "port_uuid_3",
"node_uuid": "node_uuid_2",
"address": "cc:cc:cc:cc:cc:cc",
"internal_info": {
'tenant_vif_port_id': 'neutron_port_uuid_2'
}
})
self.node1 = utils.create_mock_object({
"uuid": "node_uuid_1",
"name": "node1"
})
self.node2 = utils.create_mock_object({
"uuid": "node_uuid_2",
"name": "node2"
})
self.network = utils.create_mock_object({
"id": "network_uuid",
"name": "test_network"
})
self.neutron_port1 = utils.create_mock_object({
"id":
"neutron_port_uuid_1",
"network_id":
"network_uuid",
"name":
"node1",
"fixed_ips": [{
"ip_address": "1.1.1.1"
}],
"trunk_details":
None
})
self.neutron_port2 = utils.create_mock_object({
"id":
"neutron_port_uuid_2",
"network_id":
"network_uuid",
"name":
"node2",
"fixed_ips": [{
"ip_address": "2.2.2.2"
}],
"trunk_details":
None
})
def mock_node_get(node_uuid):
if node_uuid == "node_uuid_1":
return self.node1
elif node_uuid == "node_uuid_2":
return self.node2
return None
self.app.client_manager.baremetal.node.get.side_effect = mock_node_get
def mock_neutron_port_get(port_uuid):
if port_uuid == "neutron_port_uuid_1":
return self.neutron_port1
elif port_uuid == "neutron_port_uuid_2":
return self.neutron_port2
return None
self.app.client_manager.network.get_port.\
side_effect = mock_neutron_port_get
self.app.client_manager.network.find_network.\
return_value = self.network
self.app.client_manager.network.get_network.\
return_value = self.network
def setUp(self):
super(TestAttach, self).setUp()
self.cmd = node_network.Attach(self.app, None)
self.port1 = utils.create_mock_object({
"uuid": "port_uuid_1",
"node_uuid": "node_uuid_1",
"address": "aa:aa:aa:aa:aa:aa",
"internal_info": {
'tenant_vif_port_id': 'neutron_port_uuid_1'
}
})
self.port2 = utils.create_mock_object({
"uuid": "port_uuid_2",
"node_uuid": "node_uuid_1",
"address": "bb:bb:bb:bb:bb:bb",
"internal_info": {}
})
self.node = utils.create_mock_object({
"uuid": "node_uuid_1",
"name": "node1",
"provision_state": "active"
})
self.node_available = utils.create_mock_object({
"uuid":
"node_uuid_1",
"name":
"node1",
"provision_state":
"available"
})
self.node_manageable = utils.create_mock_object({
"uuid": "node_uuid_1",
"name": "node1",
"provision_state": "manageable",
"instance_info": {},
"driver_info": {
'deploy_ramdisk': 'fake-image'
},
})
self.node_manageable_instance_info = utils.create_mock_object({
"uuid":
"node_uuid_1",
"name":
"node1",
"provision_state":
"manageable",
"instance_info": {
'image_source': 'fake-image',
'capabilities': {}
},
"driver_info": {
'deploy_ramdisk': 'fake-image'
},
})
self.network = utils.create_mock_object({
"id": "network_uuid",
"name": "test_network"
})
self.neutron_port = utils.create_mock_object({
"id":
"neutron_port_uuid_2",
"network_id":
"network_uuid",
"name":
"node1-port",
"mac_address":
"bb:bb:bb:bb:bb:bb",
"fixed_ips": [{
"ip_address": "2.2.2.2"
}],
"trunk_details":
None
})
self.app.client_manager.network.find_network.\
return_value = self.network
self.app.client_manager.network.get_network.\
return_value = self.network
self.app.client_manager.network.create_port.\
return_value = self.neutron_port
self.app.client_manager.network.find_port.\
return_value = self.neutron_port
self.app.client_manager.network.get_port.\
return_value = self.neutron_port
def setUp(self):
super(TestRemoveNetwork, self).setUp()
self.cmd = trunk.RemoveNetwork(self.app, None)
self.network2 = utils.create_mock_object({
"id":
"network_uuid_2",
"name":
"network2",
"provider_segmentation_id":
222
})
self.network3 = utils.create_mock_object({
"id":
"network_uuid_3",
"name":
"network3",
"provider_segmentation_id":
333
})
self.subport2 = utils.create_mock_object({
"id":
"port_uuid_2",
"network_id":
"network_uuid_2",
"name":
"trunk-network2-sub-port",
"mac_address":
"bb:bb:bb:bb:bb:bb",
})
self.subport3 = utils.create_mock_object({
"id":
"port_uuid_3",
"network_id":
"network_uuid_3",
"name":
"trunk-network3-sub-port",
"mac_address":
"cc:cc:cc:cc:cc:cc",
})
self.trunk = utils.create_mock_object({
"id":
"trunk_uuid",
"name":
"trunk",
"port_id":
"port_uuid_1",
"sub_ports": [{
"port_id": 'port_uuid_2',
"segmentation_id": '222',
"segmentation_type": 'vlan'
}, {
"port_id": 'port_uuid_3',
"segmentation_id": '333',
"segmentation_type": 'vlan'
}]
})
def mock_find_trunk(trunk_name):
if trunk_name == "trunk":
return self.trunk
return None
self.app.client_manager.network.find_trunk.\
side_effect = mock_find_trunk
self.app.client_manager.network.delete_trunk_subports.\
return_value = self.trunk
self.app.client_manager.network.delete_port.\
return_value = None
def setUp(self):
super(TestList, self).setUp()
self.cmd = trunk.List(self.app, None)
self.port1 = utils.create_mock_object({
"id":
"port_uuid_1",
"network_id":
"network_uuid_1",
"name":
"trunk1-network1-trunk-port",
"mac_address":
"aa:aa:aa:aa:aa:aa",
})
self.subport2 = utils.create_mock_object({
"id":
"port_uuid_2",
"network_id":
"network_uuid_2",
"name":
"trunk1-network2-sub-port",
"mac_address":
"bb:bb:bb:bb:bb:bb",
})
self.subport3 = utils.create_mock_object({
"id":
"port_uuid_3",
"network_id":
"network_uuid_3",
"name":
"trunk1-network3-sub-port",
"mac_address":
"cc:cc:cc:cc:cc:cc",
})
self.port2 = utils.create_mock_object({
"id":
"port_uuid_4",
"network_id":
"network_uuid_4",
"name":
"trunk2-network4-trunk-port",
"mac_address":
"dd:dd:dd:dd:dd:dd",
})
self.subport3 = utils.create_mock_object({
"id":
"port_uuid_5",
"network_id":
"network_uuid_5",
"name":
"trunk2-network5-sub-port",
"mac_address":
"ee:ee:ee:ee:ee",
})
self.trunk1 = utils.create_mock_object({
"id":
"trunk_uuid_1",
"name":
"trunk1",
"port_id":
"port_uuid_1",
"sub_ports": [{
"port_id": 'port_uuid_2',
"segmentation_id": '222',
"segmentation_type": 'vlan'
}, {
"port_id": 'port_uuid_3',
"segmentation_id": '333',
"segmentation_type": 'vlan'
}]
})
self.trunk2 = utils.create_mock_object({
"id":
"trunk_uuid_2",
"name":
"trunk2",
"port_id":
"port_uuid_4",
"sub_ports": [{
"port_id": 'port_uuid_5',
"segmentation_id": '555',
"segmentation_type": 'vlan'
}]
})
self.app.client_manager.network.trunks.\
return_value = [self.trunk1, self.trunk2]
def mock_get_port(port_id):
if port_id == "port_uuid_1":
return self.port1
if port_id == "port_uuid_4":
return self.port2
return None
self.app.client_manager.network.get_port.\
side_effect = mock_get_port
def setUp(self):
super(TestCreate, self).setUp()
self.cmd = trunk.Create(self.app, None)
self.network1 = utils.create_mock_object({
"id":
"network_uuid_1",
"name":
"network1",
"provider_segmentation_id":
111
})
self.network2 = utils.create_mock_object({
"id":
"network_uuid_2",
"name":
"network2",
"provider_segmentation_id":
222
})
self.network3 = utils.create_mock_object({
"id":
"network_uuid_3",
"name":
"network3",
"provider_segmentation_id":
333
})
self.port = utils.create_mock_object({
"id":
"port_uuid_1",
"network_id":
"network_uuid_1",
"name":
"trunk-network1-trunk-port",
"mac_address":
"aa:aa:aa:aa:aa:aa",
})
self.subport2 = utils.create_mock_object({
"id":
"port_uuid_2",
"network_id":
"network_uuid_2",
"name":
"trunk-network2-sub-port",
"mac_address":
"bb:bb:bb:bb:bb:bb",
})
self.subport3 = utils.create_mock_object({
"id":
"port_uuid_3",
"network_id":
"network_uuid_3",
"name":
"trunk-network3-sub-port",
"mac_address":
"cc:cc:cc:cc:cc:cc",
})
self.trunk = utils.create_mock_object({
"id":
"trunk_uuid",
"name":
"trunk",
"port_id":
"port_uuid_1",
"sub_ports": [{
"port_id": 'port_uuid_2',
"segmentation_id": '222',
"segmentation_type": 'vlan'
}, {
"port_id": 'port_uuid_3',
"segmentation_id": '333',
"segmentation_type": 'vlan'
}]
})
def mock_find_network(network_name):
if network_name == "network1":
return self.network1
if network_name == "network2":
return self.network2
if network_name == "network3":
return self.network3
return None
self.app.client_manager.network.find_network.\
side_effect = mock_find_network
def mock_create_port(name, network_id):
if network_id == "network_uuid_1":
return self.port
if network_id == "network_uuid_2":
return self.subport2
if network_id == "network_uuid_3":
return self.subport3
return None
self.app.client_manager.network.create_port.\
side_effect = mock_create_port
self.app.client_manager.network.create_trunk.\
return_value = self.trunk