ae1d6d1b70
VPNaaS and ML2 plugin won't work, because ML2 plugin supports service version of L3. In this commit, we modify ipsec driver to use L3 plugin. This is also backward compatible change, because if L2 plugin supports L3 get_service_plugin API returns L2 plugin. Fixes bug 1238846 Change-Id: I36e541bb2e3e1df2e01f73a74f3e9005af6c38b7 Note: Exsiting unit test covers this change
93 lines
3.3 KiB
Python
93 lines
3.3 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
#
|
|
# Copyright 2013, Nachi Ueno, NTT I3, 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 mock
|
|
|
|
from neutron import context
|
|
from neutron.openstack.common import uuidutils
|
|
from neutron.plugins.common import constants
|
|
from neutron.services.vpn.service_drivers import ipsec as ipsec_driver
|
|
from neutron.tests import base
|
|
|
|
_uuid = uuidutils.generate_uuid
|
|
|
|
FAKE_VPN_CONNECTION = {
|
|
'vpnservice_id': _uuid()
|
|
}
|
|
FAKE_VPN_SERVICE = {
|
|
'router_id': _uuid()
|
|
}
|
|
FAKE_HOST = 'fake_host'
|
|
|
|
|
|
class TestIPsecDriver(base.BaseTestCase):
|
|
def setUp(self):
|
|
super(TestIPsecDriver, self).setUp()
|
|
self.addCleanup(mock.patch.stopall)
|
|
mock.patch('neutron.openstack.common.rpc.create_connection').start()
|
|
|
|
l3_agent = mock.Mock()
|
|
l3_agent.host = FAKE_HOST
|
|
plugin = mock.Mock()
|
|
plugin.get_l3_agents_hosting_routers.return_value = [l3_agent]
|
|
plugin_p = mock.patch('neutron.manager.NeutronManager.get_plugin')
|
|
get_plugin = plugin_p.start()
|
|
get_plugin.return_value = plugin
|
|
service_plugin_p = mock.patch(
|
|
'neutron.manager.NeutronManager.get_service_plugins')
|
|
get_service_plugin = service_plugin_p.start()
|
|
get_service_plugin.return_value = {constants.L3_ROUTER_NAT: plugin}
|
|
|
|
service_plugin = mock.Mock()
|
|
service_plugin.get_l3_agents_hosting_routers.return_value = [l3_agent]
|
|
service_plugin._get_vpnservice.return_value = {
|
|
'router_id': _uuid()
|
|
}
|
|
self.driver = ipsec_driver.IPsecVPNDriver(service_plugin)
|
|
|
|
def _test_update(self, func, args):
|
|
ctxt = context.Context('', 'somebody')
|
|
with mock.patch.object(self.driver.agent_rpc, 'cast') as cast:
|
|
func(ctxt, *args)
|
|
cast.assert_called_once_with(
|
|
ctxt,
|
|
{'args': {},
|
|
'namespace': None,
|
|
'method': 'vpnservice_updated'},
|
|
version='1.0',
|
|
topic='ipsec_agent.fake_host')
|
|
|
|
def test_create_ipsec_site_connection(self):
|
|
self._test_update(self.driver.create_ipsec_site_connection,
|
|
[FAKE_VPN_CONNECTION])
|
|
|
|
def test_update_ipsec_site_connection(self):
|
|
self._test_update(self.driver.update_ipsec_site_connection,
|
|
[FAKE_VPN_CONNECTION, FAKE_VPN_CONNECTION])
|
|
|
|
def test_delete_ipsec_site_connection(self):
|
|
self._test_update(self.driver.delete_ipsec_site_connection,
|
|
[FAKE_VPN_CONNECTION])
|
|
|
|
def test_update_vpnservice(self):
|
|
self._test_update(self.driver.update_vpnservice,
|
|
[FAKE_VPN_SERVICE, FAKE_VPN_SERVICE])
|
|
|
|
def test_delete_vpnservice(self):
|
|
self._test_update(self.driver.delete_vpnservice,
|
|
[FAKE_VPN_SERVICE])
|