forked from mirrors/bookwyrm
129 lines
4.9 KiB
Python
129 lines
4.9 KiB
Python
''' testing book data connectors '''
|
|
import json
|
|
import pathlib
|
|
from unittest.mock import patch
|
|
|
|
from django.test import TestCase
|
|
import responses
|
|
|
|
from bookwyrm import models
|
|
from bookwyrm.connectors.openlibrary import Connector
|
|
from bookwyrm.connectors.openlibrary import get_languages, get_description
|
|
from bookwyrm.connectors.openlibrary import pick_default_edition, \
|
|
get_openlibrary_key
|
|
from bookwyrm.connectors.abstract_connector import SearchResult
|
|
from bookwyrm.connectors.abstract_connector import ConnectorException
|
|
|
|
|
|
class Openlibrary(TestCase):
|
|
''' test loading data from openlibrary.org '''
|
|
def setUp(self):
|
|
''' creates the connector we'll use '''
|
|
models.Connector.objects.create(
|
|
identifier='openlibrary.org',
|
|
name='OpenLibrary',
|
|
connector_file='openlibrary',
|
|
base_url='https://openlibrary.org',
|
|
books_url='https://openlibrary.org',
|
|
covers_url='https://covers.openlibrary.org',
|
|
search_url='https://openlibrary.org/search?q=',
|
|
)
|
|
self.connector = Connector('openlibrary.org')
|
|
|
|
work_file = pathlib.Path(__file__).parent.joinpath(
|
|
'../data/ol_work.json')
|
|
edition_file = pathlib.Path(__file__).parent.joinpath(
|
|
'../data/ol_edition.json')
|
|
edition_list_file = pathlib.Path(__file__).parent.joinpath(
|
|
'../data/ol_edition_list.json')
|
|
self.work_data = json.loads(work_file.read_bytes())
|
|
self.edition_data = json.loads(edition_file.read_bytes())
|
|
self.edition_list_data = json.loads(edition_list_file.read_bytes())
|
|
|
|
|
|
def test_get_remote_id_from_data(self):
|
|
''' format the remote id from the data '''
|
|
data = {'key': '/work/OL1234W'}
|
|
result = self.connector.get_remote_id_from_data(data)
|
|
self.assertEqual(result, 'https://openlibrary.org/work/OL1234W')
|
|
# error handlding
|
|
with self.assertRaises(ConnectorException):
|
|
self.connector.get_remote_id_from_data({})
|
|
|
|
|
|
def test_is_work_data(self):
|
|
''' detect if the loaded json is a work '''
|
|
self.assertEqual(self.connector.is_work_data(self.work_data), True)
|
|
self.assertEqual(self.connector.is_work_data(self.edition_data), False)
|
|
|
|
|
|
@responses.activate
|
|
def test_get_edition_from_work_data(self):
|
|
''' loads a list of editions '''
|
|
data = {'key': '/work/OL1234W'}
|
|
responses.add(
|
|
responses.GET,
|
|
'https://openlibrary.org/work/OL1234W/editions',
|
|
json={'entries': []},
|
|
status=200)
|
|
with patch('bookwyrm.connectors.openlibrary.pick_default_edition') \
|
|
as pick_edition:
|
|
pick_edition.return_value = 'hi'
|
|
result = self.connector.get_edition_from_work_data(data)
|
|
self.assertEqual(result, 'hi')
|
|
|
|
|
|
@responses.activate
|
|
def test_get_work_from_edition_data(self):
|
|
''' loads a list of editions '''
|
|
data = {'works': [{'key': '/work/OL1234W'}]}
|
|
responses.add(
|
|
responses.GET,
|
|
'https://openlibrary.org/work/OL1234W',
|
|
json={'hi': 'there'},
|
|
status=200)
|
|
result = self.connector.get_work_from_edition_data(data)
|
|
self.assertEqual(result, {'hi': 'there'})
|
|
|
|
|
|
def test_pick_default_edition(self):
|
|
''' detect if the loaded json is an edition '''
|
|
edition = pick_default_edition(self.edition_list_data['entries'])
|
|
self.assertEqual(edition['key'], '/books/OL9788823M')
|
|
|
|
|
|
def test_format_search_result(self):
|
|
''' translate json from openlibrary into SearchResult '''
|
|
datafile = pathlib.Path(__file__).parent.joinpath(
|
|
'../data/ol_search.json')
|
|
search_data = json.loads(datafile.read_bytes())
|
|
results = self.connector.parse_search_data(search_data)
|
|
self.assertIsInstance(results, list)
|
|
|
|
result = self.connector.format_search_result(results[0])
|
|
self.assertIsInstance(result, SearchResult)
|
|
self.assertEqual(result.title, 'This Is How You Lose the Time War')
|
|
self.assertEqual(
|
|
result.key, 'https://openlibrary.org/works/OL20639540W')
|
|
self.assertEqual(result.author, 'Amal El-Mohtar, Max Gladstone')
|
|
self.assertEqual(result.year, 2019)
|
|
self.assertEqual(result.connector, self.connector)
|
|
|
|
|
|
def test_get_description(self):
|
|
''' should do some cleanup on the description data '''
|
|
description = get_description(self.work_data['description'])
|
|
expected = 'First in the Old Kingdom/Abhorsen series.'
|
|
self.assertEqual(description, expected)
|
|
|
|
|
|
def test_get_languages(self):
|
|
''' looks up languages from a list '''
|
|
languages = get_languages(self.edition_data['languages'])
|
|
self.assertEqual(languages, ['English'])
|
|
|
|
|
|
def test_get_ol_key(self):
|
|
''' extracts the uuid '''
|
|
key = get_openlibrary_key('/books/OL27320736M')
|
|
self.assertEqual(key, 'OL27320736M')
|