改变使用补丁创建的模拟对象方法的副作用

时间:2016-06-12 08:09:37

标签: python django unit-testing mocking python-mock

您好我想测试一下django的视图。在视图内部,我创建了两个对象,我想模拟它们的一些方法。测试看起来像这样

@mock.patch('payments_system.views.FirstObject')
@mock.patch('payments_system.helper_functions.SecondObject')
def test_make_payment_view_with_success(self, MockFirstObject, MockSecondObject):

    MockFirstObject.get_value.side_effect = get_value_side_effect   //this function is defined and implemented outside my testcase class

    MockSecondObject.is_valid.return_value = True

    factory = RequestFactory()
    request = factory.post(reverse('cardinal-term_url'), data=dict(PaRes="test_parese", MD=None))
    self._add_session_to_request(request)
    session_data = dict(amount=1000, Centinel_PIType="VISA", Card_Number="40000000000000001", ExpMonth=06,
                        ExpYear=2016, Cvv2='123')
    request = self._add_session_data_to_request(request, **session_data)
    response = term_url(request)

    self.assertRedirects(response, reverse('payments_system-success', kwargs={"token": "some_token"}))

当我调试我的测试用例并进入我的视图时,视图中创建的对象确实是Mocks的类型。但get_value方法不使用side_effect函数,但也返回MockingObject。如何在django视图中传递模拟对象的更改?补丁版本是否与以下相同?

MockFirstObject = MagicMock(spec=payments_system.views.FirstObject)
MockSecondObject = MagicMock(spec=payments_system.helper_functions.SecondOjbect)

我还需要做些什么吗?

1 个答案:

答案 0 :(得分:0)

我终于设法解决了我的问题。我做的是以下内容:

在测试函数中

mock_object1_instance = MockFirstObject1.return_value
mock_object1_instance.get_value.side_effect = get_value_side_effect

我为其他模拟对象所做的一样。