This commit is contained in:
Peter Eckersley
2016-04-01 20:17:06 -07:00
parent 9201f2691f
commit 7fbd800ddd
+8 -8
View File
@@ -46,14 +46,14 @@ class HookTest(unittest.TestCase):
mockwhich.return_value = None mockwhich.return_value = None
self.assertEqual(hooks._prog("funky"), None) self.assertEqual(hooks._prog("funky"), None)
@mock.patch('letsencrypt.hooks.logger') def _test_a_hook(self, config, hook_function, calls_expected):
def _test_a_hook(self, config, hook_function, calls_expected, mock_logger): with mock.patch('letsencrypt.hooks.logger') as mock_logger:
mock_logger.warning = mock.MagicMock() mock_logger.warning = mock.MagicMock()
with mock.patch('letsencrypt.hooks._run_hook') as mock_run_hook: with mock.patch('letsencrypt.hooks._run_hook') as mock_run_hook:
hook_function(config) hook_function(config)
hook_function(config) hook_function(config)
self.assertEqual(mock_run_hook.call_count, calls_expected) self.assertEqual(mock_run_hook.call_count, calls_expected)
return mock_logger.warning return mock_logger.warning
def test_pre_hook(self): def test_pre_hook(self):
config = mock.MagicMock(pre_hook="true") config = mock.MagicMock(pre_hook="true")