src/documents/tests/test_gallery.py

   1 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   3 #
   4 """Tests for galleries of scans."""
   5
   6 from os.path import join, basename, exists
   7 from os import makedirs, listdir
   8 from django.test import TestCase
   9 from django.contrib.auth.models import User
  10 from documents.models import Book
  11 from tempfile import mkdtemp
  12 from django.conf import settings
  13
  14
  15 class GalleryAppendTests(TestCase):
  16     def setUp(self):
  17         self.user = User.objects.create(username='tester')
  18         self.book1 = Book.create(self.user, 'book 1', slug='book1')
  19         self.book1.chunk_set.create(number=2, title='Second chunk',
  20                 slug='book1-2')
  21         c=self.book1[1]
  22         c.gallery_start=3
  23
  24         self.scandir = join(settings.MEDIA_ROOT, settings.IMAGE_DIR)
  25         if not exists(self.scandir):
  26             makedirs(self.scandir)
  27
  28     def make_gallery(self, book, files):
  29         d = mkdtemp('gallery', dir=self.scandir)
  30         for named, cont in files.items():
  31             f = open(join(d, named), 'w')
  32             f.write(cont)
  33             f.close()
  34         book.gallery = basename(d)
  35
  36
  37     def test_both_indexed(self):
  38         self.book2 = Book.create(self.user, 'book 2', slug='book2')
  39         self.book2.chunk_set.create(number=2, title='Second chunk of second book',
  40                 slug='book2-2')
  41
  42         c = self.book2[1]
  43         c.gallery_start = 3
  44         c.save()
  45
  46         self.make_gallery(self.book1, {
  47             '1-0001_1l' : 'aa',
  48             '1-0001_2r' : 'bb',
  49             '1-0002_1l' : 'cc',
  50             '1-0002_2r' : 'dd',
  51             })
  52
  53         self.make_gallery(self.book2, {
  54             '1-0001_1l' : 'dd', # the same, should not be moved
  55             '1-0001_2r' : 'ff',
  56             '2-0002_1l' : 'gg',
  57             '2-0002_2r' : 'hh',
  58             })
  59
  60         self.book1.append(self.book2)
  61
  62         files = listdir(join(self.scandir, self.book1.gallery))
  63         files.sort()
  64         self.assertEqual(files, [
  65             '1-0001_1l',
  66             '1-0001_2r',
  67             '1-0002_1l',
  68             '1-0002_2r',
  69             #            '2-0001_1l',
  70             '2-0001_2r',
  71             '3-0002_1l',
  72             '3-0002_2r',
  73             ])
  74
  75         self.assertEqual((4, 6), (self.book1[2].gallery_start, self.book1[3].gallery_start))
  76
  77
  78     def test_none_indexed(self):
  79         self.book2 = Book.create(self.user, 'book 2', slug='book2')
  80         self.make_gallery(self.book1, {
  81             '0001_1l' : 'aa',
  82             '0001_2r' : 'bb',
  83             '0002_1l' : 'cc',
  84             '0002_2r' : 'dd',
  85             })
  86
  87         self.make_gallery(self.book2, {
  88             '0001_1l' : 'ee',
  89             '0001_2r' : 'ff',
  90             '0002_1l' : 'gg',
  91             '0002_2r' : 'hh',
  92             })
  93
  94         self.book1.append(self.book2)
  95
  96         files = listdir(join(self.scandir, self.book1.gallery))
  97         files.sort()
  98         self.assertEqual(files, [
  99             '0-0001_1l',
 100             '0-0001_2r',
 101             '0-0002_1l',
 102             '0-0002_2r',
 103             '1-0001_1l',
 104             '1-0001_2r',
 105             '1-0002_1l',
 106             '1-0002_2r',
 107             ])
 108
 109
 110     def test_none_indexed(self):
 111         self.book2 = Book.create(self.user, 'book 2', slug='book2')
 112         self.make_gallery(self.book1, {
 113             '1-0001_1l' : 'aa',
 114             '1-0001_2r' : 'bb',
 115             '1002_1l' : 'cc',
 116             '1002_2r' : 'dd',
 117             })
 118
 119         self.make_gallery(self.book2, {
 120             '0001_1l' : 'ee',
 121             '0001_2r' : 'ff',
 122             '0002_1l' : 'gg',
 123             '0002_2r' : 'hh',
 124             })
 125
 126         self.book1.append(self.book2)
 127
 128         files = listdir(join(self.scandir, self.book1.gallery))
 129         files.sort()
 130         self.assertEqual(files, [
 131             '0-1-0001_1l',
 132             '0-1-0001_2r',
 133             '0-1002_1l',
 134             '0-1002_2r',
 135             '1-0001_1l',
 136             '1-0001_2r',
 137             '1-0002_1l',
 138             '1-0002_2r',
 139             ])
 140