src/catalogue/tests/test_cover.py

   1 # -*- coding: utf-8 -*-
   2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   4 #
   5 from django.core.files.base import ContentFile
   6 from catalogue.test_utils import BookInfoStub, PersonStub, info_args, WLTestCase
   7 from catalogue.models import Book
   8 from unittest.mock import patch
   9
  10
  11 class CoverTests(WLTestCase):
  12     """Checks in parent_cover_changed is properly called."""
  13     def setUp(self):
  14         WLTestCase.setUp(self)
  15         self.TEXT = """<utwor />"""
  16         self.child = BookInfoStub(
  17             genre='X-Genre',
  18             epoch='X-Epoch',
  19             kind='X-Kind',
  20             author=PersonStub(("Joe",), "Doe"),
  21             **info_args("Child")
  22         )
  23
  24         self.parent = BookInfoStub(
  25             genre='X-Genre',
  26             epoch='X-Epoch',
  27             kind='X-Kind',
  28             author=PersonStub(("Jim",), "Lazy"),
  29             cover_url="http://example.com/cover.jpg",
  30             parts=[self.child.url],
  31             **info_args("Parent")
  32         )
  33
  34     @patch.object(Book, 'parent_cover_changed', autospec=True)
  35     def test_simple_import(self, parent_cover_changed):
  36         child = Book.from_text_and_meta(ContentFile(self.TEXT), self.child)
  37         parent = Book.from_text_and_meta(ContentFile(self.TEXT), self.parent)
  38         parent_cover_changed.assert_called_with(child)
  39
  40         # Now reimport parent.
  41         parent_cover_changed.reset_mock()
  42         parent = Book.from_text_and_meta(ContentFile(self.TEXT), self.parent, overwrite=True)
  43         self.assertEqual(parent_cover_changed.call_count, 0)
  44
  45         # Now change cover in parent.
  46         parent_cover_changed.reset_mock()
  47         self.parent.cover_url = "http://example.com/other-cover.jpg"
  48         parent = Book.from_text_and_meta(ContentFile(self.TEXT), self.parent, overwrite=True)
  49         parent_cover_changed.assert_called_with(child)
  50
  51     @patch.object(Book, 'parent_cover_changed', autospec=True)
  52     def test_change_cover(self, parent_cover_changed):
  53         child = Book.from_text_and_meta(ContentFile(self.TEXT), self.child)
  54         parent = Book.from_text_and_meta(ContentFile(self.TEXT), self.parent)
  55         parent_cover_changed.assert_called_with(child)
  56
  57     @patch.object(Book, 'parent_cover_changed', autospec=True)
  58     def test_new_child(self, parent_cover_changed):
  59         # Add parent without child first.
  60         parts, self.parent.parts = self.parent.parts, []
  61         parent = Book.from_text_and_meta(ContentFile(self.TEXT), self.parent)
  62
  63         # Now import child and reimport parent.
  64         child = Book.from_text_and_meta(ContentFile(self.TEXT), self.child)
  65         self.parent.parts = parts
  66         parent = Book.from_text_and_meta(ContentFile(self.TEXT), self.parent, overwrite=True)
  67         parent_cover_changed.assert_called_with(child)
  68
  69         # Now remove the child.
  70         parent_cover_changed.reset_mock()
  71         self.parent.parts = []
  72         parent = Book.from_text_and_meta(ContentFile(self.TEXT), self.parent, overwrite=True)
  73         parent_cover_changed.assert_called_with(child)