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() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
@damorosodaragona use this: