def tearDown(self):
engine = sqlalchemy.create_engine(self.db_uri + '/postgres')
conn = engine.connect()
conn.execute("commit")
# close all active connections
conn.execute("SELECT pg_terminate_backend(pg_stat_activity.pid) FROM pg_stat_activity WHERE pg_stat_activity.datname = 'counter_test' AND pid <> pg_backend_pid()")
conn.execute("commit")
conn.execute("drop database " + self.db_name)
conn.close()
def setup_db():
"""Set up the local and test databases."""
(base_uri, local_db) = application.config['SQLALCHEMY_DATABASE_URI'].rsplit('/', 1)
engine = sqlalchemy.create_engine('/'.join([base_uri, DEFAULT_DB]))
conn = engine.connect()
conn.execute('commit')
conn.execute(CREATE_DB % local_db)
conn.execute('commit')
test_db = local_db + '_test'
conn.execute(CREATE_DB % test_db)
conn.close()
def setup_db():
(base_uri, local_db) = settings.SQLALCHEMY_DATABASE_URI.rsplit('/', 1)
engine = sqlalchemy.create_engine('/'.join([base_uri, DEFAULT_DB]))
conn = engine.connect()
conn.execute('commit')
# The dbname part of the database URI for the selected configuration.
conn.execute(CREATE_DB % local_db)
conn.execute('commit')
# By convention, the name as the normal db but with "_test" appended.
test_db = local_db + "_test"
conn.execute(CREATE_DB % test_db)
conn.close()
def addFileLogging():
if LOG_SQL:
engine = sqlalchemy.create_engine(SQLALCHEMY_DATABASE_URI, echo=True)
logging.getLogger('sqlalchemy.engine').setLevel(logging.INFO)
from logging.handlers import RotatingFileHandler
# file location, maximum size of the file, and number of backups to be kept
file_handler = RotatingFileHandler('tmp/log.log', 'a', 1 * 1024 * 1024, 10)
# format like 2013-07-13 13:01:54,083 in location
file_handler.setFormatter(logging.Formatter('%(asctime)s %(levelname)s: %(message)s [in %(pathname)s:%(lineno)d]'))
# file logging level
file_handler.setLevel(logging.INFO)
app.logger.addHandler(file_handler)
def setUp(self):
# we need to use the root user
# to be able to create the new database
self.db_username = os.environ['DB_USERNAME']
self.db_password = os.environ['DB_PASSWORD']
self.db_name = os.environ['DATABASE_NAME'] + '_test'
self.db_uri = 'postgresql://%s:%s@%s:5432' % (self.db_username, self.db_password, DB_HOST)
engine = sqlalchemy.create_engine(self.db_uri + '/postgres')
conn = engine.connect()
conn.execute("commit")
conn.execute("create database " + self.db_name)
conn.close()
self.app_factory = self.create_app()
self.app = self.app_factory.test_client()
with self.app_factory.app_context():
db.create_all()
