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