babe698caf
The foreign key constraints will be enabled for each SQLite database connection. By default the foreign key constraints are disabled in SQLite, so some test cases failed after enabling the foreign key constraints for unit tests. This fixings also fixed the failed test cases because of the foreign key enforcement. Fixes: bug #1021023 Change-Id: I89f0cbbd75bb685b50dfe6628116fa971c5e78cb
79 lines
3.1 KiB
Python
79 lines
3.1 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
#
|
|
# Copyright 2012 Isaku Yamahata <yamahata at private email ne jp>
|
|
# 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.
|
|
|
|
from contextlib import nested
|
|
import operator
|
|
import unittest2
|
|
|
|
from quantum.db import api as db
|
|
from quantum.db import models_v2
|
|
from quantum.openstack.common import cfg
|
|
from quantum.plugins.ryu.common import config
|
|
from quantum.plugins.ryu.db import api_v2 as db_api_v2
|
|
from quantum.plugins.ryu.db import models_v2 as ryu_models_v2
|
|
from quantum.plugins.ryu import ofp_service_type
|
|
from quantum.tests.unit import test_db_plugin as test_plugin
|
|
|
|
|
|
class RyuDBTest(test_plugin.QuantumDbPluginV2TestCase):
|
|
def setUp(self):
|
|
super(RyuDBTest, self).setUp()
|
|
self.hosts = [(cfg.CONF.OVS.openflow_controller,
|
|
ofp_service_type.CONTROLLER),
|
|
(cfg.CONF.OVS.openflow_rest_api,
|
|
ofp_service_type.REST_API)]
|
|
db_api_v2.set_ofp_servers(self.hosts)
|
|
|
|
def test_ofp_server(self):
|
|
session = db.get_session()
|
|
servers = session.query(ryu_models_v2.OFPServer).all()
|
|
print servers
|
|
self.assertEqual(len(servers), 2)
|
|
for s in servers:
|
|
self.assertTrue((s.address, s.host_type) in self.hosts)
|
|
|
|
@staticmethod
|
|
def _tunnel_key_sort(key_list):
|
|
key_list.sort(key=operator.attrgetter('tunnel_key'))
|
|
return [(key.network_id, key.tunnel_key) for key in key_list]
|
|
|
|
def test_key_allocation(self):
|
|
tunnel_key = db_api_v2.TunnelKey()
|
|
session = db.get_session()
|
|
with nested(self.network('network-0'),
|
|
self.network('network-1')
|
|
) as (network_0,
|
|
network_1):
|
|
network_id0 = network_0['network']['id']
|
|
key0 = tunnel_key.allocate(session, network_id0)
|
|
network_id1 = network_1['network']['id']
|
|
key1 = tunnel_key.allocate(session, network_id1)
|
|
key_list = tunnel_key.all_list()
|
|
self.assertEqual(len(key_list), 2)
|
|
|
|
expected_list = [(network_id0, key0), (network_id1, key1)]
|
|
self.assertEqual(self._tunnel_key_sort(key_list),
|
|
expected_list)
|
|
|
|
tunnel_key.delete(session, network_id0)
|
|
key_list = tunnel_key.all_list()
|
|
self.assertEqual(self._tunnel_key_sort(key_list),
|
|
[(network_id1, key1)])
|
|
|
|
tunnel_key.delete(session, network_id1)
|
|
self.assertEqual(tunnel_key.all_list(), [])
|