def test_connects_directly(self, sleep):
connector = Mock()
data_transport = DataTransport(connector)
data_transport.send = Mock()
data_transport.deliver_event(Event("test"))
connector.connect.assert_called_once()
assert sleep.call_count == 0
def test_connects_after_retries(self, sleep):
data_transport = DataTransport(
FailsAfterNTimes(2, with_exception=ConnectionError))
data_transport.send = Mock()
event = Event("test")
data_transport.deliver_event(event)
assert sleep.call_count == DataTransport.retry_n_times - 1, sleep.call_count
def test_connection_error(self, sleep):
data_transport = DataTransport(
Mock(connect=Mock(side_effect=ConnectionError)))
self.assertRaisesRegex(
ConnectionError,
"Couldn't connect after \d+ times",
data_transport.deliver_event,
Event("connection error"),
)
assert sleep.call_count == DataTransport._RETRY_TIMES
def test_connects_after_retries(self, sleep):
data_transport = DataTransport(
FailsAfterNTimes(2, with_exception=ConnectionError))
data_transport.send = Mock()
event = Event("test")
data_transport.deliver_event(event)
data_transport.send.assert_called_once_with(event)
assert (sleep.call_count == DataTransport._RETRY_TIMES -
1), sleep.call_count