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_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(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),
])