7b45f7536b
Synced from commit 5fb343faee1442921e2d610b1a5222e67418c4cd. Due to changes in oslo.db API, this sync requires a bit more work on Neutron side. Sync includes the following commits: 5b7e61c Dispose db connections pool on disconnect d1988b9 Set sql_mode callback on connect instead of checkout a1a8280 Fix excessive logging from db.sqlalchemy.session 9933bdd Get mysql_sql_mode parameter from config 96a2217 Prevent incorrect usage of _wrap_db_error() 20a7510 Add from_config() method to EngineFacade fea119e Drop special case for MySQL traditional mode, update unit tests dda24eb Introduce mysql_sql_mode option, remove old warning 0b5af67 Introduce a method to set any MySQL session SQL mode 8dccc7b Handle ibm_db_sa DBDuplicateEntry integrity errors 5b9e9f4 Fix doc build errors in db.sqlalchemy ac84a40 Update log translation domains 86707cd Remove None for dict.get() 0545121 Fix duplicating of SQL queries in logs fcf517d Update oslo log messages with translation domains 630d395 Don't use cfg.CONF in oslo.db ce69e7f Don't store engine instances in oslo.db Change-Id: I0e1d86878d3eb924b01e04dced0f90b4e57757d8
154 lines
4.8 KiB
Python
154 lines
4.8 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 abc
|
|
import functools
|
|
import os
|
|
|
|
import fixtures
|
|
import six
|
|
|
|
from neutron.openstack.common.db.sqlalchemy import session
|
|
from neutron.openstack.common.db.sqlalchemy import utils
|
|
from neutron.openstack.common.fixture import lockutils
|
|
from neutron.openstack.common import test
|
|
|
|
|
|
class DbFixture(fixtures.Fixture):
|
|
"""Basic database fixture.
|
|
|
|
Allows to run tests on various db backends, such as SQLite, MySQL and
|
|
PostgreSQL. By default use sqlite backend. To override default backend
|
|
uri set env variable OS_TEST_DBAPI_CONNECTION with database admin
|
|
credentials for specific backend.
|
|
"""
|
|
|
|
def _get_uri(self):
|
|
return os.getenv('OS_TEST_DBAPI_CONNECTION', 'sqlite://')
|
|
|
|
def __init__(self, test):
|
|
super(DbFixture, self).__init__()
|
|
|
|
self.test = test
|
|
|
|
def setUp(self):
|
|
super(DbFixture, self).setUp()
|
|
|
|
self.test.engine = session.create_engine(self._get_uri())
|
|
self.test.sessionmaker = session.get_maker(self.test.engine)
|
|
self.addCleanup(self.test.engine.dispose)
|
|
|
|
|
|
class DbTestCase(test.BaseTestCase):
|
|
"""Base class for testing of DB code.
|
|
|
|
Using `DbFixture`. Intended to be the main database test case to use all
|
|
the tests on a given backend with user defined uri. Backend specific
|
|
tests should be decorated with `backend_specific` decorator.
|
|
"""
|
|
|
|
FIXTURE = DbFixture
|
|
|
|
def setUp(self):
|
|
super(DbTestCase, self).setUp()
|
|
self.useFixture(self.FIXTURE(self))
|
|
|
|
|
|
ALLOWED_DIALECTS = ['sqlite', 'mysql', 'postgresql']
|
|
|
|
|
|
def backend_specific(*dialects):
|
|
"""Decorator to skip backend specific tests on inappropriate engines.
|
|
|
|
::dialects: list of dialects names under which the test will be launched.
|
|
"""
|
|
def wrap(f):
|
|
@functools.wraps(f)
|
|
def ins_wrap(self):
|
|
if not set(dialects).issubset(ALLOWED_DIALECTS):
|
|
raise ValueError(
|
|
"Please use allowed dialects: %s" % ALLOWED_DIALECTS)
|
|
if self.engine.name not in dialects:
|
|
msg = ('The test "%s" can be run '
|
|
'only on %s. Current engine is %s.')
|
|
args = (f.__name__, ' '.join(dialects), self.engine.name)
|
|
self.skip(msg % args)
|
|
else:
|
|
return f(self)
|
|
return ins_wrap
|
|
return wrap
|
|
|
|
|
|
@six.add_metaclass(abc.ABCMeta)
|
|
class OpportunisticFixture(DbFixture):
|
|
"""Base fixture to use default CI databases.
|
|
|
|
The databases exist in OpenStack CI infrastructure. But for the
|
|
correct functioning in local environment the databases must be
|
|
created manually.
|
|
"""
|
|
|
|
DRIVER = abc.abstractproperty(lambda: None)
|
|
DBNAME = PASSWORD = USERNAME = 'openstack_citest'
|
|
|
|
def _get_uri(self):
|
|
return utils.get_connect_string(backend=self.DRIVER,
|
|
user=self.USERNAME,
|
|
passwd=self.PASSWORD,
|
|
database=self.DBNAME)
|
|
|
|
|
|
@six.add_metaclass(abc.ABCMeta)
|
|
class OpportunisticTestCase(DbTestCase):
|
|
"""Base test case to use default CI databases.
|
|
|
|
The subclasses of the test case are running only when openstack_citest
|
|
database is available otherwise a tests will be skipped.
|
|
"""
|
|
|
|
FIXTURE = abc.abstractproperty(lambda: None)
|
|
|
|
def setUp(self):
|
|
# TODO(bnemec): Remove this once infra is ready for
|
|
# https://review.openstack.org/#/c/74963/ to merge.
|
|
self.useFixture(lockutils.LockFixture('opportunistic-db'))
|
|
credentials = {
|
|
'backend': self.FIXTURE.DRIVER,
|
|
'user': self.FIXTURE.USERNAME,
|
|
'passwd': self.FIXTURE.PASSWORD,
|
|
'database': self.FIXTURE.DBNAME}
|
|
|
|
if self.FIXTURE.DRIVER and not utils.is_backend_avail(**credentials):
|
|
msg = '%s backend is not available.' % self.FIXTURE.DRIVER
|
|
return self.skip(msg)
|
|
|
|
super(OpportunisticTestCase, self).setUp()
|
|
|
|
|
|
class MySQLOpportunisticFixture(OpportunisticFixture):
|
|
DRIVER = 'mysql'
|
|
|
|
|
|
class PostgreSQLOpportunisticFixture(OpportunisticFixture):
|
|
DRIVER = 'postgresql'
|
|
|
|
|
|
class MySQLOpportunisticTestCase(OpportunisticTestCase):
|
|
FIXTURE = MySQLOpportunisticFixture
|
|
|
|
|
|
class PostgreSQLOpportunisticTestCase(OpportunisticTestCase):
|
|
FIXTURE = PostgreSQLOpportunisticFixture
|