Python mock.patch: replace a method

Question

I'd like to replace a method in a class with mock:

from unittest.mock import patch

class A(object):
    def method(self, string):
        print(self, "method", string)

    def method2(self, string):
        print(self, "method2", string)

with patch.object(A, 'method', side_effect=method2):
    a = A()
    a.method("string")
    a.method.assert_called_with("string")

...but I get insulted by the computer:

TypeError: method2() missing 1 required positional argument: 'string'

dorian · Accepted Answer · 2016-05-15 15:09:31Z

32

The side_effect parameter indicates that a call to method should have a call to method2 as a side effect.

What you probably want is to replace method1 with method2, which you can do by using the new parameter:

with patch.object(A, 'method', new=method2):

Be aware that if you do this, you cannot use assert_called_with, as this is only available on actual Mock objects.

The alternative would be to do away with method2 altogether and just do

with patch.object(A, 'method'):

This will replace method with a Mock instance, which remembers all calls to it and allows you to do assert_called_with.

answered May 15, 2016 at 15:09

dorian

6,3321 gold badge23 silver badges40 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Python mock.patch: replace a method

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related