moviewyrm/bookwyrm/tests/connectors/test_connector_manager.py

""" interface between the app and various connectors """
from django.test import TestCase

from bookwyrm import models
from bookwyrm.connectors import connector_manager
from bookwyrm.connectors.bookwyrm_connector import Connector as BookWyrmConnector
from bookwyrm.connectors.self_connector import Connector as SelfConnector


class ConnectorManager(TestCase):
    """ interface between the app and various connectors """

    def setUp(self):
        """ we'll need some books and a connector info entry """
        self.work = models.Work.objects.create(title="Example Work")

        self.edition = models.Edition.objects.create(
            title="Example Edition", parent_work=self.work
        )
        self.work.default_edition = self.edition
        self.work.save()

        self.connector = models.Connector.objects.create(
            identifier="test_connector",
            priority=1,
            local=True,
            connector_file="self_connector",
            base_url="http://test.com/",
            books_url="http://test.com/",
            covers_url="http://test.com/",
        )

    def test_get_or_create_connector(self):
        """ loads a connector if the data source is known or creates one """
        remote_id = "https://example.com/object/1"
        connector = connector_manager.get_or_create_connector(remote_id)
        self.assertIsInstance(connector, BookWyrmConnector)
        self.assertEqual(connector.identifier, "example.com")
        self.assertEqual(connector.base_url, "https://example.com")

        same_connector = connector_manager.get_or_create_connector(remote_id)
        self.assertEqual(connector.identifier, same_connector.identifier)

    def test_get_connectors(self):
        """ load all connectors """
        remote_id = "https://example.com/object/1"
        connector_manager.get_or_create_connector(remote_id)
        connectors = list(connector_manager.get_connectors())
        self.assertEqual(len(connectors), 2)
        self.assertIsInstance(connectors[0], SelfConnector)
        self.assertIsInstance(connectors[1], BookWyrmConnector)

    def test_search(self):
        """ search all connectors """
        results = connector_manager.search("Example")
        self.assertEqual(len(results), 1)
        self.assertIsInstance(results[0]["connector"], SelfConnector)
        self.assertEqual(len(results[0]["results"]), 1)
        self.assertEqual(results[0]["results"][0].title, "Example Edition")

    def test_local_search(self):
        """ search only the local database """
        results = connector_manager.local_search("Example")
        self.assertEqual(len(results), 1)
        self.assertEqual(results[0].title, "Example Edition")

    def test_first_search_result(self):
        """ only get one search result """
        result = connector_manager.first_search_result("Example")
        self.assertEqual(result.title, "Example Edition")
        no_result = connector_manager.first_search_result("dkjfhg")
        self.assertIsNone(no_result)

    def test_load_connector(self):
        """ load a connector object from the database entry """
        connector = connector_manager.load_connector(self.connector)
        self.assertIsInstance(connector, SelfConnector)
        self.assertEqual(connector.identifier, "test_connector")