Skip to content

Commit 555646d

Browse files
author
octavia-squidington-iii
committed
Auto-fix lint and format issues
1 parent 6e79ecf commit 555646d

File tree

5 files changed

+33
-15
lines changed

5 files changed

+33
-15
lines changed

airbyte_cdk/sources/declarative/decoders/json_decoder.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ def decode(
4444
if not has_yielded:
4545
yield {}
4646

47+
4748
@dataclass
4849
class IterableDecoder(Decoder):
4950
"""

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2037,13 +2037,15 @@ def create_json_decoder(model: JsonDecoderModel, config: Config, **kwargs: Any)
20372037

20382038
@staticmethod
20392039
def create_csv_decoder(model: CsvDecoderModel, config: Config, **kwargs: Any) -> Decoder:
2040-
return CompositeRawDecoder(parser=ModelToComponentFactory._get_parser(model, config), stream_response=False)
2040+
return CompositeRawDecoder(
2041+
parser=ModelToComponentFactory._get_parser(model, config), stream_response=False
2042+
)
20412043

20422044
@staticmethod
2043-
def create_jsonl_decoder(
2044-
model: JsonlDecoderModel, config: Config, **kwargs: Any
2045-
) -> Decoder:
2046-
return CompositeRawDecoder(parser=ModelToComponentFactory._get_parser(model, config), stream_response=True)
2045+
def create_jsonl_decoder(model: JsonlDecoderModel, config: Config, **kwargs: Any) -> Decoder:
2046+
return CompositeRawDecoder(
2047+
parser=ModelToComponentFactory._get_parser(model, config), stream_response=True
2048+
)
20472049

20482050
@staticmethod
20492051
def create_iterable_decoder(
@@ -2070,8 +2072,12 @@ def _get_parser(model: BaseModel, config: Config) -> Parser:
20702072
elif isinstance(model, CsvDecoderModel):
20712073
return CsvParser(encoding=model.encoding, delimiter=model.delimiter)
20722074
elif isinstance(model, GzipDecoderModel):
2073-
return GzipParser(inner_parser=ModelToComponentFactory._get_parser(model.inner_decoder, config))
2074-
elif isinstance(model, (CustomDecoderModel, IterableDecoderModel, XmlDecoderModel, ZipfileDecoderModel)):
2075+
return GzipParser(
2076+
inner_parser=ModelToComponentFactory._get_parser(model.inner_decoder, config)
2077+
)
2078+
elif isinstance(
2079+
model, (CustomDecoderModel, IterableDecoderModel, XmlDecoderModel, ZipfileDecoderModel)
2080+
):
20752081
raise ValueError(f"Decoder type {model} does not have parser associated to it")
20762082

20772083
raise ValueError(f"Unknown decoder type {model}")

unit_tests/sources/declarative/auth/test_token_provider.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,9 +55,9 @@ def test_session_token_provider_cache_expiration():
5555
provider = create_session_token_provider()
5656
provider.get_token()
5757

58-
provider.login_requester.send_request.return_value.content = json.dumps({
59-
"nested": {"token": "updated_token"}
60-
}).encode()
58+
provider.login_requester.send_request.return_value.content = json.dumps(
59+
{"nested": {"token": "updated_token"}}
60+
).encode()
6161

6262
with freezegun.freeze_time("2001-05-21T14:00:00Z"):
6363
assert provider.get_token() == "updated_token"

unit_tests/sources/declarative/decoders/test_composite_decoder.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,9 @@ def test_composite_raw_decoder_csv_parser_values(requests_mock, encoding: str, d
201201

202202

203203
def test_given_response_already_consumed_when_decode_then_no_data_is_returned(requests_mock):
204-
requests_mock.register_uri("GET", "https://airbyte.io/", content=json.dumps({"test": "test"}).encode())
204+
requests_mock.register_uri(
205+
"GET", "https://airbyte.io/", content=json.dumps({"test": "test"}).encode()
206+
)
205207
response = requests.get("https://airbyte.io/", stream=True)
206208
composite_raw_decoder = CompositeRawDecoder(parser=JsonParser(encoding="utf-8"))
207209

@@ -212,10 +214,16 @@ def test_given_response_already_consumed_when_decode_then_no_data_is_returned(re
212214
list(composite_raw_decoder.decode(response))
213215

214216

215-
def test_given_response_is_not_streamed_when_decode_then_can_be_called_multiple_times(requests_mock):
216-
requests_mock.register_uri("GET", "https://airbyte.io/", content=json.dumps({"test": "test"}).encode())
217+
def test_given_response_is_not_streamed_when_decode_then_can_be_called_multiple_times(
218+
requests_mock,
219+
):
220+
requests_mock.register_uri(
221+
"GET", "https://airbyte.io/", content=json.dumps({"test": "test"}).encode()
222+
)
217223
response = requests.get("https://airbyte.io/")
218-
composite_raw_decoder = CompositeRawDecoder(parser=JsonParser(encoding="utf-8"), stream_response=False)
224+
composite_raw_decoder = CompositeRawDecoder(
225+
parser=JsonParser(encoding="utf-8"), stream_response=False
226+
)
219227

220228
content = list(composite_raw_decoder.decode(response))
221229
content_second_time = list(composite_raw_decoder.decode(response))

unit_tests/sources/declarative/decoders/test_json_decoder.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,10 @@ def test_json_decoder(requests_mock, response_body, first_element):
4242
def test_jsonl_decoder(requests_mock, response_body, expected_json):
4343
requests_mock.register_uri("GET", "https://airbyte.io/", text=response_body)
4444
response = requests.get("https://airbyte.io/", stream=True)
45-
assert list(CompositeRawDecoder(parser=JsonLineParser(), stream_response=True).decode(response)) == expected_json
45+
assert (
46+
list(CompositeRawDecoder(parser=JsonLineParser(), stream_response=True).decode(response))
47+
== expected_json
48+
)
4649

4750

4851
@pytest.fixture(name="large_events_response")

0 commit comments

Comments
 (0)