Created
July 5, 2012 23:05
-
-
Save santiagobasulto/3056999 to your computer and use it in GitHub Desktop.
Mocking private methods in python
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
""" This is a simple gist to show how to mock | |
private methods. I've got lots of questions | |
regarding this topic. Most people seems confused. | |
Hope it helps. | |
""" | |
import unittest | |
import mock | |
class Car: | |
def __private(self): | |
return 1 | |
def no_private(self): | |
return self.__private() | |
class CarTest(unittest.TestCase): | |
def test_exception_raises(self): | |
c = Car() | |
with self.assertRaises(AttributeError): | |
c.__private() | |
def test_car_works(self): | |
c = Car() | |
self.assertEqual(c.no_private(), 1) | |
def test_mock_private(self): | |
c = Car() | |
with mock.patch.object(c, '_Car__private', return_value=3) as method: | |
c.no_private() | |
method.assert_called_once_with() | |
if __name__ == "__main__": | |
unittest.main() |
@damorosodaragona that's an instance method, and it should work in the same way.
@santiagobasulto if i try to patch self.__private_attribute in the same way that you shown i got "no attribute error" from mock
@damorosodaragona use this:
from mock import patch, PropertyMock
with mock.patch.object(c, '_Car __private_attribute', new_callable=PropertyMock) as mock:
mock.return_value = 4. # the value of the attr
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
There is a way to mock private class attribute?
For example in the following code it's possibile patch
__private_attribute
to set it to 2?