diff --git a/airbyte-integrations/connectors/source-rki-covid/source_rki_covid/source.py b/airbyte-integrations/connectors/source-rki-covid/source_rki_covid/source.py index ca43ba02ce4e..32b3bd9567f2 100644 --- a/airbyte-integrations/connectors/source-rki-covid/source_rki_covid/source.py +++ b/airbyte-integrations/connectors/source-rki-covid/source_rki_covid/source.py @@ -109,7 +109,7 @@ def cursor_field(self) -> str: def date_to_int(self, start_date) -> int: diff = datetime.now() - datetime.strptime(start_date, "%Y-%m-%d") - if diff.days == 0: + if diff.days <= 0: return 1 return diff.days @@ -130,7 +130,7 @@ def read_records(self, stream_state: Mapping[str, Any] = None, **kwargs) -> Iter def parse_response(self, response: requests.Response, **kwargs) -> Iterable[Mapping]: if response.json().get("data"): return response.json().get("data") - return [{"error": "data unavailable on date."}] + return [{}] def path( self, stream_state: Mapping[str, Any] = None, stream_slice: Mapping[str, Any] = None, next_page_token: Mapping[str, Any] = None @@ -160,7 +160,7 @@ def cursor_field(self) -> str: def date_to_int(self, start_date) -> int: diff = datetime.now() - datetime.strptime(start_date, "%Y-%m-%d") - if diff.days == 0: + if diff.days <= 0: return 1 return diff.days @@ -181,7 +181,7 @@ def read_records(self, stream_state: Mapping[str, Any] = None, **kwargs) -> Iter def parse_response(self, response: requests.Response, **kwargs) -> Iterable[Mapping]: if response.json().get("data"): return response.json().get("data") - return [{"error": "data unavailable on date."}] + return [{}] def path( self, stream_state: Mapping[str, Any] = None, stream_slice: Mapping[str, Any] = None, next_page_token: Mapping[str, Any] = None @@ -211,7 +211,7 @@ def cursor_field(self) -> str: def date_to_int(self, start_date) -> int: diff = datetime.now() - datetime.strptime(start_date, "%Y-%m-%d") - if diff.days == 0: + if diff.days <= 0: return 1 return diff.days @@ -232,7 +232,7 @@ def read_records(self, stream_state: Mapping[str, Any] = None, **kwargs) -> Iter def parse_response(self, response: requests.Response, **kwargs) -> Iterable[Mapping]: if response.json().get("data"): return response.json().get("data") - return [{"error": "data unavailable on date."}] + return [{}] def path( self, stream_state: Mapping[str, Any] = None, stream_slice: Mapping[str, Any] = None, next_page_token: Mapping[str, Any] = None @@ -262,7 +262,7 @@ def cursor_field(self) -> str: def date_to_int(self, start_date) -> int: diff = datetime.now() - datetime.strptime(start_date, "%Y-%m-%d") - if diff.days == 0: + if diff.days <= 0: return 1 return diff.days @@ -283,7 +283,7 @@ def read_records(self, stream_state: Mapping[str, Any] = None, **kwargs) -> Iter def parse_response(self, response: requests.Response, **kwargs) -> Iterable[Mapping]: if response.json().get("data"): return response.json().get("data") - return [{"error": "data unavailable on date."}] + return [{}] def path( self, stream_state: Mapping[str, Any] = None, stream_slice: Mapping[str, Any] = None, next_page_token: Mapping[str, Any] = None @@ -313,7 +313,7 @@ def cursor_field(self) -> str: def date_to_int(self, start_date) -> int: diff = datetime.now() - datetime.strptime(start_date, "%Y-%m-%d") - if diff.days == 0: + if diff.days <= 0: return 1 return diff.days @@ -334,7 +334,7 @@ def read_records(self, stream_state: Mapping[str, Any] = None, **kwargs) -> Iter def parse_response(self, response: requests.Response, **kwargs) -> Iterable[Mapping]: if response.json().get("data"): return response.json().get("data").get("history") - return [{"error": "data unavailable on date."}] + return [{}] def path( self, stream_state: Mapping[str, Any] = None, stream_slice: Mapping[str, Any] = None, next_page_token: Mapping[str, Any] = None @@ -364,7 +364,7 @@ def cursor_field(self) -> str: def date_to_int(self, start_date) -> int: diff = datetime.now() - datetime.strptime(start_date, "%Y-%m-%d") - if diff.days == 0: + if diff.days <= 0: return 1 return diff.days @@ -385,7 +385,7 @@ def read_records(self, stream_state: Mapping[str, Any] = None, **kwargs) -> Iter def parse_response(self, response: requests.Response, **kwargs) -> Iterable[Mapping]: if response.json().get("data"): return response.json().get("data") - return [{"error": "data unavailable on date."}] + return [{}] def path( self, stream_state: Mapping[str, Any] = None, stream_slice: Mapping[str, Any] = None, next_page_token: Mapping[str, Any] = None diff --git a/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistorycases.py b/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistorycases.py index e59c76341f04..b6a7db10b82e 100644 --- a/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistorycases.py +++ b/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistorycases.py @@ -31,13 +31,13 @@ def test_parse_response(patch_incremental_german_history_cases): expected_response = response.json().get("data") assert stream.parse_response(response) == expected_response else: - expected_response = [{"error": "data unavailable on date."}] + expected_response = [{}] assert stream.parse_response(response) == expected_response def check_diff(start_date): diff = datetime.now() - datetime.strptime(start_date, "%Y-%m-%d") - if diff.days == 0: + if diff.days <= 0: return str(1) return str(diff.days) diff --git a/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistorydeaths.py b/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistorydeaths.py index 59ae8b66449d..0edf0d49382c 100644 --- a/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistorydeaths.py +++ b/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistorydeaths.py @@ -41,13 +41,13 @@ def test_parse_response(patch_incremental_german_history_deaths): expected_response = response.json().get("data") assert stream.parse_response(response) == expected_response else: - expected_response = [{"error": "data unavailable on date."}] + expected_response = [{}] assert stream.parse_response(response) == expected_response def check_diff(start_date): diff = datetime.now() - datetime.strptime(start_date, "%Y-%m-%d") - if diff.days == 0: + if diff.days <= 0: return str(1) return str(diff.days) diff --git a/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryfrozenIncidence.py b/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryfrozenIncidence.py index 6f542b699906..0a8ddc20b1f4 100644 --- a/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryfrozenIncidence.py +++ b/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryfrozenIncidence.py @@ -41,13 +41,13 @@ def test_parse_response(patch_incremental_german_history_frozenInc): expected_response = response.json().get("data").get("history") assert stream.parse_response(response) == expected_response else: - expected_response = [{"error": "data unavailable on date."}] + expected_response = [] assert stream.parse_response(response) == expected_response def check_diff(start_date): diff = datetime.now() - datetime.strptime(start_date, "%Y-%m-%d") - if diff.days == 0: + if diff.days <= 0: return str(1) return str(diff.days) diff --git a/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryhospitalization.py b/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryhospitalization.py index 0c5b8ed1d917..28518e032659 100644 --- a/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryhospitalization.py +++ b/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryhospitalization.py @@ -41,13 +41,13 @@ def test_parse_response(patch_incremental_german_history_hospitalization): expected_response = response.json().get("data") assert stream.parse_response(response) == expected_response else: - expected_response = [{"error": "data unavailable on date."}] + expected_response = [{}] assert stream.parse_response(response) == expected_response def check_diff(start_date): diff = datetime.now() - datetime.strptime(start_date, "%Y-%m-%d") - if diff.days == 0: + if diff.days <= 0: return str(1) return str(diff.days) diff --git a/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryincidence.py b/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryincidence.py index 6c5cbef8676f..8ad19cda94b6 100644 --- a/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryincidence.py +++ b/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryincidence.py @@ -41,13 +41,13 @@ def test_parse_response(patch_incremental_german_history_incidence): expected_response = response.json().get("data") assert stream.parse_response(response) == expected_response else: - expected_response = [{"error": "data unavailable on date."}] + expected_response = [{}] assert stream.parse_response(response) == expected_response def check_diff(start_date): diff = datetime.now() - datetime.strptime(start_date, "%Y-%m-%d") - if diff.days == 0: + if diff.days <= 0: return str(1) return str(diff.days) diff --git a/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryrecovered.py b/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryrecovered.py index f5b4bef29d9d..9cfe1c1d02c9 100644 --- a/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryrecovered.py +++ b/airbyte-integrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryrecovered.py @@ -41,13 +41,13 @@ def test_parse_response(patch_incremental_german_history_recovered): expected_response = response.json().get("data") assert stream.parse_response(response) == expected_response else: - expected_response = [{"error": "data unavailable on date."}] + expected_response = [{}] assert stream.parse_response(response) == expected_response def check_diff(start_date): diff = datetime.now() - datetime.strptime(start_date, "%Y-%m-%d") - if diff.days == 0: + if diff.days <= 0: return str(1) return str(diff.days)