9d0f36f8c65cbf89a56e93fc48a95a07c1deaa51
[redakcja.git] / apps / catalogue / tests / __init__.py
1 from os.path import abspath, dirname, join
2 from nose.tools import *
3 from mock import patch
4 from django.test import TestCase
5 from django.contrib.auth.models import User
6 from catalogue.models import Book, BookPublishRecord
7
8
9 def get_fixture(path):
10     f_path = join(dirname(abspath(__file__)), 'files', path)
11     with open(f_path) as f:
12         return unicode(f.read(), 'utf-8')
13
14
15 class PublishTests(TestCase):
16
17     def setUp(self):
18         self.user = User.objects.create(username='tester')
19         self.text1 = get_fixture('chunk1.xml')
20         self.book = Book.create(self.user, self.text1, slug='test-book')
21
22     @patch('apiclient.api_call')
23     def test_unpublishable(self, api_call):
24         with self.assertRaises(AssertionError):
25             self.book.publish(self.user)
26
27     @patch('apiclient.api_call')
28     def test_publish(self, api_call):
29         self.book[0].head.set_publishable(True)
30         self.book.publish(self.user)
31         api_call.assert_called_with(self.user, 'books/', {"book_xml": self.text1})
32
33     @patch('apiclient.api_call')
34     def test_publish_multiple(self, api_call):
35         self.book[0].head.set_publishable(True)
36         self.book[0].split(slug='part-2')
37         self.book[1].commit(get_fixture('chunk2.xml'))
38         self.book[1].head.set_publishable(True)
39         self.book.publish(self.user)
40         api_call.assert_called_with(self.user, 'books/', {"book_xml": get_fixture('expected.xml')})
41
42
43 class ManipulationTests(TestCase):
44
45     def setUp(self):
46         self.user = User.objects.create(username='tester')
47         self.book1 = Book.create(self.user, 'book 1', slug='book1')
48         self.book2 = Book.create(self.user, 'book 2', slug='book2')
49
50     def test_append(self):
51         self.book1.append(self.book2)
52         self.assertEqual(Book.objects.all().count(), 1)
53         self.assertEqual(len(self.book1), 2)
54
55     def test_append_to_self(self):
56         with self.assertRaises(AssertionError):
57             self.book1.append(Book.objects.get(pk=self.book1.pk))
58         self.assertEqual(Book.objects.all().count(), 2)
59         self.assertEqual(len(self.book1), 1)