transaction.atomic in importbooks
[wolnelektury.git] / apps / catalogue / tests / visit.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 catalogue import models
6 from catalogue.test_utils import BookInfoStub, PersonStub, WLTestCase, info_args
7 from django.core.files.base import ContentFile
8
9
10 class VisitTest(WLTestCase):
11     """Simply create some objects and visit some views."""
12
13     def setUp(self):
14         WLTestCase.setUp(self)
15         author = PersonStub(("Jane",), "Doe")
16         book_info = BookInfoStub(author=author, genre="Genre",
17             epoch='Epoch', kind="Kind", **info_args(u"A book"))
18         self.book = models.Book.from_text_and_meta(ContentFile('''
19             <utwor>
20             <opowiadanie>
21                 <akap>
22                     <begin id="b1" />
23                     <motyw id="m1">Theme</motyw>
24                     Test
25                     <end id="e1" />
26                 </akap>
27             </opowiadanie>
28             </utwor>
29             '''), book_info)
30         self.collection = models.Collection.objects.create(
31             title='Biblioteczka Boya', slug='boy', book_slugs='a-book')
32
33     def test_visit_urls(self):
34         """ book description should return authors, ancestors, book """
35         url_map = {
36             200: [
37                 '',
38                 'lektury/',
39                 'lektury/boy/',
40                 'nowe/',
41                 'lektura/a-book/',
42                 'lektura/a-book.html',
43                 'lektura/a-book/motyw/theme/',
44                 'motyw/theme/',
45                 'autor/jane-doe/',
46                 'autor/jane-doe/gatunek/genre/',
47                 'autor/jane-doe/gatunek/genre/motyw/theme/',
48                 'pl.json',
49                 'b/%d/mini.pl.html' % self.book.pk,
50                 'b/%d/mini_nolink.pl.html' % self.book.pk,
51                 'b/%d/short.pl.html' % self.book.pk,
52                 'b/%d/wide.pl.html' % self.book.pk,
53                 'f/%d/promo.pl.html' % self.book.fragments.all()[0].pk,
54                 'f/%d/short.pl.html' % self.book.fragments.all()[0].pk,
55                 ],
56             404: [
57                 'lektury/nonexistent/',  # Nonexistent Collection.
58                 'lektura/nonexistent/',  # Nonexistent Book.
59                 'lektura/nonexistent.html',  # Nonexistent Book's HTML.
60                 'lektura/nonexistent/motyw/theme/',  # Nonexistent Book's theme.
61                 'lektura/a-book/motyw/nonexistent/',  # Nonexistent theme in a Book.
62                 'autor/nonexistent/',  # Nonexistent author.
63                 'motyw/nonexistent/',  # Nonexistent theme.
64                 'zh.json',  # Nonexistent language.
65                 'b/%d/mini.pl.html' % (self.book.pk + 100),  # Nonexistent book.
66                 'b/%d/mini_nolink.pl.html' % (self.book.pk + 100),  # Nonexistent book.
67                 'b/%d/short.pl.html' % (self.book.pk + 100),  # Nonexistent book.
68                 'b/%d/wide.pl.html' % (self.book.pk + 100),  # Nonexistent book.
69                 'f/%d/promo.pl.html' % (self.book.fragments.all()[0].pk + 100),  # Nonexistent fragment.
70                 'f/%d/short.pl.html' % (self.book.fragments.all()[0].pk + 100),  # Nonexistent fragment.
71                 ]
72             }
73         prefix = '/katalog/'
74         for expected_status, urls in url_map.items():
75             for url in urls:
76                 print(url)
77                 status = self.client.get(prefix + url).status_code
78                 self.assertEqual(status, expected_status,
79                     "Wrong status code for '%s'. Expected %d, got %d." % (
80                         prefix + url, expected_status, status))