|
6 | 6 |
|
7 | 7 | import freezegun
|
8 | 8 | import pendulum
|
| 9 | +from airbyte_cdk.models import AirbyteStateBlob |
| 10 | +from airbyte_cdk.models import Level as LogLevel |
| 11 | +from airbyte_cdk.models import SyncMode |
9 | 12 | from airbyte_cdk.test.mock_http import HttpMocker
|
10 | 13 | from airbyte_cdk.test.mock_http.response_builder import FieldPath
|
11 | 14 | from airbyte_cdk.test.state_builder import StateBuilder
|
12 |
| -from airbyte_protocol.models import AirbyteStateBlob |
13 |
| -from airbyte_protocol.models import Level as LogLevel |
14 |
| -from airbyte_protocol.models import SyncMode |
15 | 15 |
|
16 | 16 | from .config import ConfigBuilder
|
17 | 17 | from .helpers import given_post_comments, given_posts, given_ticket_forms
|
@@ -209,7 +209,7 @@ def test_given_no_state_and_successful_sync_when_read_then_set_state_to_now(self
|
209 | 209 | assert len(output.records) == 1
|
210 | 210 |
|
211 | 211 | assert output.most_recent_state.stream_descriptor.name == "post_comment_votes"
|
212 |
| - assert output.most_recent_state.stream_state == AirbyteStateBlob.model_validate({"updated_at": post_comment_votes["updated_at"]}) |
| 212 | + assert output.most_recent_state.stream_state == AirbyteStateBlob({"updated_at": post_comment_votes["updated_at"]}) |
213 | 213 |
|
214 | 214 | @HttpMocker()
|
215 | 215 | def test_given_state_and_pagination_when_read_then_return_records(self, http_mocker):
|
@@ -267,4 +267,4 @@ def test_given_state_and_pagination_when_read_then_return_records(self, http_moc
|
267 | 267 | assert len(output.records) == 2
|
268 | 268 |
|
269 | 269 | assert output.most_recent_state.stream_descriptor.name == "post_comment_votes"
|
270 |
| - assert output.most_recent_state.stream_state == AirbyteStateBlob.model_validate({"updated_at": datetime_to_string(last_page_record_updated_at)}) |
| 270 | + assert output.most_recent_state.stream_state == AirbyteStateBlob({"updated_at": datetime_to_string(last_page_record_updated_at)}) |
0 commit comments