apps/api/tests.py

   1 # -*- coding: utf-8 -*-
   2
   3 from datetime import datetime
   4
   5 from django.test import TestCase
   6 from django.utils import simplejson as json
   7 from django.conf import settings
   8
   9 from api.helpers import timestamp
  10 from catalogue.models import Book, Tag
  11 from picture.tests.utils import RequestFactory
  12 from picture.forms import PictureImportForm
  13 from picture.models import Picture, picture_storage
  14 import picture.tests
  15 from django.core.files.uploadedfile import SimpleUploadedFile
  16
  17
  18 from os import path
  19
  20
  21 class ApiTest(TestCase):
  22
  23     def setUp(self):
  24         self.old_api_wait = settings.API_WAIT
  25         settings.API_WAIT = -1
  26
  27     def tearDown(self):
  28         settings.API_WAIT = self.old_api_wait
  29
  30
  31 class ChangesTest(ApiTest):
  32
  33     def test_basic(self):
  34         book = Book(title='A Book')
  35         book.save()
  36         tag = Tag.objects.create(category='author', name='Author')
  37         book.tags = [tag]
  38         book.save()
  39
  40         changes = json.loads(self.client.get('/api/changes/0.json?book_fields=title&tag_fields=name').content)
  41         self.assertEqual(changes['updated']['books'],
  42                          [{'id': book.id, 'title': book.title}],
  43                          'Invalid book format in changes')
  44         self.assertEqual(changes['updated']['tags'],
  45                          [{'id': tag.id, 'name': tag.name}],
  46                          'Invalid tag format in changes')
  47
  48
  49 class BookChangesTests(ApiTest):
  50
  51     def setUp(self):
  52         super(BookChangesTests, self).setUp()
  53         self.book = Book.objects.create(slug='slug')
  54
  55     def test_basic(self):
  56         # test book in book_changes.added
  57         changes = json.loads(self.client.get('/api/book_changes/0.json').content)
  58         self.assertEqual(len(changes['updated']),
  59                          1,
  60                          'Added book not in book_changes.updated')
  61
  62     def test_deleted_disappears(self):
  63         # test deleted book disappears
  64         Book.objects.all().delete()
  65         changes = json.loads(self.client.get('/api/book_changes/0.json').content)
  66         self.assertEqual(len(changes), 1,
  67                          'Deleted book should disappear.')
  68
  69     def test_shelf(self):
  70         changed_at = self.book.changed_at
  71
  72         # putting on a shelf should not update changed_at
  73         shelf = Tag.objects.create(category='set', slug='shelf')
  74         self.book.tags = [shelf]
  75         self.assertEqual(self.book.changed_at,
  76                          changed_at)
  77
  78 class TagChangesTests(ApiTest):
  79
  80     def setUp(self):
  81         super(TagChangesTests, self).setUp()
  82         self.tag = Tag.objects.create(category='author')
  83         self.book = Book.objects.create()
  84         self.book.tags = [self.tag]
  85         self.book.save()
  86
  87     def test_added(self):
  88         # test tag in tag_changes.added
  89         changes = json.loads(self.client.get('/api/tag_changes/0.json').content)
  90         self.assertEqual(len(changes['updated']),
  91                          1,
  92                          'Added tag not in tag_changes.updated')
  93
  94     def test_empty_disappears(self):
  95         self.book.tags = []
  96         self.book.save()
  97         changes = json.loads(self.client.get('/api/tag_changes/0.json').content)
  98         self.assertEqual(len(changes), 1,
  99                          'Empty or deleted tag should disappear.')
 100
 101
 102
 103 class BookTests(TestCase):
 104
 105     def setUp(self):
 106         self.tag = Tag.objects.create(category='author', slug='joe')
 107         self.book = Book.objects.create(title='A Book', slug='a-book')
 108         self.book_tagged = Book.objects.create(title='Tagged Book', slug='tagged-book')
 109         self.book_tagged.tags = [self.tag]
 110         self.book_tagged.save()
 111
 112     def test_book_list(self):
 113         books = json.loads(self.client.get('/api/books/').content)
 114         self.assertEqual(len(books), 2,
 115                          'Wrong book list.')
 116
 117     def test_tagged_books(self):
 118         books = json.loads(self.client.get('/api/authors/joe/books/').content)
 119
 120         self.assertEqual([b['title'] for b in books], [self.book_tagged.title],
 121                         'Wrong tagged book list.')
 122
 123     def test_detail(self):
 124         book = json.loads(self.client.get('/api/books/a-book/').content)
 125         self.assertEqual(book['title'], self.book.title,
 126                         'Wrong book details.')
 127
 128
 129 class TagTests(TestCase):
 130
 131     def setUp(self):
 132         self.tag = Tag.objects.create(category='author', slug='joe', name='Joe')
 133         self.book = Book.objects.create(title='A Book', slug='a-book')
 134         self.book.tags = [self.tag]
 135         self.book.save()
 136
 137     def test_tag_list(self):
 138         tags = json.loads(self.client.get('/api/authors/').content)
 139         self.assertEqual(len(tags), 1,
 140                         'Wrong tag list.')
 141
 142     def test_tag_detail(self):
 143         tag = json.loads(self.client.get('/api/authors/joe/').content)
 144         self.assertEqual(tag['name'], self.tag.name,
 145                         'Wrong tag details.')
 146
 147
 148 class PictureTests(ApiTest):
 149     def test_publish(self):
 150         slug = "kandinsky-composition-viii"
 151         xml = SimpleUploadedFile('composition8.xml', open(path.join(picture.tests.__path__[0], "files", slug + ".xml")).read())
 152         img = SimpleUploadedFile('kompozycja-8.png', open(path.join(picture.tests.__path__[0], "files", slug + ".png")).read())
 153
 154         import_form = PictureImportForm({}, {
 155             'picture_xml_file': xml,
 156             'picture_image_file': img
 157             })
 158
 159         assert import_form.is_valid()
 160         if import_form.is_valid():
 161             import_form.save()
 162
 163         pic = Picture.objects.get(slug=slug)
 164