moviewyrm/bookwyrm/tests/connectors/test_bookwyrm_connector.py

''' testing book data connectors '''
import json
import pathlib
from django.test import TestCase

from bookwyrm import models
from bookwyrm.connectors.bookwyrm_connector import Connector
from bookwyrm.connectors.abstract_connector import SearchResult


class BookWyrmConnector(TestCase):
    ''' this connector doesn't do much, just search '''
    def setUp(self):
        ''' create the connector '''
        models.Connector.objects.create(
            identifier='example.com',
            connector_file='bookwyrm_connector',
            base_url='https://example.com',
            books_url='https://example.com',
            covers_url='https://example.com/images/covers',
            search_url='https://example.com/search?q=',
        )
        self.connector = Connector('example.com')

        work_file = pathlib.Path(__file__).parent.joinpath(
            '../data/bw_work.json')
        edition_file = pathlib.Path(__file__).parent.joinpath(
            '../data/bw_edition.json')
        self.work_data = json.loads(work_file.read_bytes())
        self.edition_data = json.loads(edition_file.read_bytes())


    def test_format_search_result(self):
        ''' create a SearchResult object from search response json '''
        datafile = pathlib.Path(__file__).parent.joinpath(
            '../data/bw_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, 'Jonathan Strange and Mr Norrell')
        self.assertEqual(result.key, 'https://example.com/book/122')
        self.assertEqual(result.author, 'Susanna Clarke')
        self.assertEqual(result.year, 2017)
        self.assertEqual(result.connector, self.connector)