def setup(self):
"""
Run once before every test. Setup the test database.
"""
self.app = create_app(TEST_SETTINGS)
self.client = self.app.test_client()
with self.app.test_request_context():
db.create_all()
db.session.commit()
pass
_apply_driver_hacks(self, app, info, options)
SQLAlchemy.apply_driver_hacks = apply_driver_hacks
from sqlalchemy.engine.url import make_url
from montana import create_app, db
app = create_app()
# If we have an in-memory database, setup the schema first.
info = make_url(app.config['SQLALCHEMY_DATABASE_URI'])
if info.drivername == 'sqlite' and info.database in (None, '', ':memory:'):
print 'Setting up in-memory database'
with app.test_request_context():
db.create_all()
if __name__ == '__main__':
import glob
import sys
from montana import models
if '--load-fixtures' in sys.argv:
with app.test_request_context():
for fixture in sorted(glob.iglob('fixtures/*.fixture.json')):
print 'Loading fixture: %s' % fixture
models.load_fixture(open(fixture))
if not app.config['API_KEY']:
print 'Warning: no API key set!'
