From ece1f6a24d5abfe8991638ef4c5b5dd61ebbdc79 Mon Sep 17 00:00:00 2001 From: Blake Griffith Date: Tue, 5 Jul 2016 23:28:57 -0500 Subject: [PATCH] Don't use _private functions in tests. --- certbot/tests/error_handler_test.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/certbot/tests/error_handler_test.py b/certbot/tests/error_handler_test.py index a242c4cab..6cbc3c1fe 100644 --- a/certbot/tests/error_handler_test.py +++ b/certbot/tests/error_handler_test.py @@ -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()