get the correct stream in run_test_read

Author: lmossman

airbyte_cdk/connector_builder/connector_builder_handler.py

@@ -108,7 +108,7 @@ def read_stream(
         stream_name = configured_catalog.streams[0].stream.name
 
         stream_read = test_read_handler.run_test_read(
-            source, config, configured_catalog, state, limits.max_records
+            source, config, configured_catalog, stream_name, state, limits.max_records, 
         )
 
         return AirbyteMessage(

airbyte_cdk/connector_builder/test_reader/reader.py

@@ -86,6 +86,7 @@ def run_test_read(
         source: DeclarativeSource,
         config: Mapping[str, Any],
         configured_catalog: ConfiguredAirbyteCatalog,
+        stream_name: str,
         state: List[AirbyteStateMessage],
         record_limit: Optional[int] = None,
     ) -> StreamRead:
@@ -112,7 +113,8 @@ def run_test_read(
 
         record_limit = self._check_record_limit(record_limit)
         # The connector builder currently only supports reading from a single stream at a time
-        stream = source.streams(config)[0]
+        streams = source.streams(config)
+        stream = next(stream for stream in streams if stream.name == stream_name)
 
         # get any deprecation warnings during the component creation
         deprecation_warnings: List[LogMessage] = source.deprecation_warnings()