airbyte-source-rki-covid 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

unit_tests/test_stateshistoryrecovered.py

@@ -1,29 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-
-
-from datetime import datetime
-
-from pytest import fixture
-from source_rki_covid.source import StatesHistoryRecovered
-
-
-@fixture
-def patch_states_history_recovered(mocker):
-    # Mock abstract methods to enable instantiating abstract class
-    mocker.patch.object(StatesHistoryRecovered, "primary_key", None)
-
-
-def check_diff(start_date):
-    diff = datetime.now() - datetime.strptime(start_date, "%Y-%m-%d")
-    if diff.days <= 0:
-        return str(1)
-    return str(diff.days)
-
-
-def test_parse_with_cases(patch_states_history_recovered):
-    config = {"start_date": "2022-04-27"}
-    stream = StatesHistoryRecovered(config)
-    expected_stream_path = "states/history/recovered/" + check_diff(config.get("start_date"))
-    assert stream.path() == expected_stream_path

unit_tests/test_stream_agegroup.py

@@ -1,18 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-
-import pytest
-from source_rki_covid.source import GermanyAgeGroups
-
-
-@pytest.fixture
-def patch_age_group(mocker):
-    # Mock abstract methods to enable instantiating abstract class
-    mocker.patch.object(GermanyAgeGroups, "primary_key", None)
-
-
-def test_path(patch_age_group):
-    stream = GermanyAgeGroups()
-    expected_params = {"path": "germany/age-groups"}
-    assert stream.path() == expected_params.get("path")

unit_tests/test_stream_germany.py

@@ -1,18 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-
-import pytest
-from source_rki_covid.source import Germany
-
-
-@pytest.fixture
-def patch_germany_class(mocker):
-    # Mock abstract methods to enable instantiating abstract class
-    mocker.patch.object(Germany, "primary_key", None)
-
-
-def test_path(patch_germany_class):
-    stream = Germany()
-    expected_params = {"path": "germany/"}
-    assert stream.path() == expected_params.get("path")

unit_tests/test_stream_germany_states.py

@@ -1,18 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-
-import pytest
-from source_rki_covid.source import GermanyStates
-
-
-@pytest.fixture
-def patch_germany_states_class(mocker):
-    # Mock abstract methods to enable instantiating abstract class
-    mocker.patch.object(GermanyStates, "primary_key", None)
-
-
-def test_path(patch_germany_states_class):
-    stream = GermanyStates()
-    expected_params = {"path": "states/"}
-    assert stream.path() == expected_params.get("path")

unit_tests/test_streams.py

@@ -1,67 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-
-from http import HTTPStatus
-from unittest.mock import MagicMock
-
-import pytest
-from source_rki_covid.source import RkiCovidStream
-
-
-@pytest.fixture
-def patch_base_class(mocker):
-    # Mock abstract methods to enable instantiating abstract class
-    mocker.patch.object(RkiCovidStream, "path", "v0/example_endpoint")
-    mocker.patch.object(RkiCovidStream, "primary_key", "test_primary_key")
-    mocker.patch.object(RkiCovidStream, "__abstractmethods__", set())
-
-
-def test_request_params(patch_base_class):
-    stream = RkiCovidStream()
-    inputs = {"stream_slice": None, "stream_state": None, "next_page_token": None}
-    expected_params = {}
-    assert stream.request_params(**inputs) == expected_params
-
-
-def test_next_page_token(patch_base_class):
-    stream = RkiCovidStream()
-    inputs = {"response": MagicMock()}
-    expected_token = None
-    assert stream.next_page_token(**inputs) == expected_token
-
-
-def test_request_headers(patch_base_class):
-    stream = RkiCovidStream()
-    inputs = {"stream_slice": None, "stream_state": None, "next_page_token": None}
-    expected_headers = {}
-    assert stream.request_headers(**inputs) == expected_headers
-
-
-def test_http_method(patch_base_class):
-    stream = RkiCovidStream()
-    expected_method = "GET"
-    assert stream.http_method == expected_method
-
-
-@pytest.mark.parametrize(
-    ("http_status", "should_retry"),
-    [
-        (HTTPStatus.OK, False),
-        (HTTPStatus.BAD_REQUEST, False),
-        (HTTPStatus.TOO_MANY_REQUESTS, True),
-        (HTTPStatus.INTERNAL_SERVER_ERROR, True),
-    ],
-)
-def test_should_retry(patch_base_class, http_status, should_retry):
-    response_mock = MagicMock()
-    response_mock.status_code = http_status
-    stream = RkiCovidStream()
-    assert stream.should_retry(response_mock) == should_retry
-
-
-def test_backoff_time(patch_base_class):
-    response_mock = MagicMock()
-    stream = RkiCovidStream()
-    expected_backoff_time = None
-    assert stream.backoff_time(response_mock) == expected_backoff_time

unit_tests/test_streams_states_agegroup.py

@@ -1,18 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-
-import pytest
-from source_rki_covid.source import GermanyStatesAgeGroups
-
-
-@pytest.fixture
-def patch_states_age_group(mocker):
-    # Mock abstract methods to enable instantiating abstract class
-    mocker.patch.object(GermanyStatesAgeGroups, "primary_key", None)
-
-
-def test_path(patch_states_age_group):
-    stream = GermanyStatesAgeGroups()
-    expected_params = {"path": "states/age-groups"}
-    assert stream.path() == expected_params.get("path")