无法在 python 中使用 unittest 模拟类方法

Question

模块a.ClassA:

class ClassA():
    def __init__(self,callingString):
        print callingString

    def functionInClassA(self,val):
        return val

模块b.ClassB:

from a.ClassA import ClassA

class ClassB():
    def __init__(self,val):
        self.value=val

    def functionInsideClassB(self):
        obj=ClassA("Calling From Class B")
        value=obj.functionInClassA(self.value)

Python unittest 类

import unittest
from b.ClassB import ClassB
from mock import patch, Mock, PropertyMock,mock

class Test(unittest.TestCase):
    @patch('b.ClassB.ClassA',autospec = True)
    def _test_sample(self,classAmock):
        dummyMock=Mock()
        dummyMock.functionInClassA.return_value="mocking functionInClassA"
        classAmock.return_value=dummyMock

        obj=ClassB("dummy_val")
        obj.functionInsideClassB()

        assert dummyMock.functionInClassA.assert_called_once_with("dummy_val")

断言失败。我到底哪里错了？

Answer 1

你两次分配给return_value：

classAmock.return_value=dummyMock
classAmock.return_value=Mock()

第二个任务完全取消了您设置dummyMock 的工作；新的Mock 实例没有设置functionInClassA 属性。

您不需要创建新的模拟对象；只需使用默认的return_value 属性值：

class Test(unittest.TestCase):
    @patch('b.ClassB.ClassA', autospec=True)
    def test_sample(self, classAmock):
        instance = classAmock.return_value
        instance.functionInClassA.return_value = "mocking functionInClassA"

        obj = ClassB("dummy_val")
        obj.functionInsideClassB()

        instance.functionInClassA.assert_called_once_with("dummy_val")

您确实不需要需要断言assert_called_once_with() 的返回值，因为它始终是None（使您的额外assert 总是失败）。将断言留给assert_called_once_with() 方法，它会根据需要引发。