def app(): _app = create_app(TestConfig) ctx = _app.test_request_context() ctx.push() yield _app ctx.pop()
#!/usr/bin/env python # -*- coding: utf-8 -*- import os from flask_script import Manager, Shell, Server from flask_migrate import MigrateCommand from TALE_Toolbox.app import create_app from TALE_Toolbox.settings import DevConfig, ProdConfig if os.environ.get("TALE_TOOLBOX_ENV") == 'prod': app = create_app(ProdConfig) else: app = create_app(DevConfig) HERE = os.path.abspath(os.path.dirname(__file__)) TEST_PATH = os.path.join(HERE, 'tests') manager = Manager(app) 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} @manager.command def test(): """Run the tests.""" import pytest
def test_production_config(): app = create_app(ProdConfig) assert app.config["ENV"] == "prod" assert app.config["DEBUG"] is False assert app.config["DEBUG_TB_ENABLED"] is False assert app.config["ASSETS_DEBUG"] is False
def test_dev_config(): app = create_app(DevConfig) assert app.config["ENV"] == "dev" assert app.config["DEBUG"] is True assert app.config["ASSETS_DEBUG"] is True