319 lines
11 KiB
Python
319 lines
11 KiB
Python
# 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.
|
|
|
|
from oslo_utils import uuidutils
|
|
import testscenarios
|
|
|
|
from neutron.db.models import securitygroup as sg_models
|
|
from neutron.objects import base as obj_base
|
|
from neutron.objects.db import api as obj_db_api
|
|
from neutron.objects import network
|
|
from neutron.objects import ports
|
|
from neutron.objects.qos import policy
|
|
from neutron.tests import tools
|
|
from neutron.tests.unit.objects import test_base as obj_test_base
|
|
from neutron.tests.unit import testlib_api
|
|
|
|
|
|
class BasePortBindingDbObjectTestCase(obj_test_base._BaseObjectTestCase,
|
|
testlib_api.SqlTestCase):
|
|
def setUp(self):
|
|
super(BasePortBindingDbObjectTestCase, self).setUp()
|
|
self._create_test_network()
|
|
getter = lambda: self._create_port(network_id=self._network['id']).id
|
|
self.update_obj_fields({'port_id': getter})
|
|
|
|
|
|
class PortBindingIfaceObjTestCase(obj_test_base.BaseObjectIfaceTestCase):
|
|
_test_class = ports.PortBinding
|
|
|
|
|
|
class PortBindingDbObjectTestCase(obj_test_base.BaseDbObjectTestCase,
|
|
BasePortBindingDbObjectTestCase):
|
|
_test_class = ports.PortBinding
|
|
|
|
|
|
class DistributedPortBindingIfaceObjTestCase(
|
|
obj_test_base.BaseObjectIfaceTestCase):
|
|
_test_class = ports.DistributedPortBinding
|
|
|
|
|
|
class DistributedPortBindingDbObjectTestCase(
|
|
obj_test_base.BaseDbObjectTestCase,
|
|
BasePortBindingDbObjectTestCase):
|
|
_test_class = ports.DistributedPortBinding
|
|
|
|
|
|
# TODO(ihrachys): this test case copies some functions from the base module.
|
|
# This is because we currently cannot inherit from the base class that contains
|
|
# those functions, because that same class provides test cases that we don't
|
|
# want to execute. Ideally, we would need to copy paste, but that would require
|
|
# some significant refactoring in the base test classes. Leaving it for a
|
|
# follow up.
|
|
class PortBindingVifDetailsTestCase(testscenarios.WithScenarios,
|
|
obj_test_base._BaseObjectTestCase,
|
|
testlib_api.SqlTestCase):
|
|
|
|
scenarios = [
|
|
(cls.__name__, {'_test_class': cls})
|
|
for cls in (ports.PortBinding, ports.DistributedPortBinding)
|
|
]
|
|
|
|
def setUp(self):
|
|
super(PortBindingVifDetailsTestCase, self).setUp()
|
|
self._create_test_network()
|
|
getter = lambda: self._create_port(network_id=self._network['id']).id
|
|
self.update_obj_fields({'port_id': getter})
|
|
|
|
def _create_port(self, **port_attrs):
|
|
attrs = {'project_id': uuidutils.generate_uuid(),
|
|
'admin_state_up': True,
|
|
'status': 'ACTIVE',
|
|
'device_id': 'fake_device',
|
|
'device_owner': 'fake_owner',
|
|
'mac_address': tools.get_random_EUI()}
|
|
attrs.update(port_attrs)
|
|
port = ports.Port(self.context, **attrs)
|
|
port.create()
|
|
return port
|
|
|
|
def _create_test_network(self):
|
|
self._network = network.Network(self.context,
|
|
name='test-network1')
|
|
self._network.create()
|
|
|
|
def _make_object(self, fields):
|
|
fields = obj_test_base.get_non_synthetic_fields(
|
|
self._test_class, fields
|
|
)
|
|
return self._test_class(
|
|
self.context,
|
|
**obj_test_base.remove_timestamps_from_fields(
|
|
fields, self._test_class.fields))
|
|
|
|
def test_vif_details(self):
|
|
vif_details = {'item1': 'val1', 'item2': 'val2'}
|
|
obj = self._make_object(self.obj_fields[0])
|
|
obj.vif_details = vif_details
|
|
obj.create()
|
|
|
|
obj = self._test_class.get_object(
|
|
self.context, **obj._get_composite_keys())
|
|
self.assertEqual(vif_details, obj.vif_details)
|
|
|
|
vif_details['item1'] = 'val2'
|
|
del vif_details['item2']
|
|
vif_details['item3'] = 'val3'
|
|
|
|
obj.vif_details = vif_details
|
|
obj.update()
|
|
|
|
obj = self._test_class.get_object(
|
|
self.context, **obj._get_composite_keys())
|
|
self.assertEqual(vif_details, obj.vif_details)
|
|
|
|
obj.vif_details = None
|
|
obj.update()
|
|
|
|
obj = self._test_class.get_object(
|
|
self.context, **obj._get_composite_keys())
|
|
self.assertIsNone(obj.vif_details)
|
|
|
|
def test_null_vif_details_in_db(self):
|
|
# the null case for vif_details in our db model is an
|
|
# empty string. add that here to simulate it correctly
|
|
# in the tests
|
|
kwargs = self.get_random_fields()
|
|
kwargs['vif_details'] = ''
|
|
db_obj = self._test_class.db_model(**kwargs)
|
|
obj_fields = self._test_class.modify_fields_from_db(db_obj)
|
|
obj = self._test_class(self.context, **obj_fields)
|
|
self.assertIsNone(obj.vif_details)
|
|
|
|
|
|
class IPAllocationIfaceObjTestCase(obj_test_base.BaseObjectIfaceTestCase):
|
|
|
|
_test_class = ports.IPAllocation
|
|
|
|
|
|
class IPAllocationDbObjectTestCase(obj_test_base.BaseDbObjectTestCase,
|
|
testlib_api.SqlTestCase):
|
|
|
|
_test_class = ports.IPAllocation
|
|
|
|
def setUp(self):
|
|
super(IPAllocationDbObjectTestCase, self).setUp()
|
|
self._create_test_network()
|
|
self._create_test_subnet(self._network)
|
|
self._create_test_port(self._network)
|
|
self.update_obj_fields({'port_id': self._port.id,
|
|
'network_id': self._network.id,
|
|
'subnet_id': self._subnet.id})
|
|
|
|
|
|
class PortDNSIfaceObjTestCase(obj_test_base.BaseObjectIfaceTestCase):
|
|
|
|
_test_class = ports.PortDNS
|
|
|
|
|
|
class PortDNSDbObjectTestCase(obj_test_base.BaseDbObjectTestCase,
|
|
testlib_api.SqlTestCase):
|
|
|
|
_test_class = ports.PortDNS
|
|
|
|
def setUp(self):
|
|
super(PortDNSDbObjectTestCase, self).setUp()
|
|
self._create_test_network()
|
|
getter = lambda: self._create_port(network_id=self._network['id']).id
|
|
self.update_obj_fields({'port_id': getter})
|
|
|
|
|
|
class PortBindingLevelIfaceObjTestCase(
|
|
obj_test_base.BaseObjectIfaceTestCase):
|
|
|
|
_test_class = ports.PortBindingLevel
|
|
|
|
def setUp(self):
|
|
super(PortBindingLevelIfaceObjTestCase, self).setUp()
|
|
# for this object, the model contains segment_id but we expose it
|
|
# through an ObjectField that is loaded without a relationship
|
|
for obj in self.db_objs:
|
|
obj['segment_id'] = None
|
|
self.pager_map[self._test_class.obj_name()] = (
|
|
obj_base.Pager(sorts=[('port_id', True), ('level', True)]))
|
|
self.pager_map[network.NetworkSegment.obj_name()] = (
|
|
obj_base.Pager(
|
|
sorts=[('network_id', True), ('segment_index', True)]))
|
|
|
|
|
|
class PortBindingLevelDbObjectTestCase(
|
|
obj_test_base.BaseDbObjectTestCase):
|
|
|
|
_test_class = ports.PortBindingLevel
|
|
|
|
|
|
class PortIfaceObjTestCase(obj_test_base.BaseObjectIfaceTestCase):
|
|
|
|
_test_class = ports.Port
|
|
|
|
def setUp(self):
|
|
super(PortIfaceObjTestCase, self).setUp()
|
|
self.pager_map[ports.PortBindingLevel.obj_name()] = (
|
|
obj_base.Pager(sorts=[('port_id', True), ('level', True)]))
|
|
|
|
|
|
class PortDbObjectTestCase(obj_test_base.BaseDbObjectTestCase,
|
|
testlib_api.SqlTestCase):
|
|
|
|
_test_class = ports.Port
|
|
|
|
def _create_test_security_group(self):
|
|
return obj_db_api.create_object(
|
|
self.context, sg_models.SecurityGroup, {})
|
|
|
|
def setUp(self):
|
|
super(PortDbObjectTestCase, self).setUp()
|
|
self._create_test_network()
|
|
self._create_test_subnet(self._network)
|
|
self.update_obj_fields(
|
|
{'network_id': self._network.id,
|
|
'fixed_ips': {'subnet_id': self._subnet.id,
|
|
'network_id': self._network['id']}})
|
|
|
|
def test_security_group_ids(self):
|
|
sg1 = self._create_test_security_group()
|
|
sg2 = self._create_test_security_group()
|
|
groups = {sg1.id, sg2.id}
|
|
obj = self._make_object(self.obj_fields[0])
|
|
obj.security_group_ids = groups
|
|
obj.create()
|
|
|
|
obj = ports.Port.get_object(self.context, id=obj.id)
|
|
self.assertEqual(groups, obj.security_group_ids)
|
|
|
|
sg3 = self._create_test_security_group()
|
|
obj.security_group_ids = {sg3.id}
|
|
obj.update()
|
|
|
|
obj = ports.Port.get_object(self.context, id=obj.id)
|
|
self.assertEqual({sg3.id}, obj.security_group_ids)
|
|
|
|
obj.security_group_ids = set()
|
|
obj.update()
|
|
|
|
obj = ports.Port.get_object(self.context, id=obj.id)
|
|
self.assertFalse(obj.security_group_ids)
|
|
|
|
def test__attach_security_group(self):
|
|
obj = self._make_object(self.obj_fields[0])
|
|
obj.create()
|
|
|
|
sg = self._create_test_security_group()
|
|
obj._attach_security_group(sg.id)
|
|
|
|
obj = ports.Port.get_object(self.context, id=obj.id)
|
|
self.assertIn(sg.id, obj.security_group_ids)
|
|
|
|
sg2 = self._create_test_security_group()
|
|
obj._attach_security_group(sg2.id)
|
|
|
|
obj = ports.Port.get_object(self.context, id=obj.id)
|
|
self.assertIn(sg2.id, obj.security_group_ids)
|
|
|
|
def test_qos_policy_id(self):
|
|
policy_obj = policy.QosPolicy(self.context)
|
|
policy_obj.create()
|
|
|
|
obj = self._make_object(self.obj_fields[0])
|
|
obj.qos_policy_id = policy_obj.id
|
|
obj.create()
|
|
|
|
obj = ports.Port.get_object(self.context, id=obj.id)
|
|
self.assertEqual(policy_obj.id, obj.qos_policy_id)
|
|
|
|
policy_obj2 = policy.QosPolicy(self.context)
|
|
policy_obj2.create()
|
|
|
|
obj.qos_policy_id = policy_obj2.id
|
|
obj.update()
|
|
|
|
obj = ports.Port.get_object(self.context, id=obj.id)
|
|
self.assertEqual(policy_obj2.id, obj.qos_policy_id)
|
|
|
|
obj.qos_policy_id = None
|
|
obj.update()
|
|
|
|
obj = ports.Port.get_object(self.context, id=obj.id)
|
|
self.assertIsNone(obj.qos_policy_id)
|
|
|
|
def test__attach_qos_policy(self):
|
|
obj = self._make_object(self.obj_fields[0])
|
|
obj.create()
|
|
|
|
policy_obj = policy.QosPolicy(self.context)
|
|
policy_obj.create()
|
|
obj._attach_qos_policy(policy_obj.id)
|
|
|
|
obj = ports.Port.get_object(self.context, id=obj.id)
|
|
self.assertEqual(policy_obj.id, obj.qos_policy_id)
|
|
|
|
policy_obj2 = policy.QosPolicy(self.context)
|
|
policy_obj2.create()
|
|
obj._attach_qos_policy(policy_obj2.id)
|
|
|
|
obj = ports.Port.get_object(self.context, id=obj.id)
|
|
self.assertEqual(policy_obj2.id, obj.qos_policy_id)
|
|
|
|
def test_get_objects_queries_constant(self):
|
|
self.skipTest(
|
|
'Port object loads segment info without relationships')
|