我正在尝试为简单的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
答案 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
对象。