diff options
Diffstat (limited to 'tests/unit/db/sqlalchemy/base.py')
-rw-r--r-- | tests/unit/db/sqlalchemy/base.py | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/tests/unit/db/sqlalchemy/base.py b/tests/unit/db/sqlalchemy/base.py new file mode 100644 index 0000000..487af67 --- /dev/null +++ b/tests/unit/db/sqlalchemy/base.py @@ -0,0 +1,47 @@ +# 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 fixtures +from oslo.config import cfg + +from openstack.common.db.sqlalchemy import session +from tests import utils as test_utils + + +class SqliteInMemoryFixture(fixtures.Fixture): + """SQLite in-memory DB recreated for each test case.""" + + def __init__(self): + self.conf = cfg.CONF + self.conf.import_opt('connection', + 'openstack.common.db.sqlalchemy.session', + group='database') + + def setUp(self): + super(SqliteInMemoryFixture, self).setUp() + + self.conf.set_default('connection', "sqlite://", group='database') + self.addCleanup(self.conf.reset) + self.addCleanup(session.cleanup) + + +class DbTestCase(test_utils.BaseTestCase): + """Base class for testing of DB code (uses in-memory SQLite DB fixture).""" + + def setUp(self): + super(DbTestCase, self).setUp() + + self.useFixture(SqliteInMemoryFixture()) |