#!/usr/bin/env python # -*- coding: utf-8 -*- import os import sys import subprocess from flask.ext.script import Manager, Shell, Server from flask.ext.migrate import MigrateCommand from kraken.app import create_app from kraken.user.models import User from kraken.settings import DevConfig, ProdConfig from kraken.database import db if os.environ.get("KRAKEN_ENV") == 'prod': app = create_app(ProdConfig) else: app = create_app(DevConfig) manager = Manager(app) TEST_CMD = "nosetests" def _make_context(): '''Return context dict for a shell session so you can access app, db, and the User model by default. ''' return {'app': app, 'db': db, 'User': User} @manager.command def test(): '''Run the tests.''' status = subprocess.call(TEST_CMD, shell=True)
def create_app(self): app = create_app(TestConfig) with app.app_context(): db.create_all() return app