Skip to content

Commit e8765eb

Browse files
committed
CDK: add test
Signed-off-by: Artem Inzhyyants <[email protected]>
1 parent cfe9895 commit e8765eb

File tree

1 file changed

+29
-0
lines changed

1 file changed

+29
-0
lines changed
Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
from airbyte_cdk.sources.streams.concurrent.cursor import CursorField
2+
from airbyte_cdk.sources.streams.concurrent.state_converters.incrementing_count_stream_state_converter import (
3+
IncrementingCountStreamStateConverter,
4+
)
5+
6+
7+
def test_convert_from_sequential_state():
8+
converter = IncrementingCountStreamStateConverter(
9+
is_sequential_state=True,
10+
)
11+
12+
_, conversion = converter.convert_from_sequential_state(CursorField("id"), {"id": 12345}, 0)
13+
14+
assert conversion["state_type"] == "integer"
15+
assert conversion["legacy"] == {"id": 12345}
16+
assert len(conversion["slices"]) == 1
17+
assert conversion["slices"][0] == {"end": 12345, "most_recent_cursor_value": 12345, "start": 0}
18+
19+
20+
def test_convert_to_sequential_state():
21+
converter = IncrementingCountStreamStateConverter(
22+
is_sequential_state=True,
23+
)
24+
concurrent_state = {
25+
"legacy": {"id": 12345},
26+
"slices": [{"end": 12345, "most_recent_cursor_value": 12345, "start": 0}],
27+
"state_type": "integer",
28+
}
29+
assert converter.convert_to_state_message(CursorField("id"), concurrent_state) == {"id": 12345}

0 commit comments

Comments
 (0)