在Python中添加Mock对象

My code under test does sth like this:def to_be_tested(x): return round((x.a + x.b).c())I would like to test it by passing a Mock object as x. I tried to do it like this:import unittestimport unittest.mockclass Test_X(unittest.TestCase): def test_x(self): m = unittest.mock.Mock() to_be_tested(m) # now check if the proper call has taken place on mThe call to x.a and the one to x.b work as expected. They deliver new mock objects which can be asked how they were created (e. g. via q._mock_parent and q._mock_new_name), so this step works just fine.But then the addition is supposed to take place which just raises an error (TypeError: unsupported operand type(s) for +: 'Mock' and 'Mock'). I hoped this also would return a mock object so that the call to .c() can take place and (again) return a mock object.I also considered m.__add__ = lambda a, b: unittest.mock.Mock(a, b) prior to the call of the code under test, but that wouldn't help since not my original Mock is going to be added but a newly created one.I also tried the (already quite cumbersome) m.a.__add__ = lambda a, b: unittest.mock.Mock(a, b). But that (to my surprise) led to raising an AttributeError: Mock object has no attribute 'c' when calling the code under test. Which I do not understand because the Mock I create there should accept that I called c() in it, right?Is there a way I can achieve what I want? How can I create a Mock which is capable of being added to another Mock?Or is there another standard way of unittesting code like mine?EDIT: I'm not interested in providing a specialized code which prepares the passed mocks for the expected calls. I only want to check after the call that everything has been happening as expected by examining the passed and returned mock objects. I think this way is supposed to be possible and in this (and other complex cases like this) I could make use of it. 解决方案 Adding objects requires those objects to at least implement __add__, a special method, called magic methods by Mock, see the Mocking Magic Methods section in the documentation:The easiest way to get access to those magic methods that are supported by mock, you can create an instance of the MagicMock class, which provides default implementations for those (each returning a now MagicMock instance by default).This gives you access to the x.a + x.b call:>>> from unittest import mock>>> m = mock.MagicMock()>>> m.a + m.b<MagicMock name='mock.a.__add__()' id='4500141448'>>>> m.mock_calls[call.a.__add__(<MagicMock name='mock.b' id='4500112160'>)]A call to m.a.__add__() has been recorded, with the argument being m.b; this is something we can now assert in a test!Next, that same m.a.__add__() mock is then used to supply the .c() mock:>>> (m.a + m.b).c()<MagicMock name='mock.a.__add__().c()' id='4500162544'>Again, this is something we can assert. Note that if you repeat this call, you'll find that mocks are singletons; when accessing attributes or calling a mock, more mocks of the same type are created and stored, you can later use these stored objects to assert that the right object has been handed out; you can reach the result of a call with the Mock.return_value attribute:>>> m.a.__add__.return_value.c.return_value<MagicMock name='mock.a.__add__().c()' id='4500162544'>>>> (m.a + m.b).c() is m.a.__add__.return_value.c.return_valueTrueNow, on to round(). round() calls a magic method too, the __round__() method. Unfortunately, this is not on the list of supported methods:>>> round(mock.MagicMock())Traceback (most recent call last): File "<stdin>", line 1, in <module>TypeError: type MagicMock doesn't define __round__ methodThis is probably an oversight, since other numeric methods such as __trunc__ and __ceil__ are included. I filed a bug report to request it to be added. You can manually add this to the MagicMock supported methods list with:mock._magics.add('__round__') # set of magic methods MagicMock supports_magics is a set; adding __round__ when it already exists in that set is harmless, so the above is future-proof. An alternative work-around is to mock the round() built-in function, using mock.patch() to set a new round global in the module where your function-under-test is located.Next, when testing, you have 3 options:Drive tests by setting return values for calls, including types other than mocks. For example, you can set up your mock to return a floating point value for the .c() call, so you can assert that you get correctly rounded results: >>> m.a.__add__.return_value.c.return_value = 42.12 # (m.a + ??).c() returns 42.12 >>> round((m.a + m.b).c()) == 42 TrueAssert that specific calls have taken place. There are a whole series of assert_call* methods that help you with testing for a call, all calls, calls in a specific order, etc. There are also attributes such as .called, .call_count, and mock_calls. Do check those out.Asserting that m.a + m.b took place means asserting that m.a.__add__ was called with m.b as an argument:>>> m = mock.MagicMock()>>> m.a + m.b<MagicMock name='mock.a.__add__()' id='4500337776'>>>> m.a.__add__.assert_called_with(m.b) # returns None, so successIf you want to test a Mock instance return value, traverse to the expected mock object, and use is to test for identity:>>> mock._magics.add('__round__')>>> m = mock.MagicMock()>>> r = round((m.a + m.b).c())>>> mock_c_result = m.a.__add__.return_value.c.return_value>>> r is mock_c_result.__round__.return_valueTrueThere is never a need to go back from a mock result to parents, etc. Just traverse the other way.The reason your lambda for __add__ doesn't work is because you created a Mock() instance with arguments. The first two arguments are the spec and the side_effect arguments. The spec argument limits what attributes a mock supports, and since you passed in a as a mock object specification and that a object has no attribute c, you get an attribute error on c. 这篇关于在Python中添加Mock对象的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！