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模块中的函数file 。 Foo.b通过file.urandom绑定访问该函数。因此，Foo.b的测试必须修补file.urandom，而不是os.urandom。

Python 3 unittest不能在函数内部使用mock方法

1 个答案: