updated unit tests.

airbytehq · Aug 25, 2022 · 249e450 · 249e450
1 parent e2ff0e8
commit 249e450
Show file tree

Hide file tree

Showing 7 changed files with 24 additions and 24 deletions.
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
@@ -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/...ntegrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistorycases.py b/...ntegrations/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/...tegrations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistorydeaths.py b/...tegrations/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/...s/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryfrozenIncidence.py b/...s/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/...s/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryhospitalization.py b/...s/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/...rations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryincidence.py b/...rations/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/...rations/connectors/source-rki-covid/unit_tests/test_incremental_germanhistoryrecovered.py b/...rations/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)