Add missing coverage (betodealmeida#453)

ClearTax · Jun 4, 2024 · 70a60df · 70a60df
1 parent 69fa7bf
commit 70a60df
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/shillelagh/adapters/api/generic_json.py b/src/shillelagh/adapters/api/generic_json.py
@@ -142,7 +142,7 @@ def get_data(  # pylint: disable=unused-argument, too-many-arguments
         response = self._session.get(self.uri)
         payload = response.json()
         if not response.ok:
-            raise ProgrammingError(f'Error: {payload["message"]}')
+            raise ProgrammingError(f'Error: {payload["error"]["message"]}')
 
         for i, row in enumerate(jsonpath.findall(self.path, payload)):
             row = {

diff --git a/tests/adapters/api/generic_json_test.py b/tests/adapters/api/generic_json_test.py
@@ -99,12 +99,14 @@ def test_generic_json(requests_mock: Mocker) -> None:
 
     requests_mock.get(
         "https://example.org/data.json",
-        json={"message": "An error occurred"},
+        json={
+            "error": {"code": 1002, "message": "API key is invalid or not provided."},
+        },
         status_code=500,
     )
     with pytest.raises(ProgrammingError) as excinfo:
         list(cursor.execute(sql))
-    assert str(excinfo.value) == "Error: An error occurred"
+    assert str(excinfo.value) == "Error: API key is invalid or not provided."
 
 
 def test_generic_json_complex_type(requests_mock: Mocker) -> None: