121 lines
4.6 KiB
Python
121 lines
4.6 KiB
Python
# Copyright 2016 Red Hat, Inc
|
|
# All Rights Reserved.
|
|
#
|
|
# 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 fixtures
|
|
import testtools
|
|
|
|
from neutron.plugins.ml2.drivers.openvswitch.agent import vlanmanager
|
|
from neutron.tests import base
|
|
|
|
|
|
class LocalVlanManagerFixture(fixtures.Fixture):
|
|
def _setUp(self):
|
|
super(LocalVlanManagerFixture, self)._setUp()
|
|
self.vlan_manager = vlanmanager.LocalVlanManager()
|
|
self.addCleanup(self.restore_manager)
|
|
# Remove _instance attribute from VlanManager in order to not obtain a
|
|
# singleton
|
|
del vlanmanager.LocalVlanManager._instance
|
|
self.manager = vlanmanager.LocalVlanManager()
|
|
|
|
def restore_manager(self):
|
|
vlanmanager.LocalVlanManager._instance = self.vlan_manager
|
|
|
|
|
|
class TestLocalVLANMapping(base.BaseTestCase):
|
|
def test___eq___equal(self):
|
|
mapping1 = vlanmanager.LocalVLANMapping(1, 2, 3, 4, 5)
|
|
mapping2 = vlanmanager.LocalVLANMapping(1, 2, 3, 4, 5)
|
|
self.assertEqual(mapping1, mapping2)
|
|
|
|
def test___eq___different(self):
|
|
mapping1 = vlanmanager.LocalVLANMapping(1, 2, 3, 4, 5)
|
|
mapping2 = vlanmanager.LocalVLANMapping(1, 2, 4, 4, 5)
|
|
self.assertNotEqual(mapping1, mapping2)
|
|
|
|
def test___eq___different_type(self):
|
|
mapping = vlanmanager.LocalVLANMapping(1, 2, 3, 4, 5)
|
|
self.assertNotEqual(mapping, "foo")
|
|
|
|
|
|
class TestLocalVlanManager(base.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestLocalVlanManager, self).setUp()
|
|
self.vlan_manager = self.useFixture(LocalVlanManagerFixture()).manager
|
|
|
|
def test_is_singleton(self):
|
|
self.vlan_manager.add(1, None, None, None, None)
|
|
new_vlan_manager = vlanmanager.LocalVlanManager()
|
|
self.assertIs(new_vlan_manager, self.vlan_manager)
|
|
self.assertItemsEqual(new_vlan_manager.mapping,
|
|
self.vlan_manager.mapping)
|
|
|
|
def test_in_operator_on_key(self):
|
|
self.vlan_manager.add(1, None, None, None, None)
|
|
self.assertIn(1, self.vlan_manager)
|
|
self.assertNotIn(2, self.vlan_manager)
|
|
|
|
def test_iterator_returns_vlan_mappings(self):
|
|
created_vlans = []
|
|
for val in range(3):
|
|
self.vlan_manager.add(val, val, val, val, val)
|
|
created_vlans.append(self.vlan_manager.get(val))
|
|
|
|
self.assertItemsEqual(created_vlans, list(self.vlan_manager))
|
|
|
|
def test_get_net_uuid_existing(self):
|
|
port_id = 'port-id'
|
|
vlan_data = (2, 3, 4, 5, {port_id: 'port'})
|
|
net_id = 1
|
|
self.vlan_manager.add(net_id, *vlan_data)
|
|
obtained_net_id = self.vlan_manager.get_net_uuid(port_id)
|
|
self.assertEqual(net_id, obtained_net_id)
|
|
|
|
def test_get_net_uuid_non_existing_raises_exception(self):
|
|
vlan_data = (1, 2, 3, 4, 5, {'port_id': 'port'})
|
|
self.vlan_manager.add(*vlan_data)
|
|
with testtools.ExpectedException(vlanmanager.VifIdNotFound):
|
|
self.vlan_manager.get_net_uuid('non-existing-port')
|
|
|
|
def test_add_and_get(self):
|
|
vlan_data = (2, 3, 4, 5, 6)
|
|
expected_vlan_mapping = vlanmanager.LocalVLANMapping(*vlan_data)
|
|
self.vlan_manager.add(1, *vlan_data)
|
|
vlan_mapping = self.vlan_manager.get(1)
|
|
self.assertEqual(expected_vlan_mapping, vlan_mapping)
|
|
|
|
def test_add_existing_raises_exception(self):
|
|
vlan_data = (2, 3, 4, 5, 6)
|
|
self.vlan_manager.add(1, *vlan_data)
|
|
with testtools.ExpectedException(vlanmanager.MappingAlreadyExists):
|
|
self.vlan_manager.add(1, *vlan_data)
|
|
|
|
def test_get_non_existing_raises_keyerror(self):
|
|
with testtools.ExpectedException(vlanmanager.MappingNotFound):
|
|
self.vlan_manager.get(1)
|
|
|
|
def test_pop(self):
|
|
vlan_data = (2, 3, 4, 5, 6)
|
|
expected_vlan_mapping = vlanmanager.LocalVLANMapping(*vlan_data)
|
|
self.vlan_manager.add(1, *vlan_data)
|
|
vlan_mapping = self.vlan_manager.pop(1)
|
|
self.assertEqual(expected_vlan_mapping, vlan_mapping)
|
|
self.assertFalse(self.vlan_manager.mapping)
|
|
|
|
def test_pop_non_existing_raises_exception(self):
|
|
with testtools.ExpectedException(vlanmanager.MappingNotFound):
|
|
self.vlan_manager.pop(1)
|