Python 模拟补丁无法按预期用于 public 方法

Question

我正在尝试为我的烧瓶应用程序修补 public 方法，但它似乎不起作用。

这是我在 mrss.feed_burner

中的代码

def get_feed(env=os.environ):
   return 'something'

这就是我的使用方式

@app.route("/feed")
    def feed():
        mrss_feed = get_feed(env=os.environ)
        response = make_response(mrss_feed)
        response.headers["Content-Type"] = "application/xml"

        return response

这是我的测试，它没有解析。

def test_feed(self):
    with patch('mrss.feed_burner.get_feed', new=lambda: '<xml></xml>'):
        response = self.app.get('/feed')
        self.assertEquals('<xml></xml>', response.data)

Answer 1

我认为你的问题是你没有在正确的命名空间中打补丁。请参阅 where_to_patch unittest.mock.patch 的文档。

本质上，您是在修补 mrss.feed_burner 中 get_feed() 的定义，但您的视图处理程序 feed() 已经引用了原始 mrss.feed_burner.get_feed()。要解决此问题，您需要修补视图文件中的引用。

根据您在视图函数中对 get_feed 的使用，我假设您正在像这样导入 get_feed

view_file.py

from mrss.feed_burner import get_feed

如果是这样，您应该像这样修补 view_file.get_feed：

def test_feed(self):
    with patch('view_file.get_feed', new=lambda: '<xml></xml>'):
        ...

Python 模拟补丁无法按预期用于 public 方法

Python mock patch doesn't work as expected for public method

python

unit-testing

mocking

flask

magicmock