Python Mock 获取地址而不是 return_value

Question

我正在尝试转接电话以检查用户全名。我已经使用 mock 编写了以下方法。

def test_UserDisplayName(self):
    appModel = Mock()
    eval = appModel.eval.return_value
    eval.userDisplayName.return_value = 'JohnDoe'
    self._SummaryModel.AppModel = appModel
    actual = self._SummaryModel.UserDisplayName()
    self.assertEqual(actual, 'JohnDoe')

这是它正在调用的方法。

def UserDisplayName(self):
    return self.AppModel().eval().userDisplayName()

如果我将调试器附加到上述行并在 shell 中运行它就可以工作

[PAUSED] >>> self.AppModel().eval().userDisplayName()
'JohnDoe'
[PAUSED] >>> 

但是当运行它的测试用例失败时，我得到的是地址而不是值。

   ======================================================================
   FAIL: test_UserDisplayName (unittests.model.SummaryModelTest)
   ----------------------------------------------------------------------
   Traceback (most recent call last):
     File "/unittests/summary_model.py", line 112, in test_UserDisplayName
       self.assertEqual(actual, 'JohnDoe')
   AssertionError: <Mock name='mock.userDisplayName()' id='233406864'> != 'JohnDoe'

   ----------------------------------------------------------------------
   Ran 1 test in 0.010s

   FAILED:  (failures=1)

我做错了什么？

Answer 1

您的代码也调用了AppModel，因此您需要为此调整其他引用：

eval = appModel.return_value.eval.return_value

演示：

>>> from unittest.mock import Mock
>>> appModel = Mock()
>>> eval = appModel.return_value.eval.return_value
>>> eval.userDisplayName.return_value = 'JohnDoe'
>>> appModel().eval().userDisplayName()
'JohnDoe'

但是，您的<Mock name='mock.userDisplayName()' id='233406864'> 对象显示名称为mock.userDisplayName()，这表明返回值是通过使用self.AppModel.userDisplayName()直接 生成的。