对烧瓶蓝图进行单元测试是否有良好的做法?
http://flask.pocoo.org/docs/testing/
我没有找到帮助我的东西,或者说这很简单。
//编辑
这是我的代码:
# -*- coding: utf-8 -*-
import sys
import os
import unittest
import flask
sys.path = [os.path.abspath('')] + sys.path
from app import create_app
from views import bp
class SimplepagesTestCase(unittest.TestCase):
def setUp(self):
self.app = create_app('development.py')
self.test_client = self.app.test_client()
def tearDown(self):
pass
def test_show(self):
page = self.test_client.get('/')
assert '404 Not Found' not in page.data
if __name__ == '__main__':
unittest.main()
在这种情况下,我测试蓝图。不是整个应用程序。为了测试蓝图,我已将应用的根路径添加到sys.path
。现在我可以导入create_app
函数来创建应用程序。我还初始化了test_client
。
我想我找到了一个很好的解决方案。或者会有更好的方法吗?
答案 0 :(得分:9)
如果这有助于任何人,我做了以下事情。我基本上把测试文件变成了我的Flask应用程序
from flask import Flask
import unittest
app = Flask(__name__)
from blueprint_file import blueprint
app.register_blueprint(blueprint, url_prefix='')
class BluePrintTestCase(unittest.TestCase):
def setUp(self):
self.app = app.test_client()
def test_health(self):
rv = self.app.get('/blueprint_path')
print rv.data
if __name__ == '__main__':
unittest.main()
答案 1 :(得分:8)
蓝图与应用程序非常相似。我想你想要测试test_client
个请求。
如果您希望将测试蓝图作为应用程序的一部分,那么看起来应用程序没有任何差异。
如果您想将测试蓝图作为扩展,那么您可以使用自己的蓝图创建测试应用程序并进行测试。
答案 2 :(得分:0)
我在一个应用程序中有多个API,因此使用url_prefix
具有多个蓝图。我不喜欢在测试这些API时必须在所有路径前面加上前缀。我使用以下类包装test_client
的蓝图:
class BlueprintClient():
def __init__(self, app_client, blueprint_url_prefix):
self.app_client = app_client
self.blueprint_url_prefix = blueprint_url_prefix.strip('/')
def _delegate(self, method, path, *args, **kwargs):
app_client_function = getattr(self.app_client, method)
prefixed_path = '/%s/%s' % (self.blueprint_url_prefix, path.lstrip('/'))
return app_client_function(prefixed_path, *args, **kwargs)
def get(self, *args, **kwargs):
return self._delegate('get', *args, **kwargs)
def post(self, *args, **kwargs):
return self._delegate('post', *args, **kwargs)
def put(self, *args, **kwargs):
return self._delegate('put', *args, **kwargs)
def delete(self, *args, **kwargs):
return self._delegate('delete', *args, **kwargs)
app_client = app.test_client()
api_client = BlueprintClient(app_client, '/api/v1')
api2_client = BlueprintClient(app_client, '/api/v2')