Fix error during report generation

[wolnelektury.git] / apps / api / tests.py

diff --git a/apps/api/tests.py b/apps/api/tests.py
index 5a981a2..0a638dc 100644 (file)
--- a/apps/api/tests.py
+++ b/apps/api/tests.py
@@ -1,31 +1,28 @@
 # -*- coding: utf-8 -*-
 # -*- coding: utf-8 -*-

+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from os import path

 
 

-from datetime import datetime
-
+from django.core.files.uploadedfile import SimpleUploadedFile

 from django.test import TestCase
 from django.test import TestCase

-from django.utils import simplejson as json
-from django.conf import settings
+from django.test.utils import override_settings
+import json

 
 

-from api.helpers import timestamp

 from catalogue.models import Book, Tag
 from catalogue.models import Book, Tag

-from picture.tests.utils import RequestFactory

 from picture.forms import PictureImportForm
 from picture.forms import PictureImportForm

-from picture.models import Picture, picture_storage
+from picture.models import Picture

 import picture.tests
 import picture.tests

-from django.core.files.uploadedfile import SimpleUploadedFile
-
- 
-from os import path

 
 
 
 

+@override_settings(
+    API_WAIT=-1,
+    CACHES = {'api': {'BACKEND': 'django.core.cache.backends.dummy.DummyCache'},
+              'default': {'BACKEND': 'django.core.cache.backends.dummy.DummyCache'},
+              'permanent': {'BACKEND': 'django.core.cache.backends.dummy.DummyCache'}}
+)

 class ApiTest(TestCase):
 class ApiTest(TestCase):

-
-    def setUp(self):
-        self.old_api_wait = settings.API_WAIT
-        settings.API_WAIT = -1
-
-    def tearDown(self):
-        settings.API_WAIT = self.old_api_wait
+    pass

 
 
 class ChangesTest(ApiTest):
 
 
 class ChangesTest(ApiTest):


@@ -38,10 +35,10 @@ class ChangesTest(ApiTest):
         book.save()
 
         changes = json.loads(self.client.get('/api/changes/0.json?book_fields=title&tag_fields=name').content)
         book.save()
 
         changes = json.loads(self.client.get('/api/changes/0.json?book_fields=title&tag_fields=name').content)

-        self.assertEqual(changes['updated']['books'], 
+        self.assertEqual(changes['updated']['books'],

                          [{'id': book.id, 'title': book.title}],
                          'Invalid book format in changes')
                          [{'id': book.id, 'title': book.title}],
                          'Invalid book format in changes')

-        self.assertEqual(changes['updated']['tags'], 
+        self.assertEqual(changes['updated']['tags'],

                          [{'id': tag.id, 'name': tag.name}],
                          'Invalid tag format in changes')
 
                          [{'id': tag.id, 'name': tag.name}],
                          'Invalid tag format in changes')
 


@@ -160,5 +157,4 @@ class PictureTests(ApiTest):
         if import_form.is_valid():
             import_form.save()
 
         if import_form.is_valid():
             import_form.save()
 

-        pic = Picture.objects.get(slug=slug)
-
+        Picture.objects.get(slug=slug)