x/gantt
Code Issues Proposed changes

Merge "Parameterize database connection in test.py"

Browse Source
This commit is contained in:
Jenkins
2012-12-20 23:10:43 +00:00
committed by Gerrit Code Review
parent fa0338f8ff f97d4c1cf5
commit 3a06c2101f
2 changed files with 22 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
nova/test.py
View File

@@ -72,39 +72,44 @@ _DB_CACHE = None
class Database(fixtures.Fixture):
def __init__(self, db_session, db_migrate):
def __init__(self, db_session, db_migrate, sql_connection,
sqlite_db, sqlite_clean_db):
self.sql_connection = sql_connection
self.sqlite_db = sqlite_db
self.sqlite_clean_db = sqlite_clean_db
self.engine = db_session.get_engine()
self.engine.dispose()
conn = self.engine.connect()
if CONF.sql_connection == "sqlite://":
if sql_connection == "sqlite://":
if db_migrate.db_version() > db_migrate.INIT_VERSION:
return
else:
testdb = os.path.join(CONF.state_path, CONF.sqlite_db)
testdb = os.path.join(CONF.state_path, sqlite_db)
if os.path.exists(testdb):
return
db_migrate.db_sync()
self.post_migrations()
if CONF.sql_connection == "sqlite://":
if sql_connection == "sqlite://":
conn = self.engine.connect()
self._DB = "".join(line for line in conn.connection.iterdump())
self.engine.dispose()
else:
cleandb = os.path.join(CONF.state_path, CONF.sqlite_clean_db)
cleandb = os.path.join(CONF.state_path, sqlite_clean_db)
shutil.copyfile(testdb, cleandb)
def setUp(self):
super(Database, self).setUp()
if CONF.sql_connection == "sqlite://":
if self.sql_connection == "sqlite://":
conn = self.engine.connect()
conn.connection.executescript(self._DB)
self.addCleanup(self.engine.dispose)
else:
shutil.copyfile(os.path.join(CONF.state_path,
CONF.sqlite_clean_db),
self.sqlite_clean_db),
os.path.join(CONF.state_path,
CONF.sqlite_db))
self.sqlite_db))
def post_migrations(self):
"""Any addition steps that are needed outside of the migrations."""
@@ -205,7 +210,10 @@ class TestCase(testtools.TestCase):
global _DB_CACHE
if not _DB_CACHE:
_DB_CACHE = Database(session, migration)
_DB_CACHE = Database(session, migration,
sql_connection=CONF.sql_connection,
sqlite_db=CONF.sqlite_db,
sqlite_clean_db=CONF.sqlite_clean_db)
self.useFixture(_DB_CACHE)
mox_fixture = self.useFixture(MoxStubout())

7
nova/tests/baremetal/db/base.py
View File

@@ -40,9 +40,12 @@ class BMDBTestCase(test.TestCase):
def setUp(self):
super(BMDBTestCase, self).setUp()
self.flags(baremetal_sql_connection='sqlite:///:memory:')
self.flags(baremetal_sql_connection='sqlite://')
global _DB_CACHE
if not _DB_CACHE:
_DB_CACHE = Database(bm_session, bm_migration)
_DB_CACHE = Database(bm_session, bm_migration,
sql_connection=CONF.baremetal_sql_connection,
sqlite_db=None,
sqlite_clean_db=None)
self.useFixture(_DB_CACHE)
self.context = nova_context.get_admin_context()