Python模拟：运行多个测试时模拟对象未更新

Question

我正在尝试模拟一个调用remote_api的函数：

def get_remote_value():
    ret = make_distant_call()
    return ret > 0

此函数在另一个函数中调用：

from another_file import get_remote_value

def check_remote_value():
    remote_value = get_remote_value()
    # Actually do some computation but it doesn't change the issue
    return remote_value

这是我的测试：

@mock.patch('another_file.get_remote_value')
class MyTest(TestCase):
    def first_test(self, mock_get_remote_value):
        mock_get_remote_value.return_value = True
        self.assertEqual(check_remote_value(), True)

    def second_test(self, mock_get_remote_value):
        mock_get_remote_value.return_value = False
        self.assertEqual(check_remote_value(), False)

当我自己运行每个测试时，它可以正常工作。 当我运行整个类时，第二个测试失败，因为get_remote_value返回True而不是False 。

我在想check_remote_value函数仍在使用旧的模拟，这就是导致问题的原因。 我对吗？ 无论如何，如何更改测试以使其顺利运行？

我尝试在每个函数上使用装饰器，使用修补程序上下文管理器，但无济于事。 模拟整个check_remote_value并不是一个真正的选择，因为这是我要测试的选择。

Answer 1

您需要修补check_remote_value实际使用的名称。

@mock.path('mymodule.utils.another_file.get_remote_value')
class MyTest(TestCase):
    def first_test(self, mock_get_remote_value):
        mock_get_remote_value.return_value = True
        self.assertEqual(check_remote_value(), True)

    def second_test(self, mock_get_remote_value):
        mock_get_remote_value.return_value = False
        self.assertEqual(check_remote_value(), False)

这是由于函数如何查找全局值。 check_remote_value引用了在mymodule.utils定义的全局范围，而不是您的测试脚本，因此在需要查找get_remote_value 。