Python 模拟补丁不是 mocking object

Question

我正在使用 python mocking 库，但我不确定为什么会得到这个结果。 为什么只有第二个被嘲笑而不是第一个？ 最好的方法是什么？

import unittest
from unittest.mock import patch, Mock

import requests
from requests import Session


def my_func():
    s = Session()
    print('my_func', type(s))


def my_func2():
    s = requests.Session()
    print('my_func2', type(s))


class Test(unittest.TestCase):

    @patch("requests.Session", new=Mock)
    def test1(self, *args):
        my_func()

    @patch("requests.Session", new=Mock)
    def test2(self, *args):
        my_func2()

Output

my_func <class 'requests.sessions.Session'>
my_func2 <class 'unittest.mock.Mock'>

Answer 1

这是关于 Python 命名空间的一个微妙的细微差别。 在第一个my_func()中，您只使用了Session （在使用from requests import Session将其导入您的模块命名空间之后）。 在my_func2()你使用requests.Session 。

@patch('requests.Session')正在替换requests模块中的Session ，因此requests.Session将是模拟 object。 但它并不能取代每个模块中对Session的所有引用——事实上，在 Python 中很难做到这一点。

实际上，在基本层面上， patch()所做的只是设置requests.Session = Mock() 。 但是要替换对Session的引用，它已经在你的模块的全局命名空间中，它还必须设置globals()['Session'] = Mock() 。

换句话说，您必须在正确的命名空间中修补 object。

如果你有一些模块foo.py包含：

from requests import Session

def my_func():
    s = Session()
    ...

然后是一个单独的测试模块test_foo.py ，您要在其中模拟Session class，您必须在foo的命名空间中对其进行修补，例如：

import foo

@patch('foo.Session', ...)
def test_my_func():
    ...