def setUp(self):
"""Set up a test data base.
"""
try:
db = pgconn.connect("dbname=planet user=osm")
db.set_isolation_level(psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)
cur = psycopg2.extensions.connection.cursor(db)
cur.execute("""CREATE DATABASE testdb
WITH TEMPLATE osmosis_test_template
ENCODING 'UTF-8';
""")
self.db = pgconn.connect("dbname=testdb user=osm")
self.fillDB()
except:
self.tearDown()
raise
def tearDown(self):
"""Delete the test database.
"""
if not hasattr(self, 'db'):
return
if not self.db.closed:
self.db.close()
db = pgconn.connect("dbname=planet user=osm")
db.set_isolation_level(psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)
cur = psycopg2.extensions.connection.cursor(db)
cur.execute("DROP DATABASE testdb;")
