使用 unittest 模拟函数

Question

我正在尝试对调用另一个函数“test()”的函数“check()”进行单元测试。

我想模拟 test() 并调用一个函数 mocked() 来代替它。

这是我的代码：

import unittest
import mock

def check():
    return test()
def test():
    return "test"

def mocked():
    return "mocked"

class CheckTest(unittest.TestCase):
    @mock.patch(target='test.test', spec=True, new_callable=mocked)
    def test_test(self, mocked):
        self.assertEqual(check(), "mocked")




if __name__ == '__main__':
    unittest.main()

我只想用我的字符串替换 test() 返回的字符串。而且我不能覆盖或更改 test() 或 check() 函数。

Answer 1

有点简单：设置return_value即可：

import unittest
import mock

def check():
    return test()
def test():
    return "test"

class CheckTest(unittest.TestCase):
    @mock.patch('__main__.test')
    def test_test(self, mocked):
        mocked.return_value = "mocked"
        self.assertEqual(check(), "mocked")


if __name__ == '__main__':
    unittest.main()

mocked 是 test 函数的模拟版本。