mirrors/bookwyrm
1
0
Fork 1
mirror of https://github.com/bookwyrm-social/bookwyrm.git synced 2024-09-26 13:30:13 +00:00
Code Issues Projects Releases Wiki Activity

Fixes tests

Browse source
This commit is contained in:
Mouse Reeve 2021-11-16 10:16:28 -08:00
parent 50ca16a46f
commit 6dd7eebd98
2 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
bookwyrm/connectors/inventaire.py
View file

@ -67,7 +67,7 @@ class Connector(AbstractConnector):
extracted = list(data.get("entities").values())
try:
data = extracted[0]
except KeyError:
except (KeyError, IndexError):
raise ConnectorException("Invalid book data")
# flatten the data so that images, uri, and claims are on the same level
return {

10
bookwyrm/tests/connectors/test_inventaire_connector.py
View file

@ -269,10 +269,14 @@ class Inventaire(TestCase):
responses.GET,
"https://inventaire.io/?action=by-uris&uris=hello",
)
data = {"wdt:P629": "hello"}
self.connector.get_work_from_edition_data(data)
data = {"wdt:P629": ["hello"]}
with patch("bookwyrm.connectors.inventaire.Connector.get_book_data") as mock:
self.connector.get_work_from_edition_data(data)
self.assertEqual(mock.call_count, 1)
args = mock.call_args[0]
self.assertEqual(args[0], "https://inventaire.io?action=by-uris&uris=hello")
data = {"wdt:P629": None}
data = {"wdt:P629": [None]}
with self.assertRaises(ConnectorException):
self.connector.get_work_from_edition_data(data)