Patch a module.object with python

Question

Hello I'm trying the following test scenario

def test_main(monkeypatch):
  with patch("library.mockme.MockMe") as mock:
    message = MainObejct().message()
    assert message == "Mock me"

MainObject implementation

from library.mockme import MockMe

class MainObejct():
  def __init__(self):
    self.mock_me = MockMe()

  def message(self):
    return self.mock_me.message

The problem here is that the MockMe object is not patched... but if I change the import to from library import mockme.MockMe it actually works, is there a way to make it work with my original implementation?

Thanks!

Owen · Accepted Answer · 2020-03-13 23:54:46Z

4

Hi this is all about where and what to patch. In your example you are mocking the MockMe class in the the mockme module. You need to mock the class that is imported into your main.py module. Have a look at where to patch in the Python docs.

Hope this helps!

test_main.py

from main import MainObejct


def test_main(mocker):
    m_mockerme = mocker.patch("main.MockMe")
    m_mockerme.return_value.hello.return_value = "goodbye"
    message = MainObejct().message()
    assert message == "goodbye"

main.py

from library.mockme import MockMe


class MainObejct:
    def __init__(self):
        self.mock_me = MockMe()

    def message(self):
        return self.mock_me.hello()

library/mockme.py

class MockMe:
    def hello(self):
        return "hello"

edited Mar 13, 2020 at 23:54

answered Mar 13, 2020 at 17:18

Owen

1084 bronze badges

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

1 Comment

hoefling Over a year ago

@AlvinEstrada if it answers your question, you should accept and upvote it, see stackoverflow.com/help/someone-answers

Collectives™ on Stack Overflow

Patch a module.object with python

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related