low-code-connectors: handle single records (#15346)

* handle single records * comment * comment
airbytehq · Aug 5, 2022 · c5c13f0 · c5c13f0
1 parent 5031e72
commit c5c13f0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/airbyte-cdk/python/airbyte_cdk/sources/declarative/extractors/record_selector.py b/airbyte-cdk/python/airbyte_cdk/sources/declarative/extractors/record_selector.py
@@ -39,6 +39,9 @@ def select_records(
         next_page_token: Optional[Mapping[str, Any]] = None,
     ) -> List[Record]:
         all_records = self.extractor.extract_records(response)
+        # Some APIs don't wrap single records in a list
+        if not isinstance(all_records, list):
+            all_records = [all_records]
         if self.record_filter:
             return self.record_filter.filter_records(
                 all_records, stream_state=stream_state, stream_slice=stream_slice, next_page_token=next_page_token

diff --git a/airbyte-cdk/python/unit_tests/sources/declarative/extractors/test_record_selector.py b/airbyte-cdk/python/unit_tests/sources/declarative/extractors/test_record_selector.py
@@ -36,6 +36,13 @@
             {"data": [{"id": 1, "created_at": "06-06-21"}, {"id": 2, "created_at": "06-07-21"}, {"id": 3, "created_at": "06-08-21"}]},
             [{"id": 3, "created_at": "06-08-21"}],
         ),
+        (
+            "test_read_single_record",
+            "_.data",
+            None,
+            {"data": {"id": 1, "created_at": "06-06-21"}},
+            [{"id": 1, "created_at": "06-06-21"}],
+        ),
     ],
 )
 def test_record_filter(test_name, transform_template, filter_template, body, expected_records):