Python模拟获取地址而不是return_value

时间:2014-06-26 12:53:42

标签: python unit-testing mocking return-value

我正在尝试发出呼叫以检查用户的全名。我使用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)

我做错了什么?

1 个答案:

答案 0 :(得分:0)

您的代码也会调用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() 生成的。