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

Question

作为一个最小的例子，我的 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')被调用？

Answer 1

您可以通过在测试模块中导入的模块访问它，从而在模块的mock元素上使用 Python 。你的测试模块应该是这样的：viewsapplication

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