我如何在烧瓶测试中模拟我的烧瓶应用程序的视图模块的依赖关系?

How do I mock dependencies of the views module of my Flask application in flask-testing?

作为一个最小的例子,我的 Flask 应用程序有一个 views 模块,例如

from flask import render_template
from something import some_service

def home():
    foo = some_service.do_thing('bar')
    return render_template('index.html', foo=foo)

我有一个像

这样的测试设置
from application import app
from flask.ext.testing import TestCase

class MyTest(TestCase):

    def create_app(self):
        app.config['TESTING'] = True
        app.config['WTF_CSRF_ENABLED'] = False
        return app

    def setUp(self):
        self.app = app.test_client()

    def test_home(self):
        rv = self.app.get('/home')
        ???

如何编写 test_home 以便它断言调用了 some_service.do_thing('bar')

您可以在 views 模块的元素上使用 Python mock,方法是通过测试模块中导入的 application 模块访问它。你的测试模块应该是这样的:

import application
from flask.ext.testing import TestCase

class MyTest(TestCase):

    def create_app(self):
        application.app.config['TESTING'] = True
        application.app.config['WTF_CSRF_ENABLED'] = False
        return application.app

    def setUp(self):
        self.app = application.app.test_client()

    def test_home(self):
        mock_service = mock.MagicMock()
        application.views.some_service = mock_service
        self.app.get('/home')
        mock_service.do_thing.assert_called_with('bar')