fix tests

Author: brianjlai

airbyte_cdk/sources/file_based/stream/concurrent/adapters.py

@@ -32,6 +32,7 @@
 from airbyte_cdk.sources.message import MessageRepository
 from airbyte_cdk.sources.source import ExperimentalClassWarning
 from airbyte_cdk.sources.streams.concurrent.abstract_stream_facade import AbstractStreamFacade
+from airbyte_cdk.sources.streams.concurrent.cursor import CursorField
 from airbyte_cdk.sources.streams.concurrent.default_stream import DefaultStream
 from airbyte_cdk.sources.streams.concurrent.exceptions import ExceptionWithDisplayMessage
 from airbyte_cdk.sources.streams.concurrent.helpers import (
@@ -97,7 +98,7 @@ def create_from_stream(
                 name=stream.name,
                 json_schema=stream.get_json_schema(),
                 primary_key=pk,
-                cursor_field=cursor_field,
+                cursor_field=CursorField(cursor_field_key=cursor_field) if cursor_field else None,
                 logger=logger,
                 namespace=stream.namespace,
                 cursor=cursor,

airbyte_cdk/sources/streams/concurrent/adapters.py

@@ -25,7 +25,7 @@
 from airbyte_cdk.sources.source import ExperimentalClassWarning
 from airbyte_cdk.sources.streams import Stream
 from airbyte_cdk.sources.streams.concurrent.abstract_stream_facade import AbstractStreamFacade
-from airbyte_cdk.sources.streams.concurrent.cursor import Cursor, FinalStateCursor
+from airbyte_cdk.sources.streams.concurrent.cursor import Cursor, CursorField, FinalStateCursor
 from airbyte_cdk.sources.streams.concurrent.default_stream import DefaultStream
 from airbyte_cdk.sources.streams.concurrent.exceptions import ExceptionWithDisplayMessage
 from airbyte_cdk.sources.streams.concurrent.helpers import (
@@ -97,7 +97,7 @@ def create_from_stream(
                 namespace=stream.namespace,
                 json_schema=stream.get_json_schema(),
                 primary_key=pk,
-                cursor_field=cursor_field,
+                cursor_field=CursorField(cursor_field_key=cursor_field) if cursor_field else None,
                 logger=logger,
                 cursor=cursor,
             ),

unit_tests/sources/streams/concurrent/test_adapters.py

@@ -22,7 +22,7 @@
     StreamPartition,
     StreamPartitionGenerator,
 )
-from airbyte_cdk.sources.streams.concurrent.cursor import Cursor
+from airbyte_cdk.sources.streams.concurrent.cursor import Cursor, CursorField
 from airbyte_cdk.sources.streams.concurrent.exceptions import ExceptionWithDisplayMessage
 from airbyte_cdk.sources.streams.core import Stream
 from airbyte_cdk.sources.types import Record
@@ -328,6 +328,7 @@ def test_create_from_stream_stream(self):
 
         assert facade.name == "stream"
         assert facade.cursor_field == "cursor"
+        # assert facade.cursor_field == CursorField(cursor_field_key="cursor")
         assert facade._abstract_stream._primary_key == ["id"]
 
     def test_create_from_stream_stream_with_none_primary_key(self):

unit_tests/sources/streams/concurrent/test_default_stream.py

@@ -8,7 +8,7 @@
 
 from airbyte_cdk.models import AirbyteStream, SyncMode
 from airbyte_cdk.sources.message import InMemoryMessageRepository
-from airbyte_cdk.sources.streams.concurrent.cursor import Cursor, FinalStateCursor
+from airbyte_cdk.sources.streams.concurrent.cursor import Cursor, CursorField, FinalStateCursor
 from airbyte_cdk.sources.streams.concurrent.default_stream import DefaultStream
 from airbyte_cdk.sources.streams.concurrent.partitions.partition import Partition
 from airbyte_cdk.sources.streams.concurrent.partitions.partition_generator import PartitionGenerator
@@ -187,7 +187,7 @@ def test_as_airbyte_stream_with_a_cursor(self):
             self._name,
             json_schema,
             self._primary_key,
-            "date",
+            CursorField(cursor_field_key="date"),
             self._logger,
             FinalStateCursor(
                 stream_name=self._name,