X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/967eed676fc83d15b26149047f353ac61faa8217..59947fddf28d67c6822267c8a0fc28e13a6e9175:/src/api/tests/tests.py diff --git a/src/api/tests/tests.py b/src/api/tests/tests.py index eccdd04a8..63ac8d70f 100644 --- a/src/api/tests/tests.py +++ b/src/api/tests/tests.py @@ -1,27 +1,23 @@ -# -*- 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. +# This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later. +# Copyright © Fundacja Wolne Lektury. See NOTICE for more information. # from base64 import b64encode -from os import path import hashlib import hmac -import json from io import BytesIO +import json +from os import path from time import time +from unittest.mock import patch from urllib.parse import quote, urlencode, parse_qs from django.contrib.auth.models import User from django.core.files.uploadedfile import SimpleUploadedFile from django.test import TestCase from django.test.utils import override_settings -from unittest.mock import patch -from api.models import Consumer, Token from catalogue.models import Book, Tag -from picture.forms import PictureImportForm -from picture.models import Picture -import picture.tests +from api.models import Consumer, Token @override_settings( @@ -30,6 +26,8 @@ import picture.tests 'BACKEND': 'django.core.cache.backends.dummy.DummyCache'}}, ) class ApiTest(TestCase): + maxDiff = None + def load_json(self, url): content = self.client.get(url).content try: @@ -104,32 +102,6 @@ class TagTests(ApiTest): 'Wrong tag details.') -class PictureTests(ApiTest): - def test_publish(self): - slug = "kandinsky-composition-viii" - xml = SimpleUploadedFile( - 'composition8.xml', - open(path.join( - picture.tests.__path__[0], "files", slug + ".xml" - ), 'rb').read()) - img = SimpleUploadedFile( - 'kompozycja-8.png', - open(path.join( - picture.tests.__path__[0], "files", slug + ".png" - ), 'rb').read()) - - import_form = PictureImportForm({}, { - 'picture_xml_file': xml, - 'picture_image_file': img - }) - - assert import_form.is_valid() - if import_form.is_valid(): - import_form.save() - - Picture.objects.get(slug=slug) - - class BooksTests(ApiTest): fixtures = ['test-books.yaml'] @@ -178,12 +150,15 @@ class BooksTests(ApiTest): '/api/filter-books/?lektura=true', []) - self.assert_slugs( - '/api/filter-books/?preview=true', - ['grandchild']) + Book.objects.filter(slug='grandchild').update(preview=True) + # Skipping: we don't allow previewed books in filtered list. + #self.assert_slugs( + # '/api/filter-books/?preview=true', + # ['grandchild']) self.assert_slugs( '/api/filter-books/?preview=false', ['child', 'parent']) + Book.objects.filter(slug='grandchild').update(preview=False) self.assert_slugs( '/api/filter-books/?audiobook=true', @@ -232,11 +207,6 @@ class BlogTests(ApiTest): self.assertEqual(self.load_json('/api/blog'), []) -class PreviewTests(ApiTest): - def unauth(self): - self.assert_json_response('/api/preview/', 'preview.json') - - class OAuth1Tests(ApiTest): @classmethod def setUpClass(cls): @@ -278,7 +248,11 @@ class OAuth1Tests(ApiTest): # Request token authorization. self.client.login(username='test', password='test') - response = self.client.get('/api/oauth/authorize/?oauth_token=%s&oauth_callback=test://oauth.callback/' % request_token) + response = self.client.get( + '/api/oauth/authorize/?oauth_token=%s&oauth_callback=test://oauth.callback/' % ( + request_token, + ) + ) post_data = response.context['form'].initial response = self.client.post('/api/oauth/authorize/?' + urlencode(post_data)) @@ -305,8 +279,8 @@ class OAuth1Tests(ApiTest): ).digest() h = b64encode(h).rstrip(b'\n') sign = quote(h) - query = u"{}&oauth_signature={}".format(base_query, sign) - response = self.client.get(u'/api/oauth/access_token/?' + query) + query = "{}&oauth_signature={}".format(base_query, sign) + response = self.client.get('/api/oauth/access_token/?' + query) access_token_data = parse_qs(response.content.decode('latin1')) access_token = access_token_data['oauth_token'][0] @@ -437,6 +411,8 @@ class AuthorizedTests(ApiTest): ['parent']) def test_subscription(self): + Book.objects.filter(slug='grandchild').update(preview=True) + self.assert_slugs('/api/preview/', ['grandchild']) self.assertEqual( self.signed_json('/api/username/'), @@ -445,28 +421,24 @@ class AuthorizedTests(ApiTest): self.signed('/api/epub/grandchild/').status_code, 403) - with patch('api.fields.user_is_subscribed', return_value=True): + with patch('club.models.Membership.is_active_for', return_value=True): self.assertEqual( self.signed_json('/api/username/'), {"username": "test", "premium": True}) - with patch('paypal.permissions.user_is_subscribed', return_value=True): with patch('django.core.files.storage.Storage.open', return_value=BytesIO(b"")): self.assertEqual( self.signed('/api/epub/grandchild/').content, b"") + Book.objects.filter(slug='grandchild').update(preview=False) + def test_publish(self): response = self.signed('/api/books/', method='POST', data={"data": json.dumps({})}) self.assertEqual(response.status_code, 403) - response = self.signed('/api/pictures/', - method='POST', - data={"data": json.dumps({})}) - self.assertEqual(response.status_code, 403) - self.user.is_superuser = True self.user.save() @@ -479,15 +451,5 @@ class AuthorizedTests(ApiTest): self.assertTrue(mock.called) self.assertEqual(response.status_code, 201) - with patch('picture.models.Picture.from_xml_file') as mock: - response = self.signed('/api/pictures/', - method='POST', - data={"data": json.dumps({ - "picture_xml": "", - "picture_image_data": "Kg==", - })}) - self.assertTrue(mock.called) - self.assertEqual(response.status_code, 201) - self.user.is_superuser = False self.user.save()