-# -*- 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(
'BACKEND': 'django.core.cache.backends.dummy.DummyCache'}},
)
class ApiTest(TestCase):
+ maxDiff = None
+
def load_json(self, url):
content = self.client.get(url).content
try:
'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']
'/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',
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):
# 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))
).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]
['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/'),
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"<epub>")):
self.assertEqual(
self.signed('/api/epub/grandchild/').content,
b"<epub>")
+ 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()
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": "<utwor/>",
- "picture_image_data": "Kg==",
- })})
- self.assertTrue(mock.called)
- self.assertEqual(response.status_code, 201)
-
self.user.is_superuser = False
self.user.save()
fnp / wolnelektury.git / blobdiff