14e779bbac
Now that we no longer support py27, we can use the standard library unittest.mock module instead of the third party mock lib. Change-Id: I2de669d8e89b8daeb7ee5405ffab35af6307c40b
145 lines
6.4 KiB
Python
145 lines
6.4 KiB
Python
# Copyright 2013 NEC Corporation
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import collections
|
|
from unittest import mock
|
|
|
|
import netaddr
|
|
|
|
from django.test.utils import override_settings
|
|
|
|
from openstack_dashboard import api
|
|
from openstack_dashboard.test import helpers as test
|
|
|
|
|
|
class NetworkApiNeutronTests(test.APIMockTestCase):
|
|
def setUp(self):
|
|
super(NetworkApiNeutronTests, self).setUp()
|
|
neutronclient = mock.patch.object(api.neutron, 'neutronclient').start()
|
|
self.qclient = neutronclient.return_value
|
|
|
|
def _get_expected_addresses(self, server, no_fip_expected=True):
|
|
server_ports = self.ports.filter(device_id=server.id)
|
|
addresses = collections.defaultdict(list)
|
|
for p in server_ports:
|
|
net_name = self.networks.get(id=p['network_id']).name
|
|
for ip in p.fixed_ips:
|
|
version = netaddr.IPAddress(ip['ip_address']).version
|
|
addresses[net_name].append(
|
|
{'version': version,
|
|
'addr': ip['ip_address'],
|
|
'OS-EXT-IPS-MAC:mac_addr': p.mac_address,
|
|
'OS-EXT-IPS:type': 'fixed'})
|
|
if no_fip_expected:
|
|
continue
|
|
fips = self.floating_ips.filter(port_id=p['id'])
|
|
if not fips:
|
|
continue
|
|
# Only one FIP should match.
|
|
fip = fips[0]
|
|
addresses[net_name].append(
|
|
{'version': 4,
|
|
'addr': fip.floating_ip_address,
|
|
'OS-EXT-IPS-MAC:mac_addr': p.mac_address,
|
|
'OS-EXT-IPS:type': 'floating'})
|
|
return addresses
|
|
|
|
def _check_server_address(self, res_server_data, no_fip_expected=False):
|
|
expected_addresses = self._get_expected_addresses(res_server_data,
|
|
no_fip_expected)
|
|
self.assertEqual(len(expected_addresses),
|
|
len(res_server_data.addresses))
|
|
for net, addresses in expected_addresses.items():
|
|
self.assertIn(net, res_server_data.addresses)
|
|
self.assertEqual(addresses, res_server_data.addresses[net])
|
|
|
|
def _test_servers_update_addresses(self, router_enabled=True):
|
|
tenant_id = self.request.user.tenant_id
|
|
|
|
servers = self.servers.list()
|
|
server_ids = tuple([server.id for server in servers])
|
|
server_ports = [p for p in self.api_ports.list()
|
|
if p['device_id'] in server_ids]
|
|
server_port_ids = tuple([p['id'] for p in server_ports])
|
|
if router_enabled:
|
|
assoc_fips = [fip for fip in self.api_floating_ips.list()
|
|
if fip['port_id'] in server_port_ids]
|
|
server_network_ids = [p['network_id'] for p in server_ports]
|
|
server_networks = [net for net in self.api_networks.list()
|
|
if net['id'] in server_network_ids]
|
|
|
|
list_ports_retvals = [{'ports': server_ports}]
|
|
self.qclient.list_ports.side_effect = list_ports_retvals
|
|
if router_enabled:
|
|
self.qclient.list_floatingips.return_value = {'floatingips':
|
|
assoc_fips}
|
|
list_ports_retvals.append({'ports': self.api_ports.list()})
|
|
self.qclient.list_networks.return_value = {'networks': server_networks}
|
|
self.qclient.list_subnets.return_value = {'subnets':
|
|
self.api_subnets.list()}
|
|
|
|
api.network.servers_update_addresses(self.request, servers)
|
|
|
|
self.assertEqual(self.servers.count(), len(servers))
|
|
self.assertEqual([server.id for server in self.servers.list()],
|
|
[server.id for server in servers])
|
|
|
|
no_fip_expected = not router_enabled
|
|
|
|
# server[0] has one fixed IP and one floating IP
|
|
# if router ext isenabled.
|
|
self._check_server_address(servers[0], no_fip_expected)
|
|
# The expected is also calculated, we examine the result manually once.
|
|
addrs = servers[0].addresses['net1']
|
|
if router_enabled:
|
|
self.assertEqual(3, len(addrs))
|
|
self.assertEqual('fixed', addrs[0]['OS-EXT-IPS:type'])
|
|
self.assertEqual('fixed', addrs[1]['OS-EXT-IPS:type'])
|
|
self.assertEqual('floating', addrs[2]['OS-EXT-IPS:type'])
|
|
else:
|
|
self.assertEqual(2, len(addrs))
|
|
self.assertEqual('fixed', addrs[0]['OS-EXT-IPS:type'])
|
|
self.assertEqual('fixed', addrs[1]['OS-EXT-IPS:type'])
|
|
|
|
# server[1] has one fixed IP.
|
|
self._check_server_address(servers[1], no_fip_expected)
|
|
# manual check.
|
|
addrs = servers[1].addresses['net2']
|
|
self.assertEqual(1, len(addrs))
|
|
self.assertEqual('fixed', addrs[0]['OS-EXT-IPS:type'])
|
|
|
|
# server[2] has no corresponding ports in neutron_data,
|
|
# so it should be an empty dict.
|
|
self.assertFalse(servers[2].addresses)
|
|
|
|
expected_list_ports = [mock.call(device_id=server_ids)]
|
|
if router_enabled:
|
|
self.qclient.list_floatingips.assert_called_once_with(
|
|
tenant_id=tenant_id, port_id=server_port_ids)
|
|
expected_list_ports.append(mock.call(tenant_id=tenant_id))
|
|
else:
|
|
self.assertEqual(0, self.qclient.list_floatingips.call_count)
|
|
self.qclient.list_ports.assert_has_calls(expected_list_ports)
|
|
self.qclient.list_networks.assert_called_once_with(
|
|
id=frozenset(server_network_ids))
|
|
self.qclient.list_subnets.assert_called_once_with()
|
|
|
|
@override_settings(OPENSTACK_NEUTRON_NETWORK={'enable_router': True})
|
|
def test_servers_update_addresses(self):
|
|
self._test_servers_update_addresses()
|
|
|
|
@override_settings(OPENSTACK_NEUTRON_NETWORK={'enable_router': False})
|
|
def test_servers_update_addresses_router_disabled(self):
|
|
self._test_servers_update_addresses(router_enabled=False)
|