def test_ticket_metrics_timeout_error_without_parameter( self, mock_get, mock_sleep): """We mock request method to raise a `Timeout` and expect the tap to retry this up to 5 times when `request_timeout` does not passed, """ mock_get.side_effect = requests.exceptions.Timeout ticket_metrics = streams.TicketMetrics(config={ 'subdomain': '34', 'access_token': 'df' }) try: responses = list(ticket_metrics.sync('i1')) except requests.exceptions.Timeout as e: pass # Verify the request retry 5 times on timeout self.assertEqual(mock_get.call_count, 5)
def test_ticket_metrics_timeout_error_with_float_value( self, mock_get, mock_sleep): """We mock request method to raise a `Timeout` and expect the tap to retry this up to 5 times when int value of `request_timeout` passed, """ mock_get.side_effect = requests.exceptions.Timeout ticket_metrics = streams.TicketMetrics( config={ 'subdomain': '34', 'access_token': 'df', 'request_timeout': REQUEST_TIMEOUT_FLOAT }) try: responses = list(ticket_metrics.sync('i1')) except requests.exceptions.Timeout as e: pass # Verify the request retry 5 times on timeout self.assertEqual(mock_get.call_count, 5)