moviewyrm/bookwyrm/tests/connectors/test_openlibrary_connector.py

''' 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'})


    @responses.activate
    def test_get_authors_from_data(self):
        ''' find authors in data '''
        responses.add(
            responses.GET,
            'https://openlibrary.org/authors/OL382982A',
            json={'hi': 'there'},
            status=200)
        results = self.connector.get_authors_from_data(self.work_data)
        for result in results:
            self.assertIsInstance(result, models.Author)


    def test_get_cover_url(self):
        ''' formats a url that should contain the cover image '''
        blob = ['image']
        result = self.connector.get_cover_url(blob)
        self.assertEqual(
            result, 'https://covers.openlibrary.org/b/id/image-M.jpg')

    def test_parse_search_result(self):
        ''' extract the results from the search json response '''
        datafile = pathlib.Path(__file__).parent.joinpath(
            '../data/ol_search.json')
        search_data = json.loads(datafile.read_bytes())
        result = self.connector.parse_search_data(search_data)
        self.assertIsInstance(result, list)
        self.assertEqual(len(result), 2)


    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)


    @responses.activate
    def test_load_edition_data(self):
        ''' format url from key and make request '''
        key = 'OL1234W'
        responses.add(
            responses.GET,
            'https://openlibrary.org/works/OL1234W/editions',
            json={'hi': 'there'}
        )
        result = self.connector.load_edition_data(key)
        self.assertEqual(result, {'hi': 'there'})


    @responses.activate
    def test_expand_book_data(self):
        ''' given a book, get more editions '''
        work = models.Work.objects.create(
            title='Test Work', openlibrary_key='OL1234W')
        edition = models.Edition.objects.create(
            title='Test Edition', parent_work=work)

        responses.add(
            responses.GET,
            'https://openlibrary.org/works/OL1234W/editions',
            json={'entries': []},
        )
        with patch(
                'bookwyrm.connectors.abstract_connector.AbstractConnector.' \
                    'create_edition_from_data'):
            self.connector.expand_book_data(edition)
            self.connector.expand_book_data(work)


    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_openlibrary_key(self):
        ''' extracts the uuid '''
        key = get_openlibrary_key('/books/OL27320736M')
        self.assertEqual(key, 'OL27320736M')


    def test_get_languages(self):
        ''' looks up languages from a list '''
        languages = get_languages(self.edition_data['languages'])
        self.assertEqual(languages, ['English'])


    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')