Don't use _private functions in tests.

This commit is contained in:
Blake Griffith 2016-07-05 23:28:57 -05:00
parent e9f879a443
commit ece1f6a24d

View file

@ -70,7 +70,11 @@ class ErrorHandlerTest(unittest.TestCase):
def test_bad_recovery(self):
bad_func = mock.MagicMock(side_effect=[ValueError])
self.handler.register(bad_func)
self.handler._call_registered()
try:
with self.handler:
raise ValueError
except ValueError:
pass
self.init_func.assert_called_once_with(*self.init_args,
**self.init_kwargs)
bad_func.assert_called_once_with()