def connection(app, request): """Session-wide test database.""" # Connect and create the temporary database print "=" * 80 print "CREATING TEMPORARY DATABASE FOR TESTS" print "=" * 80 subprocess.call(['dropdb', DBNAME]) subprocess.call(['createdb', DBNAME]) # Bind and create the database tables _db.clear() engine_url = '%s/%s' % (BIND, DBNAME) db_engine = create_engine( engine_url, encoding='utf-8', poolclass=NullPool) # AKA models.start() _db.Session.bind = db_engine _db.metadata.bind = _db.Session.bind _db.Base.metadata.create_all(db_engine) _db.commit() _db.clear() def teardown(): _db.Base.metadata.drop_all(db_engine) request.addfinalizer(teardown) # Slap our test app on it _db.app = app return _db
def connection(app, request): """Session-wide test database.""" # Connect and create the temporary database print "=" * 80 print "CREATING TEMPORARY DATABASE FOR TESTS" print "=" * 80 subprocess.call(['dropdb', DBNAME]) subprocess.call(['createdb', DBNAME]) # Bind and create the database tables _db.clear() engine_url = '%s/%s' % (BIND, DBNAME) db_engine = create_engine(engine_url, encoding='utf-8', poolclass=NullPool) # AKA models.start() _db.Session.bind = db_engine _db.metadata.bind = _db.Session.bind _db.Base.metadata.create_all(db_engine) _db.commit() _db.clear() #connection = db_engine.connect() def teardown(): _db.Base.metadata.drop_all(db_engine) request.addfinalizer(teardown) # Slap our test app on it _db.app = app return _db
def after_return(self, status, retval, task_id, args, kwargs, einfo): models.clear()