Skip to content

Commit d2233c5

Browse files
style: apply ruff format to new code
Co-Authored-By: alfredo.garcia@airbyte.io <freddy.garcia7.fg@gmail.com>
1 parent 34cdcb1 commit d2233c5

2 files changed

Lines changed: 15 additions & 5 deletions

File tree

airbyte_cdk/sources/streams/http/http_client.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -591,7 +591,10 @@ def send_request(
591591
mutable_request_kwargs: Dict[str, Any] = {**request_kwargs, **env_settings}
592592

593593
if "timeout" not in mutable_request_kwargs:
594-
mutable_request_kwargs["timeout"] = (self._DEFAULT_CONNECT_TIMEOUT, self._DEFAULT_READ_TIMEOUT)
594+
mutable_request_kwargs["timeout"] = (
595+
self._DEFAULT_CONNECT_TIMEOUT,
596+
self._DEFAULT_READ_TIMEOUT,
597+
)
595598

596599
response: requests.Response = self._send_with_retry(
597600
request=request,

unit_tests/sources/streams/http/test_http_client.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1076,8 +1076,13 @@ def test_send_request_applies_default_timeout_when_not_provided(mocker):
10761076
call_kwargs = mock_send.call_args
10771077
# The timeout should be passed as part of the keyword arguments to session.send()
10781078
# session.send(request, **request_kwargs) unpacks request_kwargs, so timeout appears as a kwarg
1079-
assert call_kwargs.kwargs.get("timeout") == (HttpClient._DEFAULT_CONNECT_TIMEOUT, HttpClient._DEFAULT_READ_TIMEOUT) or \
1080-
call_kwargs[1].get("timeout") == (HttpClient._DEFAULT_CONNECT_TIMEOUT, HttpClient._DEFAULT_READ_TIMEOUT)
1079+
assert call_kwargs.kwargs.get("timeout") == (
1080+
HttpClient._DEFAULT_CONNECT_TIMEOUT,
1081+
HttpClient._DEFAULT_READ_TIMEOUT,
1082+
) or call_kwargs[1].get("timeout") == (
1083+
HttpClient._DEFAULT_CONNECT_TIMEOUT,
1084+
HttpClient._DEFAULT_READ_TIMEOUT,
1085+
)
10811086

10821087

10831088
def test_send_request_respects_explicit_timeout(mocker):
@@ -1097,5 +1102,7 @@ def test_send_request_respects_explicit_timeout(mocker):
10971102
assert mock_send.call_count == 1
10981103
call_kwargs = mock_send.call_args
10991104
# The explicit timeout should be preserved, not overridden by the default
1100-
assert call_kwargs.kwargs.get("timeout") == custom_timeout or \
1101-
call_kwargs[1].get("timeout") == custom_timeout
1105+
assert (
1106+
call_kwargs.kwargs.get("timeout") == custom_timeout
1107+
or call_kwargs[1].get("timeout") == custom_timeout
1108+
)

0 commit comments

Comments
 (0)