61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
#
|
|
# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
|
|
#
|
|
|
|
|
|
from airbyte_cdk.models import SyncMode
|
|
from pytest import fixture
|
|
from source_rss.source import IncrementalRssStream
|
|
|
|
|
|
@fixture
|
|
def patch_incremental_base_class(mocker):
|
|
# Mock abstract methods to enable instantiating abstract class
|
|
mocker.patch.object(IncrementalRssStream, "path", "v0/example_endpoint")
|
|
mocker.patch.object(IncrementalRssStream, "primary_key", "test_primary_key")
|
|
mocker.patch.object(IncrementalRssStream, "__abstractmethods__", set())
|
|
|
|
|
|
def test_cursor_field(patch_incremental_base_class):
|
|
stream = IncrementalRssStream()
|
|
expected_cursor_field = "published"
|
|
assert stream.cursor_field == expected_cursor_field
|
|
|
|
|
|
def test_get_updated_state(patch_incremental_base_class):
|
|
stream = IncrementalRssStream()
|
|
|
|
inputs = {
|
|
"current_stream_state": {"published": "2022-10-24T16:16:00+00:00"},
|
|
"latest_record": {"published": "2022-10-30T16:16:00+00:00"},
|
|
}
|
|
|
|
expected_state = {"published": "2022-10-30T16:16:00+00:00"}
|
|
assert stream.get_updated_state(**inputs) == expected_state
|
|
|
|
|
|
def test_stream_slices(patch_incremental_base_class):
|
|
stream = IncrementalRssStream()
|
|
# TODO: replace this with your input parameters
|
|
inputs = {"sync_mode": SyncMode.incremental, "cursor_field": ["published"], "stream_state": {}}
|
|
# TODO: replace this with your expected stream slices list
|
|
expected_stream_slice = [None]
|
|
assert stream.stream_slices(**inputs) == expected_stream_slice
|
|
|
|
|
|
def test_supports_incremental(patch_incremental_base_class, mocker):
|
|
mocker.patch.object(IncrementalRssStream, "cursor_field", "dummy_field")
|
|
stream = IncrementalRssStream()
|
|
assert stream.supports_incremental
|
|
|
|
|
|
def test_source_defined_cursor(patch_incremental_base_class):
|
|
stream = IncrementalRssStream()
|
|
assert stream.source_defined_cursor
|
|
|
|
|
|
def test_stream_checkpoint_interval(patch_incremental_base_class):
|
|
stream = IncrementalRssStream()
|
|
expected_checkpoint_interval = None
|
|
assert stream.state_checkpoint_interval == expected_checkpoint_interval
|