David Ames
70 lines
2.7 KiB
Python
70 lines
2.7 KiB
Python
import mock
|
|
import unit_tests.utils as ut_utils
|
|
from zaza.utilities import openstack_utils
|
|
|
|
|
|
class TestOpenStackUtils(ut_utils.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestOpenStackUtils, self).setUp()
|
|
self.port_name = 'port_name'
|
|
self.net_uuid = 'uuid'
|
|
self.port = {
|
|
'port': {'id': 'port_id',
|
|
'name': self.port_name,
|
|
'network_id': self.net_uuid}}
|
|
|
|
self.ports = {'ports': [self.port['port']]}
|
|
self.floatingip = {
|
|
'floatingip': {'id': 'floatingip_id',
|
|
'floating_network_id': self.net_uuid,
|
|
'port_id': 'port_id'}}
|
|
|
|
self.floatingips = {'floatingips': [self.floatingip['floatingip']]}
|
|
|
|
self.neutronclient = mock.MagicMock()
|
|
self.neutronclient.list_ports.return_value = self.ports
|
|
self.neutronclient.create_port.return_value = self.port
|
|
|
|
self.neutronclient.list_floatingips.return_value = self.floatingips
|
|
self.neutronclient.create_floatingip.return_value = self.floatingip
|
|
self.ext_net = 'ext_net'
|
|
self.private_net = 'private_net'
|
|
|
|
def test_create_port(self):
|
|
self.patch_object(openstack_utils, 'get_net_uuid')
|
|
self.get_net_uuid.return_value = self.net_uuid
|
|
|
|
# Already exists
|
|
port = openstack_utils.create_port(
|
|
self.neutronclient, self.port_name, self.private_net)
|
|
self.assertEqual(port, self.port['port'])
|
|
self.neutronclient.create_port.assert_not_called()
|
|
|
|
# Does not yet exist
|
|
self.neutronclient.list_ports.return_value = {'ports': []}
|
|
self.port['port'].pop('id')
|
|
port = openstack_utils.create_port(
|
|
self.neutronclient, self.port_name, self.private_net)
|
|
self.assertEqual(port, self.port['port'])
|
|
self.neutronclient.create_port.assert_called_once_with(self.port)
|
|
|
|
def test_create_floating_ip(self):
|
|
self.patch_object(openstack_utils, 'get_net_uuid')
|
|
self.get_net_uuid.return_value = self.net_uuid
|
|
|
|
# Already exists
|
|
floatingip = openstack_utils.create_floating_ip(
|
|
self.neutronclient, self.ext_net, port=self.port['port'])
|
|
self.assertEqual(floatingip, self.floatingip['floatingip'])
|
|
self.neutronclient.create_floatingip.assert_not_called()
|
|
|
|
# Does not yet exist
|
|
self.neutronclient.list_floatingips.return_value = {'floatingips': []}
|
|
self.floatingip['floatingip'].pop('id')
|
|
floatingip = openstack_utils.create_floating_ip(
|
|
self.neutronclient, self.private_net, port=self.port['port'])
|
|
self.assertEqual(floatingip, self.floatingip['floatingip'])
|
|
self.neutronclient.create_floatingip.assert_called_once_with(
|
|
self.floatingip)
|