Merge pull request #167 from mouse-reeve/abstract-connector-tests

Tests for abstract connector

fedireads/connectors/abstract_connector.py

@@ -182,7 +182,7 @@ class AbstractConnector(ABC):
     def match_from_mappings(self, data, model):
         ''' try to find existing copies of this book using various keys '''
         relevent_mappings = [m for m in self.key_mappings if \
-                m.model and model == m.model]
+                not m.model or model == m.model]
         for mapping in relevent_mappings:
             # check if this field is present in the data
             value = data.get(mapping.remote_field)

fedireads/tests/test_connector_abstract.py

@@ -0,0 +1,122 @@
+''' testing book data connectors '''
+from django.test import TestCase
+
+from fedireads import models
+from fedireads.connectors.abstract_connector import Mapping,\
+        update_from_mappings
+from fedireads.connectors.fedireads_connector import Connector
+
+
+class FedireadsConnector(TestCase):
+    def setUp(self):
+        self.book = models.Edition.objects.create(title='Example Edition')
+
+        models.Connector.objects.create(
+            identifier='example.com',
+            connector_file='fedireads_connector',
+            base_url='https://example.com',
+            books_url='https:/example.com',
+            covers_url='https://example.com',
+            search_url='https://example.com/search?q=',
+        )
+        self.connector = Connector('example.com')
+
+        self.data = {
+            'title': 'Unused title',
+            'ASIN': 'A00BLAH',
+            'isbn_10': '1234567890',
+            'isbn_13': 'blahhh',
+            'blah': 'bip',
+            'format': 'hardcover',
+            'series': ['one', 'two'],
+        }
+        self.connector.key_mappings = [
+            Mapping('isbn_10', model=models.Edition),
+            Mapping('isbn_13'),
+            Mapping('lccn', model=models.Work),
+            Mapping('asin', remote_field='ASIN'),
+        ]
+
+
+    def test_create_mapping(self):
+        mapping = Mapping('isbn')
+        self.assertEqual(mapping.local_field, 'isbn')
+        self.assertEqual(mapping.remote_field, 'isbn')
+        self.assertEqual(mapping.model, None)
+        self.assertEqual(mapping.formatter('bb'), 'bb')
+
+
+    def test_create_mapping_with_remote(self):
+        mapping = Mapping('isbn', remote_field='isbn13')
+        self.assertEqual(mapping.local_field, 'isbn')
+        self.assertEqual(mapping.remote_field, 'isbn13')
+        self.assertEqual(mapping.model, None)
+        self.assertEqual(mapping.formatter('bb'), 'bb')
+
+
+    def test_create_mapping_with_formatter(self):
+        formatter = lambda x: 'aa' + x
+        mapping = Mapping('isbn', formatter=formatter)
+        self.assertEqual(mapping.local_field, 'isbn')
+        self.assertEqual(mapping.remote_field, 'isbn')
+        self.assertEqual(mapping.formatter, formatter)
+        self.assertEqual(mapping.model, None)
+        self.assertEqual(mapping.formatter('bb'), 'aabb')
+
+
+    def test_update_from_mappings(self):
+        data = {
+            'title': 'Unused title',
+            'isbn_10': '1234567890',
+            'isbn_13': 'blahhh',
+            'blah': 'bip',
+            'format': 'hardcover',
+            'series': ['one', 'two'],
+        }
+        mappings = [
+            Mapping('isbn_10'),
+            Mapping('blah'),# not present on self.book
+            Mapping('physical_format', remote_field='format'),
+            Mapping('series', formatter=lambda x: x[0]),
+        ]
+        book = update_from_mappings(self.book, data, mappings)
+        self.assertEqual(book.title, 'Example Edition')
+        self.assertEqual(book.isbn_10, '1234567890')
+        self.assertEqual(book.isbn_13, None)
+        self.assertEqual(book.physical_format, 'hardcover')
+        self.assertEqual(book.series, 'one')
+
+
+    def test_match_from_mappings(self):
+        edition = models.Edition.objects.create(
+            title='Blah',
+            isbn_13='blahhh',
+        )
+        match = self.connector.match_from_mappings(self.data, models.Edition)
+        self.assertEqual(match, edition)
+
+
+    def test_match_from_mappings_with_model(self):
+        edition = models.Edition.objects.create(
+            title='Blah',
+            isbn_10='1234567890',
+        )
+        match = self.connector.match_from_mappings(self.data, models.Edition)
+        self.assertEqual(match, edition)
+
+
+    def test_match_from_mappings_with_remote(self):
+        edition = models.Edition.objects.create(
+            title='Blah',
+            asin='A00BLAH',
+        )
+        match = self.connector.match_from_mappings(self.data, models.Edition)
+        self.assertEqual(match, edition)
+
+
+    def test_match_from_mappings_no_match(self):
+        edition = models.Edition.objects.create(
+            title='Blah',
+        )
+        match = self.connector.match_from_mappings(self.data, models.Edition)
+        self.assertEqual(match, None)