use flags for sqlite db names and fix flags in dhcpbridge

4 files changed, 16 insertions, 7 deletions

diff --git a/nova/flags.py b/nova/flags.py
index f64a62da9..ab1adc6e3 100644
--- a/nova/flags.py
+++ b/nova/flags.py
@@ -285,8 +285,9 @@ DEFINE_string('state_path', os.path.join(os.path.dirname(__file__), '../'),
 DEFINE_string('logdir', None, 'output to a per-service log file in named '
                               'directory')
 
+DEFINE_string('sqlite_db', 'nova.sqlite', 'file name for sqlite')
 DEFINE_string('sql_connection',
-              'sqlite:///$state_path/nova.sqlite',
+              'sqlite:///$state_path/$sqlite_db',
               'connection string for sql database')
 DEFINE_integer('sql_idle_timeout',
               3600,
diff --git a/nova/test.py b/nova/test.py
index 6cbbb9e8e..9c961a987 100644
--- a/nova/test.py
+++ b/nova/test.py
@@ -22,12 +22,14 @@ Allows overriding of flags for use of fakes,
 and some black magic for inline callbacks.
 """
 
+
 import datetime
+import os
+import shutil
 import uuid
 import unittest
 
 import mox
-import shutil
 import stubout
 
 from nova import context
@@ -39,8 +41,8 @@ from nova import service
 
 
 FLAGS = flags.FLAGS
-flags.DEFINE_bool('flush_db', True,
-                  'Flush the database before running fake tests')
+flags.DEFINE_string('sqlite_clean_db', 'clean.sqlite',
+                    'File name of clean sqlite db')
 flags.DEFINE_bool('fake_tests', True,
                   'should we use everything for testing')
 
@@ -65,7 +67,8 @@ class TestCase(unittest.TestCase):
         #             now that we have some required db setup for the system
         #             to work properly.
         self.start = datetime.datetime.utcnow()
-        shutil.copyfile("clean.sqlite", "tests.sqlite")
+        shutil.copyfile(os.path.join(FLAGS.state_path, FLAGS.sqlite_clean_db),
+                        os.path.join(FLAGS.state_path, FLAGS.sqlite_db))
 
         # emulate some of the mox stuff, we can't use the metaclass
         # because it screws with our generators
diff --git a/nova/tests/__init__.py b/nova/tests/__init__.py
index 5afd9389d..dbd433054 100644
--- a/nova/tests/__init__.py
+++ b/nova/tests/__init__.py
@@ -37,13 +37,17 @@ setattr(__builtin__, '_', lambda x: x)
 
 
 def setup():
+    import os
     import shutil
+
     from nova import context
     from nova import flags
     from nova.db import migration
     from nova.network import manager as network_manager
     from nova.tests import fake_flags
+
     FLAGS = flags.FLAGS
+
     migration.db_sync()
     ctxt = context.get_admin_context()
     network_manager.VlanManager().create_networks(ctxt,
@@ -54,4 +58,5 @@ def setup():
                                                   FLAGS.vlan_start,
                                                   FLAGS.vpn_start,
                                                   )
-    shutil.copyfile("tests.sqlite", "clean.sqlite")
+    shutil.copyfile(os.path.join(FLAGS.state_path, FLAGS.sqlite_db),
+                    os.path.join(FLAGS.state_path, FLAGS.sqlite_clean_db))
diff --git a/nova/tests/fake_flags.py b/nova/tests/fake_flags.py
index a8291a968..dcc8a676d 100644
--- a/nova/tests/fake_flags.py
+++ b/nova/tests/fake_flags.py
@@ -39,6 +39,6 @@ FLAGS.num_shelves = 2
 FLAGS.blades_per_shelf = 4
 FLAGS.iscsi_num_targets = 8
 FLAGS.verbose = True
-FLAGS.sql_connection = 'sqlite:///tests.sqlite'
+FLAGS.sqlite_db = "tests.sqlite"
 FLAGS.use_ipv6 = True
 FLAGS.logfile = 'tests.log'