Flask蓝图单元测试

时间:2013-11-13 19:19:00

标签: python unit-testing flask blueprint

对烧瓶蓝图进行单元测试是否有良好的做法?

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

我想我找到了一个很好的解决方案。或者会有更好的方法吗?

3 个答案:

答案 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')