Fix unit test

Author: lazebnyi

unit_tests/sources/declarative/retrievers/test_state_delegating_retriever.py

@@ -13,13 +13,14 @@
     AirbyteStateType,
     AirbyteStreamState,
     ConfiguredAirbyteCatalog,
+    ConfiguredAirbyteStream,
+    DestinationSyncMode,
     StreamDescriptor,
     Type,
 )
 from airbyte_cdk.sources.declarative.concurrent_declarative_source import (
     ConcurrentDeclarativeSource,
 )
-from airbyte_cdk.sources.embedded.catalog import to_configured_catalog, to_configured_stream
 from airbyte_cdk.test.mock_http import HttpMocker, HttpRequest, HttpResponse
 
 _CONFIG = {"start_date": "2024-07-01T00:00:00.000Z"}
@@ -108,6 +109,28 @@
 }
 
 
+def to_configured_stream(
+    stream,
+    sync_mode=None,
+    destination_sync_mode=DestinationSyncMode.append,
+    cursor_field=None,
+    primary_key=None,
+) -> ConfiguredAirbyteStream:
+    return ConfiguredAirbyteStream(
+        stream=stream,
+        sync_mode=sync_mode,
+        destination_sync_mode=destination_sync_mode,
+        cursor_field=cursor_field,
+        primary_key=primary_key,
+    )
+
+
+def to_configured_catalog(
+    configured_streams,
+) -> ConfiguredAirbyteCatalog:
+    return ConfiguredAirbyteCatalog(streams=configured_streams)
+
+
 def create_configured_catalog(
     source: ConcurrentDeclarativeSource, config: dict
 ) -> ConfiguredAirbyteCatalog: