Don't use _private functions in tests.

This commit is contained in:
Blake Griffith
2016-07-05 23:32:22 -05:00
parent e9f879a443
commit ece1f6a24d
+5 -1
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()