from daklib.config import Config
from daklib.dbconn import DBConn
-from sqlalchemy import create_engine
+from sqlalchemy import create_engine, __version__
from sqlalchemy.exc import SADeprecationWarning
import pickle
class DBDakTestCase(DakTestCase):
def setUp(self):
cnf = Config()
+ if cnf["DB::Name"] in ('backports', 'obscurity', 'projectb'):
+ self.fail("You have configured an invalid database name: '%s'." % \
+ cnf["DB::Name"])
if cnf["DB::Host"]:
# TCP/IP
connstr = "postgres://%s" % cnf["DB::Host"]
if cnf["DB::Port"] and cnf["DB::Port"] != "-1":
connstr += "?port=%s" % cnf["DB::Port"]
- pickle_file = open(fixture('db-metadata.pkl'), 'r')
+ pickle_filename = 'db-metadata-%s.pkl' % __version__
+ pickle_file = open(fixture(pickle_filename), 'r')
self.metadata = pickle.load(pickle_file)
self.metadata.ddl_listeners = pickle.load(pickle_file)
pickle_file.close()
self.session = DBConn().session()
def tearDown(self):
- #pass
self.session.close()
#self.metadata.drop_all()