def test_reliable_link_not_found(self, sleep): url = 'test-url' fn = Mock(side_effect=_NotFound) # test wrapped = reliable(fn) self.assertRaises(NotFound, wrapped, None) self.assertFalse(sleep.called)
def test_reliable(self): fn = Mock() messenger = Mock() args = (messenger, 2, 3) kwargs = {'A': 1} # test wrapped = reliable(fn) wrapped(*args, **kwargs) # validation fn.assert_called_once_with(*args, **kwargs)
def test_reliable_link_detached(self, sleep): url = 'test-url' fn = Mock(side_effect=[LinkError, None]) messenger = Mock(url=url, connection=Mock()) args = (messenger, 2, 3) kwargs = {'A': 1} # test wrapped = reliable(fn) wrapped(*args, **kwargs) # validation sleep.assert_called_once_with(DELAY) messenger.repair.assert_called_once_with() self.assertEqual(fn.call_args_list, [ (args, kwargs), (args, kwargs), ])
def test_reliable_link_detached(self, sleep): url = 'test-url' fn = Mock(side_effect=[LinkError, None]) messenger = Mock(url=url, connection=Mock()) args = (messenger, 2, 3) kwargs = {'A': 1} # test wrapped = reliable(fn) wrapped(*args, **kwargs) # validation sleep.assert_called_once_with(DELAY) messenger.repair.assert_called_once_with() self.assertEqual( fn.call_args_list, [ (args, kwargs), (args, kwargs), ])