Auto-fix lint and format issues

airbytehq · Dec 18, 2024 · ec9e480 · ec9e480
1 parent fb21cd4
commit ec9e480
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 17 deletions.
diff --git a/airbyte_cdk/sources/declarative/resolvers/http_components_resolver.py b/airbyte_cdk/sources/declarative/resolvers/http_components_resolver.py
@@ -89,7 +89,9 @@ def resolve_components(
         kwargs = {"stream_template_config": stream_template_config}
 
         for stream_slice in self.retriever.stream_slices():
-            for components_values in self.retriever.read_records(records_schema={}, stream_slice=stream_slice):
+            for components_values in self.retriever.read_records(
+                records_schema={}, stream_slice=stream_slice
+            ):
                 updated_config = deepcopy(stream_template_config)
                 kwargs["components_values"] = components_values  # type: ignore[assignment] # component_values will always be of type Mapping[str, Any]
                 kwargs["stream_slice"] = stream_slice  # type: ignore[assignment] # stream_slice will always be of type Mapping[str, Any]

diff --git a/unit_tests/sources/declarative/resolvers/test_http_components_resolver.py b/unit_tests/sources/declarative/resolvers/test_http_components_resolver.py
@@ -284,23 +284,24 @@
                                         },
                                         "record_selector": {
                                             "type": "RecordSelector",
-                                            "extractor": {"type": "DpathExtractor", "field_path": []},
+                                            "extractor": {
+                                                "type": "DpathExtractor",
+                                                "field_path": [],
+                                            },
                                         },
                                     },
                                     "schema_loader": {
                                         "type": "InlineSchemaLoader",
                                         "schema": {
                                             "$schema": "http://json-schema.org/schema#",
-                                            "properties": {
-                                                "id": {"type": "integer"}
-                                            },
+                                            "properties": {"id": {"type": "integer"}},
                                             "type": "object",
                                         },
                                     },
-                                }
+                                },
                             }
-                        ]
-                    }
+                        ],
+                    },
                 },
                 "components_mapping": [
                     {
@@ -322,6 +323,8 @@
         }
     ],
 }
+
+
 @pytest.mark.parametrize(
     "components_mapping, retriever_data, stream_template_config, expected_result",
     [
@@ -358,6 +361,7 @@ def test_http_components_resolver(
     result = list(resolver.resolve_components(stream_template_config=stream_template_config))
     assert result == expected_result
 
+
 @pytest.mark.parametrize(
     "components_mapping, retriever_data, stream_template_config, expected_result",
     [
@@ -394,6 +398,7 @@ def test_http_components_resolver_with_stream_slices(
     result = list(resolver.resolve_components(stream_template_config=stream_template_config))
     assert result == expected_result
 
+
 def test_dynamic_streams_read_with_http_components_resolver():
     expected_stream_names = ["item_1", "item_2"]
     with HttpMocker() as http_mocker:
@@ -466,16 +471,18 @@ def test_duplicated_dynamic_streams_read_with_http_components_resolver():
             == "Dynamic streams list contains a duplicate name: item_2. Please contact Airbyte Support."
         )
 
+
 def test_dynamic_streams_with_http_components_resolver_retriever_with_parent_stream():
     expected_stream_names = [
-        "parent_1_item_1", "parent_1_item_2", "parent_2_item_1", "parent_2_item_2"
+        "parent_1_item_1",
+        "parent_1_item_2",
+        "parent_2_item_1",
+        "parent_2_item_2",
     ]
     with HttpMocker() as http_mocker:
         http_mocker.get(
             HttpRequest(url="https://api.test.com/parents"),
-            HttpResponse(
-                body=json.dumps([{"id": 1}, {"id": 2}])
-            ),
+            HttpResponse(body=json.dumps([{"id": 1}, {"id": 2}])),
         )
         parent_ids = [1, 2]
         for parent_id in parent_ids:
@@ -494,13 +501,14 @@ def test_dynamic_streams_with_http_components_resolver_retriever_with_parent_str
         for dynamic_stream_path in dynamic_stream_paths:
             http_mocker.get(
                 HttpRequest(url=f"https://api.test.com/{dynamic_stream_path}"),
-                HttpResponse(
-                    body=json.dumps([{"ABC": 1, "AED": 2}])
-                ),
+                HttpResponse(body=json.dumps([{"ABC": 1, "AED": 2}])),
             )
 
         source = ConcurrentDeclarativeSource(
-            source_config=_MANIFEST_WITH_HTTP_COMPONENT_RESOLVER_WITH_RETRIEVER_WITH_PARENT_STREAM, config=_CONFIG, catalog=None, state=None
+            source_config=_MANIFEST_WITH_HTTP_COMPONENT_RESOLVER_WITH_RETRIEVER_WITH_PARENT_STREAM,
+            config=_CONFIG,
+            catalog=None,
+            state=None,
         )
 
         actual_catalog = source.discover(logger=source.logger, config=_CONFIG)
@@ -520,4 +528,4 @@ def test_dynamic_streams_with_http_components_resolver_retriever_with_parent_str
     assert len(actual_catalog.streams) == 4
     assert [stream.name for stream in actual_catalog.streams] == expected_stream_names
     assert len(records) == 4
-    assert [record.stream for record in records] == expected_stream_names
+    assert [record.stream for record in records] == expected_stream_names