烧瓶测试 - 测试失败的原因

时间:2016-04-22 21:13:44

标签: python flask flask-testing

我正在尝试为简单的Flask应用程序编写测试。该项目的结构如下:

app/
    static/
    templates/
    forms.py
    models.py
    views.py
migrations/
config.py
manage.py
tests.py

tests.py

import unittest
from app import create_app, db
from flask import current_app
from flask.ext.testing import TestCase

class AppTestCase(TestCase):
    def create_app(self):
        return create_app('test_config')

    def setUp(self):
        db.create_all()

    def tearDown(self):
        db.session.remove()
        db.drop_all()

    def test_hello(self):
        response = self.client.get('/')
        self.assert_200(response)

应用程序/的初始化的.py

# app/__init__.py

from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy
from config import config

db = SQLAlchemy()

def create_app(config_name):
    app = Flask(__name__)
    app.config.from_object(config[config_name])
    db.init_app(app)
    return app

app = create_app('default')

from . import views

当我启动测试时,test_hello失败,因为response.status_code是404.请告诉我,我该如何解决?看起来,app实例对views.py中的视图函数一无所知。如果需要整个代码,可以找到here

1 个答案:

答案 0 :(得分:1)

您的views.py文件会在app文件中创建的__init__.py中装入路由。

您必须在create_app测试方法中将这些路线绑定到您创建的应用。

我建议你颠倒依赖。而是views.py导入您的代码,您可以从init_app或从测试文件中导入和调用__init__.py

# views.py
def init_app(app):
    app.add_url_rule('/', 'index', index)
    # repeat to each route

使用Blueprint可以做得更好。

def init_app(app):
    app.register_blueprint(blueprint)

这样,您的测试文件就可以导入此init_app并将蓝图绑定到测试app对象。