handle case where stream is not found

Author: lmossman

airbyte_cdk/connector_builder/test_reader/reader.py

@@ -114,14 +114,14 @@ 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
         streams = source.streams(config)
-        stream = next(stream for stream in streams if stream.name == stream_name)
+        stream = next((stream for stream in streams if stream.name == stream_name), None)
 
         # get any deprecation warnings during the component creation
         deprecation_warnings: List[LogMessage] = source.deprecation_warnings()
 
         schema_inferrer = SchemaInferrer(
-            self._pk_to_nested_and_composite_field(stream.primary_key),
-            self._cursor_field_to_nested_and_composite_field(stream.cursor_field),
+            self._pk_to_nested_and_composite_field(stream.primary_key) if stream else None,
+            self._cursor_field_to_nested_and_composite_field(stream.cursor_field) if stream else None,
         )
         datetime_format_inferrer = DatetimeFormatInferrer()