Python 3 unittest 不能模拟函数内部使用的方法

Question

我正在尝试在 Python 中进行单元测试。

test.py

from unittest import TestCase,main
from unittest.mock import patch

import file

def mock_return(*args):
    return -1

class Tests(TestCase):
    @patch("file.Foo.a", side_effect=mock_return)
    def test_1(self, mock_fun):
        self.assertEqual(file.Foo().a(), -1)

    @patch("os.urandom", side_effect=mock_return)
    def test_2(self, mock_fun):
        self.assertEqual(file.Foo().b(), -1)

if __name__ == "__main__":
    main()

file.py

from os import urandom

class Foo:
    def a(self):
        return 1

    def b(self):
        return urandom(1)

为什么test_1 通过但test_2 失败？有没有办法模拟其他类使用的方法？

Answer 1

您必须修补被测试函数使用的绑定。

from os import urandom  # in file.py

将名称 urandom 绑定到函数 os.urandom in file 模块。 Foo.b 通过 file.urandom 绑定访问该函数。所以Foo.b 的测试必须修补file.urandom，而不是os.urandom。