Flask蓝图单元测试

2024-03-29 07:22:00 发布

您现在位置:Python中文网/ 问答频道 /正文

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

http://flask.pocoo.org/docs/testing/

我没有找到对我有帮助的或是足够简单的东西。

//Edit
这是我的代码:

# -*- 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

我想我找到了一个很好的解决办法。还是会有更好的办法?


Tags: pathfromtestimportselfclientapp应用程序
3条回答

蓝图与应用程序非常相似。我想您需要测试请求。

如果您希望将测试蓝图作为应用程序的一部分,那么看起来与应用程序没有区别。

如果您希望将测试蓝图作为扩展,那么可以使用自己的蓝图创建测试应用程序并对其进行测试。

如果这对任何人都有帮助的话,我会做以下的。我基本上把测试文件做成了我的烧瓶申请表

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()

我在一个应用程序中有多个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')

相关问题 更多 >