40 lines
1.6 KiB
Python
40 lines
1.6 KiB
Python
# Copyright (c) 2015 OpenStack Foundation.
|
|
# 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 mock
|
|
from neutron_lib.api.definitions import port_security as psec
|
|
|
|
from neutron.plugins.ml2.extensions import port_security
|
|
from neutron.tests.unit.plugins.ml2 import test_plugin
|
|
|
|
|
|
class TestML2ExtensionPortSecurity(test_plugin.Ml2PluginV2TestCase):
|
|
def _test_extend_dict_no_port_security(self, func):
|
|
"""Test extend_*_dict won't crash if port_security item is None."""
|
|
for db_data in ({'port_security': None, 'name': 'net1'}, {}):
|
|
response_data = {}
|
|
session = mock.Mock()
|
|
|
|
driver = port_security.PortSecurityExtensionDriver()
|
|
getattr(driver, func)(session, db_data, response_data)
|
|
|
|
self.assertTrue(response_data[psec.PORTSECURITY])
|
|
|
|
def test_extend_port_dict_no_port_security(self):
|
|
self._test_extend_dict_no_port_security('extend_port_dict')
|
|
|
|
def test_extend_network_dict_no_port_security(self):
|
|
self._test_extend_dict_no_port_security('extend_network_dict')
|