src/catalogue/tests/test_gallery.py

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