Partial patch in Python with mock

Question

How can I use mock.patch on a function, so I have an access to methods .assert_called etc. and simultaneously I can still preserve original functionality of the functions?

Here is example code:

from unittest import mock

def foo(arg):
    print(arg)

def tested():
    foo('hi')

@mock.patch('__main__.foo')
def test(foo):
    tested()
    foo.assert_called_once()

test()

I want it to test if the foo function was called just once but I still need it to print hi.

Thyrst · Accepted Answer · 2017-04-16 10:02:25Z

1

Oh. I have it solved already. I'v just needed to add parameter side_effect to the decorator :-)

Like this:

@mock.patch('__main__.foo', side_effect=foo)
def test(foo):
    ...

answered Apr 16, 2017 at 10:02

Thyrst

2,6483 gold badges26 silver badges28 bronze badges

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

Collectives™ on Stack Overflow

Partial patch in Python with mock

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