38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
import subprocess
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker, scoped_session,create_session
|
|
|
|
from sqlalchemy.pool import NullPool
|
|
from artifice.models import Resource, Tenant, UsageEntry, SalesOrder, Base
|
|
|
|
DATABASE_NAME = "test_artifice"
|
|
|
|
PG_DATABASE_URI = "postgresql://aurynn:postgres@localhost/%s" % DATABASE_NAME
|
|
MY_DATABASE_URI = "mysql://root:password@localhost/%s" % DATABASE_NAME
|
|
|
|
def setUp():
|
|
subprocess.call(["/usr/bin/createdb","%s" % DATABASE_NAME])
|
|
subprocess.call(["mysql", "-u", "root","--password=password", "-e", "CREATE DATABASE %s" % DATABASE_NAME])
|
|
mysql_engine = create_engine(MY_DATABASE_URI, poolclass=NullPool)
|
|
pg_engine = create_engine(PG_DATABASE_URI, poolclass=NullPool)
|
|
Base.metadata.create_all(bind=mysql_engine)
|
|
Base.metadata.create_all(bind=pg_engine)
|
|
|
|
mysql_engine.dispose()
|
|
pg_engine.dispose()
|
|
|
|
def tearDown():
|
|
|
|
mysql_engine = create_engine(MY_DATABASE_URI, poolclass=NullPool)
|
|
pg_engine = create_engine(PG_DATABASE_URI, poolclass=NullPool)
|
|
|
|
Base.metadata.drop_all(bind=mysql_engine)
|
|
Base.metadata.drop_all(bind=pg_engine)
|
|
|
|
mysql_engine.dispose()
|
|
pg_engine.dispose()
|
|
|
|
|
|
subprocess.call(["/usr/bin/dropdb","%s" % DATABASE_NAME])
|
|
subprocess.call(["mysql", "-u", "root", "--password=password", "-e", "DROP DATABASE %s" % DATABASE_NAME])
|