6bdfccaf1b
This patch add the type_driver GRE to enable the creation of GRE tunnels with the OVS agent. No Endpoints ID are managed. Only Endpoint IP are stored in DB, and this IP is proposed as an endpoint ID for the OVS agent. It also fixes the bug 1201471. Implements: blueprint ml2-gre Change-Id: I1a33a4bd3ebc4c97eecf17a59ce16b8c2066ec66
177 lines
6.9 KiB
Python
177 lines
6.9 KiB
Python
# Copyright (c) 2013 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 testtools
|
|
from testtools import matchers
|
|
|
|
from neutron.common import exceptions as exc
|
|
import neutron.db.api as db
|
|
from neutron.plugins.ml2 import db as ml2_db
|
|
from neutron.plugins.ml2 import driver_api as api
|
|
from neutron.plugins.ml2.drivers import type_gre
|
|
from neutron.tests import base
|
|
|
|
TUNNEL_IP_ONE = "10.10.10.10"
|
|
TUNNEL_IP_TWO = "10.10.10.20"
|
|
TUN_MIN = 100
|
|
TUN_MAX = 109
|
|
TUNNEL_RANGES = [(TUN_MIN, TUN_MAX)]
|
|
UPDATED_TUNNEL_RANGES = [(TUN_MIN + 5, TUN_MAX + 5)]
|
|
|
|
|
|
class GreTypeTest(base.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(GreTypeTest, self).setUp()
|
|
ml2_db.initialize()
|
|
self.driver = type_gre.GreTypeDriver()
|
|
self.driver.gre_id_ranges = TUNNEL_RANGES
|
|
self.driver._sync_gre_allocations()
|
|
self.session = db.get_session()
|
|
|
|
def test_validate_provider_segment(self):
|
|
segment = {api.NETWORK_TYPE: 'gre',
|
|
api.PHYSICAL_NETWORK: 'phys_net',
|
|
api.SEGMENTATION_ID: None}
|
|
|
|
with testtools.ExpectedException(exc.InvalidInput):
|
|
self.driver.validate_provider_segment(segment)
|
|
|
|
segment[api.PHYSICAL_NETWORK] = None
|
|
with testtools.ExpectedException(exc.InvalidInput):
|
|
self.driver.validate_provider_segment(segment)
|
|
|
|
def test_sync_tunnel_allocations(self):
|
|
self.assertIsNone(
|
|
self.driver.get_gre_allocation(self.session,
|
|
(TUN_MIN - 1))
|
|
)
|
|
self.assertFalse(
|
|
self.driver.get_gre_allocation(self.session,
|
|
(TUN_MIN)).allocated
|
|
)
|
|
self.assertFalse(
|
|
self.driver.get_gre_allocation(self.session,
|
|
(TUN_MIN + 1)).allocated
|
|
)
|
|
self.assertFalse(
|
|
self.driver.get_gre_allocation(self.session,
|
|
(TUN_MAX - 1)).allocated
|
|
)
|
|
self.assertFalse(
|
|
self.driver.get_gre_allocation(self.session,
|
|
(TUN_MAX)).allocated
|
|
)
|
|
self.assertIsNone(
|
|
self.driver.get_gre_allocation(self.session,
|
|
(TUN_MAX + 1))
|
|
)
|
|
|
|
self.driver.gre_id_ranges = UPDATED_TUNNEL_RANGES
|
|
self.driver._sync_gre_allocations()
|
|
|
|
self.assertIsNone(
|
|
self.driver.get_gre_allocation(self.session,
|
|
(TUN_MIN + 5 - 1))
|
|
)
|
|
self.assertFalse(
|
|
self.driver.get_gre_allocation(self.session,
|
|
(TUN_MIN + 5)).allocated
|
|
)
|
|
self.assertFalse(
|
|
self.driver.get_gre_allocation(self.session,
|
|
(TUN_MIN + 5 + 1)).allocated
|
|
)
|
|
self.assertFalse(
|
|
self.driver.get_gre_allocation(self.session,
|
|
(TUN_MAX + 5 - 1)).allocated
|
|
)
|
|
self.assertFalse(
|
|
self.driver.get_gre_allocation(self.session,
|
|
(TUN_MAX + 5)).allocated
|
|
)
|
|
self.assertIsNone(
|
|
self.driver.get_gre_allocation(self.session,
|
|
(TUN_MAX + 5 + 1))
|
|
)
|
|
|
|
def test_reserve_provider_segment(self):
|
|
segment = {api.NETWORK_TYPE: 'gre',
|
|
api.PHYSICAL_NETWORK: 'None',
|
|
api.SEGMENTATION_ID: 101}
|
|
self.driver.reserve_provider_segment(self.session, segment)
|
|
alloc = self.driver.get_gre_allocation(self.session,
|
|
segment[api.SEGMENTATION_ID])
|
|
self.assertTrue(alloc.allocated)
|
|
|
|
with testtools.ExpectedException(exc.TunnelIdInUse):
|
|
self.driver.reserve_provider_segment(self.session, segment)
|
|
|
|
self.driver.release_segment(self.session, segment)
|
|
alloc = self.driver.get_gre_allocation(self.session,
|
|
segment[api.SEGMENTATION_ID])
|
|
self.assertFalse(alloc.allocated)
|
|
|
|
segment[api.SEGMENTATION_ID] = 1000
|
|
self.driver.reserve_provider_segment(self.session, segment)
|
|
alloc = self.driver.get_gre_allocation(self.session,
|
|
segment[api.SEGMENTATION_ID])
|
|
self.assertTrue(alloc.allocated)
|
|
|
|
self.driver.release_segment(self.session, segment)
|
|
alloc = self.driver.get_gre_allocation(self.session,
|
|
segment[api.SEGMENTATION_ID])
|
|
self.assertEqual(None, alloc)
|
|
|
|
def test_allocate_tenant_segment(self):
|
|
tunnel_ids = set()
|
|
for x in xrange(TUN_MIN, TUN_MAX + 1):
|
|
segment = self.driver.allocate_tenant_segment(self.session)
|
|
self.assertThat(segment[api.SEGMENTATION_ID],
|
|
matchers.GreaterThan(TUN_MIN - 1))
|
|
self.assertThat(segment[api.SEGMENTATION_ID],
|
|
matchers.LessThan(TUN_MAX + 1))
|
|
tunnel_ids.add(segment[api.SEGMENTATION_ID])
|
|
|
|
segment = self.driver.allocate_tenant_segment(self.session)
|
|
self.assertEqual(None, segment)
|
|
|
|
segment = {api.NETWORK_TYPE: 'gre',
|
|
api.PHYSICAL_NETWORK: 'None',
|
|
api.SEGMENTATION_ID: tunnel_ids.pop()}
|
|
self.driver.release_segment(self.session, segment)
|
|
segment = self.driver.allocate_tenant_segment(self.session)
|
|
self.assertThat(segment[api.SEGMENTATION_ID],
|
|
matchers.GreaterThan(TUN_MIN - 1))
|
|
self.assertThat(segment[api.SEGMENTATION_ID],
|
|
matchers.LessThan(TUN_MAX + 1))
|
|
tunnel_ids.add(segment[api.SEGMENTATION_ID])
|
|
|
|
for tunnel_id in tunnel_ids:
|
|
segment[api.SEGMENTATION_ID] = tunnel_id
|
|
self.driver.release_segment(self.session, segment)
|
|
|
|
def test_gre_endpoints(self):
|
|
tun_1 = self.driver.add_endpoint(TUNNEL_IP_ONE)
|
|
tun_2 = self.driver.add_endpoint(TUNNEL_IP_TWO)
|
|
self.assertEqual(TUNNEL_IP_ONE, tun_1.ip_address)
|
|
self.assertEqual(TUNNEL_IP_TWO, tun_2.ip_address)
|
|
|
|
# Get all the endpoints
|
|
endpoints = self.driver.get_endpoints()
|
|
for endpoint in endpoints:
|
|
self.assertIn(endpoint['ip_address'],
|
|
[TUNNEL_IP_ONE, TUNNEL_IP_TWO])
|