Skip to content

Commit 24d48c1

Browse files
Fix formatting and lint issues
Co-Authored-By: Aaron <AJ> Steers <[email protected]>
1 parent e8541be commit 24d48c1

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

unit_tests/sources/streams/http/test_http.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,8 @@
1010

1111
import pytest
1212
import requests
13+
from requests.exceptions import InvalidURL
14+
1315
from airbyte_cdk.models import AirbyteLogMessage, AirbyteMessage, Level, SyncMode, Type
1416
from airbyte_cdk.sources.streams import CheckpointMixin
1517
from airbyte_cdk.sources.streams.checkpoint import ResumableFullRefreshCursor
@@ -19,7 +21,10 @@
1921
from airbyte_cdk.sources.streams.core import StreamData
2022
from airbyte_cdk.sources.streams.http import HttpStream, HttpSubStream
2123
from airbyte_cdk.sources.streams.http.error_handlers import ErrorHandler, HttpStatusErrorHandler
22-
from airbyte_cdk.sources.streams.http.error_handlers.response_models import FailureType, ResponseAction
24+
from airbyte_cdk.sources.streams.http.error_handlers.response_models import (
25+
FailureType,
26+
ResponseAction,
27+
)
2328
from airbyte_cdk.sources.streams.http.exceptions import (
2429
DefaultBackoffException,
2530
RequestBodyException,
@@ -28,7 +33,6 @@
2833
from airbyte_cdk.sources.streams.http.http_client import MessageRepresentationAirbyteTracedErrors
2934
from airbyte_cdk.sources.streams.http.requests_native_auth import TokenAuthenticator
3035
from airbyte_cdk.utils.airbyte_secrets_utils import update_secrets
31-
from requests.exceptions import InvalidURL
3236

3337

3438
class StubBasicReadHttpStream(HttpStream):
@@ -335,6 +339,7 @@ class StubHttpStreamWithErrorHandler(StubBasicReadHttpStream):
335339
def get_error_handler(self) -> Optional[ErrorHandler]:
336340
return HttpStatusErrorHandler(logging.getLogger())
337341

342+
338343
def test_dns_resolution_error_retry():
339344
"""Test that DNS resolution errors are retried"""
340345
stream = StubHttpStreamWithErrorHandler()

0 commit comments

Comments
 (0)