Skip to content

Commit e154d08

Browse files
fix: make mock cursor ensure_at_least_one_state_emitted return iterable in substream tests
Co-Authored-By: gl_anatolii.yatsuk <gl_anatolii.yatsuk@airbyte.io>
1 parent 87db0cc commit e154d08

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

unit_tests/sources/declarative/partition_routers/test_substream_partition_router.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1208,6 +1208,7 @@ def test_substream_partition_router_closes_all_partitions_even_when_no_records()
12081208

12091209
mock_cursor = Mock()
12101210
mock_cursor.stream_slices.return_value = []
1211+
mock_cursor.ensure_at_least_one_state_emitted.return_value = []
12111212

12121213
partition_router = SubstreamPartitionRouter(
12131214
parent_stream_configs=[
@@ -1270,6 +1271,7 @@ def test_substream_partition_router_closes_partition_even_when_parent_key_missin
12701271

12711272
mock_cursor = Mock()
12721273
mock_cursor.stream_slices.return_value = []
1274+
mock_cursor.ensure_at_least_one_state_emitted.return_value = []
12731275

12741276
partition_router = SubstreamPartitionRouter(
12751277
parent_stream_configs=[

0 commit comments

Comments
 (0)