I am currently dealing with a class in a module structured like this:
class Foo:
def bar1(self) -> str:
return "bar1"
def bar2(self) -> str:
bar = bar1()
return bar
def bar3(self) -> str:
bar = bar1()
return bar
def bar4(self) -> str:
bar = bar1()
return bar
In another module, I have a test class where I need to mock the bar1 method for each function like barX, except when testing the bar1 method itself. Since there are numerous barX methods and tests, it is not ideal to individually apply the patch decorator above each of the barX test methods (excluding the test_bar1 function).
To address this issue, I decided to patch the bar1 method at the class level using a @patch decorator. However, how can I override this decorator for a specific test method (in this case, my test for bar1())?
Below is my test code:
foo = Foo()
def mock_bar1() -> str:
return "mocked_bar1"
@patch("foo_module.Foo.bar1", mock_bar1)
class FooTest(TestCase):
def test_bar1(self) -> None:
# This fails because bar1() is mocked. I want to disable the mock here.
self.assertEqual("bar1", foo.bar1())
def test_bar2(self) -&> None:
self.assertEqual("mocked_bar1", foo.bar2())
def test_bar3(self) -> None:
self.assertEqual("mocked_bar1", foo.bar3())
def test_bar4(self) -> None:
self.assertEqual("mocked_bar1", foo.bar4())
How can I exclude/negate the patched method for my test_bar1 method?