mirror of
https://github.com/bookwyrm-social/bookwyrm.git
synced 2024-10-31 22:19:00 +00:00
56 lines
2.4 KiB
Python
56 lines
2.4 KiB
Python
""" testing book data connectors """
|
|
from unittest.mock import patch
|
|
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")
|
|
|
|
def test_get_or_create_book_existing(self):
|
|
"""load book activity"""
|
|
with patch("bookwyrm.preview_images.generate_edition_preview_image_task.delay"):
|
|
work = models.Work.objects.create(title="Test Work")
|
|
book = models.Edition.objects.create(title="Test Edition", parent_work=work)
|
|
result = self.connector.get_or_create_book(book.remote_id)
|
|
self.assertEqual(book, result)
|
|
|
|
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)
|
|
|
|
def test_format_isbn_search_result(self):
|
|
"""just gotta attach the connector"""
|
|
datafile = pathlib.Path(__file__).parent.joinpath("../data/bw_search.json")
|
|
search_data = json.loads(datafile.read_bytes())
|
|
results = self.connector.parse_isbn_search_data(search_data)
|
|
result = self.connector.format_isbn_search_result(results[0])
|
|
self.assertEqual(result.connector, self.connector)
|