moviewyrm/bookwyrm/tests/models/test_book_model.py

''' testing models '''
from dateutil.parser import parse
from django.test import TestCase
from django.utils import timezone

from bookwyrm import models, settings
from bookwyrm.models.book import isbn_10_to_13, isbn_13_to_10


class Book(TestCase):
    ''' not too much going on in the books model but here we are '''
    def setUp(self):
        ''' we'll need some books '''
        self.work = models.Work.objects.create(
            title='Example Work',
            remote_id='https://example.com/book/1'
        )
        self.first_edition = models.Edition.objects.create(
            title='Example Edition',
            parent_work=self.work,
        )
        self.second_edition = models.Edition.objects.create(
            title='Another Example Edition',
            parent_work=self.work,
        )

    def test_remote_id(self):
        ''' fanciness with remote/origin ids '''
        remote_id = 'https://%s/book/%d' % (settings.DOMAIN, self.work.id)
        self.assertEqual(self.work.get_remote_id(), remote_id)
        self.assertEqual(self.work.remote_id, remote_id)

    def test_create_book(self):
        ''' you shouldn't be able to create Books (only editions and works) '''
        self.assertRaises(
            ValueError,
            models.Book.objects.create,
            title='Invalid Book'
        )

    def test_isbn_10_to_13(self):
        ''' checksums and so on '''
        isbn_10 = '178816167X'
        isbn_13 = isbn_10_to_13(isbn_10)
        self.assertEqual(isbn_13, '9781788161671')

        isbn_10 = '1-788-16167-X'
        isbn_13 = isbn_10_to_13(isbn_10)
        self.assertEqual(isbn_13, '9781788161671')


    def test_isbn_13_to_10(self):
        ''' checksums and so on '''
        isbn_13 = '9781788161671'
        isbn_10 = isbn_13_to_10(isbn_13)
        self.assertEqual(isbn_10, '178816167X')

        isbn_13 = '978-1788-16167-1'
        isbn_10 = isbn_13_to_10(isbn_13)
        self.assertEqual(isbn_10, '178816167X')


    def test_get_edition_info(self):
        ''' text slug about an edition '''
        book = models.Edition.objects.create(title='Test Edition')
        self.assertEqual(book.edition_info, '')

        book.physical_format = 'worm'
        book.save()
        self.assertEqual(book.edition_info, 'worm')

        book.languages = ['English']
        book.save()
        self.assertEqual(book.edition_info, 'worm')

        book.languages = ['Glorbish', 'English']
        book.save()
        self.assertEqual(book.edition_info, 'worm, Glorbish language')

        book.published_date = timezone.make_aware(parse('2020'))
        book.save()
        self.assertEqual(book.edition_info, 'worm, Glorbish language, 2020')
        self.assertEqual(
            book.alt_text, 'Test Edition cover (worm, Glorbish language, 2020)')


    def test_get_rank(self):
        ''' sets the data quality index for the book '''
        # basic rank
        self.assertEqual(self.first_edition.edition_rank, 0)

        self.first_edition.description = 'hi'
        self.first_edition.save()
        self.assertEqual(self.first_edition.edition_rank, 1)

        # default edition
        self.work.default_edition = self.first_edition
        self.work.save()
        self.first_edition.refresh_from_db()
        self.assertEqual(self.first_edition.edition_rank, 20)