def test_prod_config(self):
""" Tests if the production config loads correctly """
app = create_app("zeus.settings.ProdConfig")
assert app.config["SQLALCHEMY_DATABASE_URI"] == "sqlite:///../database.db"
assert app.config["CACHE_TYPE"] == "simple"
def test_dev_config(self):
""" Tests if the development config loads correctly """
app = create_app("zeus.settings.DevConfig")
assert app.config["DEBUG"] is True
assert app.config["SQLALCHEMY_DATABASE_URI"] == "sqlite:///../database.db"
assert app.config["CACHE_TYPE"] == "null"
def test_test_config(self):
""" Tests if the test config loads correctly """
app = create_app("zeus.settings.TestConfig")
assert app.config["DEBUG"] is True
assert app.config["SQLALCHEMY_ECHO"] is True
assert app.config["CACHE_TYPE"] == "null"
def testapp(request):
app = create_app('zeus.settings.TestConfig')
client = app.test_client()
db.app = app
db.create_all()
if getattr(request.module, "create_user", True):
admin = User('admin', 'supersafepassword')
db.session.add(admin)
db.session.commit()
def teardown():
db.session.remove()
db.drop_all()
request.addfinalizer(teardown)
return client
#!/usr/bin/env python
import os
from flask.ext.script import Manager, Server
from flask.ext.script.commands import ShowUrls, Clean
from zeus import create_app
from zeus.models import db, User
# default to dev config because no one should use this in
# production anyway
env = os.environ.get('zeus_ENV', 'dev')
app = create_app('zeus.settings.%sConfig' % env.capitalize())
manager = Manager(app)
manager.add_command("server", Server())
manager.add_command("show-urls", ShowUrls())
manager.add_command("clean", Clean())
@manager.shell
def make_shell_context():
""" Creates a python REPL with several default imports
in the context of the app
"""
return dict(app=app, db=db, User=User)
@manager.command
def createdb():
""" Creates a database with all of the tables defined in
