regluit/core/tests.py

1356 lines
58 KiB
Python
Raw Permalink Normal View History

2017-08-03 21:09:42 +00:00
# encoding: utf-8
2018-04-07 22:38:33 +00:00
#external library imports
import os
from datetime import datetime, timedelta
2013-06-03 16:31:39 +00:00
from decimal import Decimal as D
from math import factorial
2018-04-20 01:26:29 +00:00
import unittest
2013-06-03 16:31:39 +00:00
from urlparse import parse_qs, urlparse
from tempfile import NamedTemporaryFile
2018-04-20 01:26:29 +00:00
from time import sleep, mktime
2018-04-07 22:38:33 +00:00
2013-06-03 16:31:39 +00:00
from celery.task.sets import TaskSet
import requests
2016-12-30 22:24:20 +00:00
import requests_mock
2018-04-07 22:38:33 +00:00
#django imports
from django.conf import settings
2013-06-03 16:31:39 +00:00
from django.contrib.auth.models import User
2012-02-10 01:51:10 +00:00
from django.contrib.contenttypes.models import ContentType
from django.contrib.sites.models import Site
from django.core.files import File as DjangoFile
2013-06-03 16:31:39 +00:00
from django.db import IntegrityError
2016-04-09 17:24:44 +00:00
from django.db import transaction
from django.http import Http404
2013-06-03 16:31:39 +00:00
from django.test import TestCase
from django.test.client import Client
from django.test.utils import override_settings
2018-04-19 16:24:34 +00:00
from django.utils.timezone import now
2018-04-07 22:38:33 +00:00
from django_comments.models import Comment
#regluit imports
2013-06-03 16:31:39 +00:00
from regluit.core import (
isbn,
bookloader,
models,
search,
goodreads,
librarything,
tasks,
parameters,
2013-06-03 16:31:39 +00:00
)
from regluit.core.models import (
Campaign,
Work,
UnglueitError,
Edition,
RightsHolder,
Claim,
Key,
Ebook,
Premium,
Subject,
Publisher,
2013-08-22 18:23:47 +00:00
Offer,
2013-08-27 22:03:35 +00:00
EbookFile,
Acq,
Hold,
2013-06-03 16:31:39 +00:00
)
from regluit.libraryauth.models import Library
from regluit.core.parameters import TESTING, LIBRARY, RESERVE
from regluit.core.loaders.utils import (load_from_books, loaded_book_ok, )
from regluit.core.validation import valid_subject
2013-06-03 16:31:39 +00:00
from regluit.frontend.views import safe_get_work
from regluit.payment.models import Transaction
from regluit.payment.parameters import PAYMENT_TYPE_AUTHORIZATION
from regluit.pyepub import EPUB
2018-04-19 16:24:34 +00:00
from regluit.utils.localdatetime import date_today
2014-01-15 13:32:55 +00:00
from .epub import test_epub
2018-04-07 22:38:33 +00:00
from .pdf import test_pdf
2016-12-30 22:24:20 +00:00
TESTDIR = os.path.join(os.path.dirname(__file__), '../test/')
YAML_VERSIONFILE = os.path.join(TESTDIR, 'versiontest.yaml')
YAML_HUCKFILE = os.path.join(TESTDIR, 'raw/master/metadata.yaml')
class BookLoaderTests(TestCase):
2018-04-07 22:38:33 +00:00
fixtures = ['initial_data.json', 'bookloader.json']
2016-12-30 22:24:20 +00:00
def setUp(self):
self.user = User.objects.create_user('core_test', 'test@example.org', 'core_test')
self.client = Client()
self.client.login(username='core_test', password='core_test')
2018-04-07 22:38:33 +00:00
def test_add_by_local_yaml(self):
noebook_id = bookloader.load_from_yaml(YAML_VERSIONFILE)
noebook = models.Work.objects.get(id=noebook_id)
2018-04-07 22:38:33 +00:00
self.assertEqual(noebook.first_ebook(), None)
2016-03-07 21:17:33 +00:00
huck_id = bookloader.load_from_yaml(YAML_HUCKFILE, test_mode=True)
2015-08-18 18:34:52 +00:00
huck = models.Work.objects.get(id=huck_id)
2018-04-07 22:38:33 +00:00
self.assertTrue(huck.ebooks().count() > 1)
def test_add_by_yaml(self):
space_id = bookloader.load_from_yaml(
'https://github.com/gitenberg-dev/metadata/raw/master/samples/pandata.yaml'
)
huck_id = bookloader.load_from_yaml(
'https://github.com/GITenberg/Adventures-of-Huckleberry-Finn_76/raw/master/metadata.yaml'
)
space = models.Work.objects.get(id=space_id)
huck = models.Work.objects.get(id=huck_id)
#test ebook archiving
2018-04-07 22:38:33 +00:00
num_ebf = EbookFile.objects.all().count()
for ebook in huck.ebooks().all():
f = ebook.get_archive()
2018-04-07 22:38:33 +00:00
self.assertTrue(EbookFile.objects.all().count() > num_ebf)
2016-12-30 22:24:20 +00:00
def test_add_by_isbn_mock(self):
with requests_mock.Mocker(real_http=True) as m:
with open(os.path.join(TESTDIR, 'gb_hamilton.json')) as gb:
m.get('https://www.googleapis.com/books/v1/volumes', content=gb.read())
self.test_add_by_isbn(mocking=True)
2018-04-07 22:38:33 +00:00
2016-12-30 22:24:20 +00:00
def test_add_by_isbn(self, mocking=False):
if not (mocking or settings.TEST_INTEGRATION):
return
# edition
2017-09-22 22:31:06 +00:00
edition = bookloader.add_by_isbn('9780143034759')
self.assertEqual(edition.title, u'Alexander Hamilton')
2017-09-22 22:31:06 +00:00
self.assertEqual(edition.publication_date, u'2005')
self.assertEqual(edition.publisher, u'Penguin')
self.assertEqual(edition.isbn_10, '0143034758')
self.assertEqual(edition.isbn_13, '9780143034759')
self.assertEqual(edition.googlebooks_id, '4iafgTEhU3QC')
# authors
self.assertEqual(edition.authors.all().count(), 1)
self.assertEqual(edition.authors.all()[0].name, u'Ron Chernow')
# work
self.assertTrue(edition.work)
2017-09-22 22:31:06 +00:00
self.assertEqual(edition.work.googlebooks_id, '4iafgTEhU3QC')
self.assertEqual(edition.work.first_isbn_13(), '9780143034759')
2018-04-07 22:38:33 +00:00
# test duplicate pubname
ed2 = Edition.objects.create(work=edition.work)
2017-09-22 22:31:06 +00:00
ed2.set_publisher(u'Penguin')
2018-04-07 22:38:33 +00:00
# publisher names
old_pub_name = edition.publisher_name
edition.set_publisher('test publisher name')
self.assertEqual(edition.publisher, u'test publisher name')
pub = Publisher(name=edition.publisher_name)
pub.save()
self.assertEqual(edition.work.publishers().count(), 1)
old_pub_name.publisher = pub
old_pub_name.save()
2017-09-22 22:31:06 +00:00
edition.set_publisher(u'Penguin')
self.assertEqual(edition.publisher, u'test publisher name') # Perseus has been aliased
2016-04-29 15:55:12 +00:00
2016-12-30 22:24:20 +00:00
def test_language_locale_mock(self):
with requests_mock.Mocker(real_http=True) as m:
with open(os.path.join(TESTDIR, 'zhCN.json')) as gb:
2016-12-30 22:24:20 +00:00
m.get('https://www.googleapis.com/books/v1/volumes', content=gb.read())
self.test_language_locale(mocking=True)
2018-04-07 22:38:33 +00:00
2016-12-30 22:24:20 +00:00
def test_language_locale(self, mocking=False):
if not (mocking or settings.TEST_INTEGRATION):
return
edition = bookloader.add_by_isbn('9787104030126')
2018-04-20 01:26:29 +00:00
self.assertEqual(edition.work.language, u'zh-CN')
2016-12-30 22:24:20 +00:00
def test_update_edition_mock(self):
with requests_mock.Mocker(real_http=True) as m:
with open(os.path.join(TESTDIR, 'python4da.json')) as gb:
m.get('https://www.googleapis.com/books/v1/volumes', content=gb.read())
self.test_update_edition(mocking=True)
2018-04-07 22:38:33 +00:00
def test_update_edition(self, mocking=False):
2016-12-30 22:24:20 +00:00
if not (mocking or settings.TEST_INTEGRATION):
return
w = models.Work(title='silly title', language='xx')
w.save()
2018-04-07 22:38:33 +00:00
e = models.Edition(title=w.title, work=w)
e.save()
models.Identifier(type='isbn', value='9781449319793', work=w, edition=e).save()
bookloader.update_edition(e)
self.assertEqual(e.work.language, 'en')
self.assertEqual(e.title, 'Python for Data Analysis')
def test_double_add(self):
2016-12-30 22:24:20 +00:00
edbefore = models.Edition.objects.all().count()
autbefore = models.Author.objects.all().count()
before = models.Work.objects.all().count()
bookloader.add_by_isbn('0441007465') # in fixture
self.assertEqual(models.Edition.objects.all().count(), edbefore)
self.assertEqual(models.Author.objects.all().count(), autbefore)
self.assertEqual(models.Work.objects.all().count(), before)
2018-04-07 22:38:33 +00:00
def test_missing_isbn(self):
2018-04-25 21:20:44 +00:00
e = bookloader.add_by_isbn_from_google('9781938616990') #unassigned in gluejar block
self.assertEqual(e, None)
2016-12-30 22:24:20 +00:00
def test_thingisbn_mock(self):
with requests_mock.Mocker(real_http=True) as m:
with open(os.path.join(TESTDIR, '9780441569595.xml')) as lt:
2017-07-27 14:33:13 +00:00
m.get('https://www.librarything.com/api/thingISBN/0441007465', content=lt.read())
2016-12-30 22:24:20 +00:00
self.test_thingisbn(mocking=True)
def test_thingisbn(self, mocking=False):
if not (mocking or settings.TEST_INTEGRATION):
return
isbns = bookloader.thingisbn('0441007465') #Neuromancer
2011-10-13 01:59:46 +00:00
self.assertTrue(len(isbns) > 20)
self.assertTrue('0441007465' in isbns)
2011-10-13 01:59:46 +00:00
self.assertTrue('3453313895' in isbns)
def test_add_related(self):
# add one edition
2018-04-07 22:38:33 +00:00
#Neuromancer; editions in fixture not joined
edition = bookloader.add_by_isbn('0441007465')
2016-12-30 22:24:20 +00:00
edbefore = models.Edition.objects.count()
before = models.Work.objects.count()
2018-04-07 22:38:33 +00:00
lang = edition.work.language
2016-12-30 22:24:20 +00:00
langbefore = models.Work.objects.filter(language=lang).count()
# ask for related editions to be added using the work we just created
2016-12-30 22:24:20 +00:00
with requests_mock.Mocker(real_http=True) as m:
with open(os.path.join(TESTDIR, '9780441569595.xml')) as lt:
2017-07-27 14:33:13 +00:00
m.get('https://www.librarything.com/api/thingISBN/0441007465', content=lt.read())
2016-12-30 22:24:20 +00:00
bookloader.add_related('0441007465') # should join the editions
self.assertTrue(models.Edition.objects.count() >= edbefore)
self.assertTrue(models.Work.objects.filter(language=lang).count() < langbefore)
self.assertTrue(edition.work.editions.count() > 9)
self.assertTrue(edition.work.reverse_related.count() > 0)
# is edition.work found among the from_work of all the to_work of edition.work?
back_point = True
to_works = [wr.to_work for wr in edition.work.works_related_from.all()]
for to_work in to_works:
if edition.work_id not in [wr1.from_work.id for wr1 in to_work.works_related_to.all()]:
back_point = False
break
self.assertTrue(back_point)
def test_populate_edition(self):
2016-12-30 22:24:20 +00:00
edition = bookloader.add_by_isbn('9780606301121') # A People's History Of The United States
with requests_mock.Mocker(real_http=True) as m:
with open(os.path.join(TESTDIR, '9780061989834.xml')) as lt:
2017-07-27 14:33:13 +00:00
m.get('https://www.librarything.com/api/thingISBN/9780606301121', content=lt.read())
2016-12-30 22:24:20 +00:00
edition = tasks.populate_edition.run(edition.isbn_13)
self.assertTrue(edition.work.editions.all().count() > 10)
self.assertTrue(edition.work.subjects.all().count() > 8)
2012-01-17 04:28:34 +00:00
self.assertTrue(edition.work.publication_date)
edition.publication_date = None
self.assertTrue(edition.work.publication_date)
self.assertTrue(len(edition.work.description) > 20)
self.assertTrue(edition.work.identifiers.filter(type='oclc')[0])
2018-04-07 22:38:33 +00:00
def test_merge_works_mechanics(self):
2018-04-07 22:38:33 +00:00
"""Make sure then merge_works is still okay when we try to merge
works with themselves and with deleted works"""
2016-12-30 22:24:20 +00:00
before = models.Work.objects.count()
wasbefore = models.WasWork.objects.count()
2018-04-07 22:38:33 +00:00
sub1 = Subject(name='test1')
sub1.save()
2018-04-07 22:38:33 +00:00
sub2 = Subject(name='test2')
sub2.save()
w1 = Work(title="Work 1")
w1.save()
w1.subjects.add(sub1)
2018-04-07 22:38:33 +00:00
w2 = Work(title="Work 2")
w2.save()
2018-04-07 22:38:33 +00:00
w2.subjects.add(sub1, sub2)
e1 = Edition(work=w1)
e1.save()
2018-04-07 22:38:33 +00:00
e2 = Edition(work=w2)
e2.save()
2018-04-07 22:38:33 +00:00
eb1 = Ebook(edition=e2)
2014-12-13 17:37:35 +00:00
eb1.save()
2018-04-07 22:38:33 +00:00
e2a = Edition(work=w2)
e2a.save()
2018-04-07 22:38:33 +00:00
self.assertTrue(e1)
self.assertTrue(e2)
self.assertTrue(e2a)
self.assertTrue(e1.work)
self.assertTrue(e2.work)
2016-12-30 22:24:20 +00:00
self.assertEqual(models.Work.objects.count(), before + 2)
2018-04-07 22:38:33 +00:00
2014-12-13 17:37:35 +00:00
self.assertTrue(w2.is_free)
self.assertFalse(w1.is_free)
2018-04-07 22:38:33 +00:00
w1_id = w1.id
w2_id = w2.id
2018-04-07 22:38:33 +00:00
# first try to merge work 1 into itself -- should not do anything
2018-04-07 22:38:33 +00:00
bookloader.merge_works(w1, w1)
2016-12-30 22:24:20 +00:00
self.assertEqual(models.Work.objects.count(), before + 2)
2018-04-07 22:38:33 +00:00
# merge the second work into the first
bookloader.merge_works(e1.work, e2.work)
2016-12-30 22:24:20 +00:00
self.assertEqual(models.Work.objects.count(), before + 1)
self.assertEqual(models.WasWork.objects.count(), wasbefore + 1)
self.assertEqual(w1.subjects.count(), 2)
2018-04-07 22:38:33 +00:00
2014-12-13 17:37:35 +00:00
self.assertTrue(w1.is_free)
# getting proper view?
anon_client = Client()
r = anon_client.get("/work/%s/" % w1_id)
self.assertEqual(r.status_code, 200)
r = anon_client.get("/work/%s/" % w2_id)
2018-04-07 22:38:33 +00:00
self.assertEqual(r.status_code, 200)
# try to do it twice -- nothing should happen
bookloader.merge_works(e1.work, e2a.work)
r = anon_client.get("/work/%s/" % w1_id)
self.assertEqual(r.status_code, 200)
r = anon_client.get("/work/%s/" % w2_id)
2018-04-07 22:38:33 +00:00
self.assertEqual(r.status_code, 200)
# if the work has a selected edition, then don't touch the work.
2018-04-07 22:38:33 +00:00
w3 = Work(title='work 3')
e_pref = Edition(work=w1)
w1.selected_edition = e_pref
bookloader.merge_works(w3, w1)
2018-04-07 22:38:33 +00:00
self.assertTrue(w1.title == 'Work 1')
def test_merge_works(self):
2016-12-30 22:24:20 +00:00
before = models.Work.objects.count()
# add two editions and see that there are two stub works
2018-04-07 22:38:33 +00:00
2016-12-30 22:24:20 +00:00
# for this test, use two unjoined Neuromancer works in fixture
isbn1 = '9780441569595'
isbn2 = '9780441012039'
e1 = bookloader.add_by_isbn(isbn1)
e2 = bookloader.add_by_isbn(isbn2)
2011-11-06 22:42:09 +00:00
self.assertTrue(e1)
self.assertTrue(e2)
self.assertTrue(e1.work)
self.assertTrue(e2.work)
# add the stub works to a wishlist
user = User.objects.create_user('test', 'test@example.org', 'testpass')
user.wishlist.add_work(e1.work, 'test')
user.wishlist.add_work(e2.work, 'test')
manager = User.objects.create_user('manager', 'manager@example.org', 'managerpass')
2018-04-07 22:38:33 +00:00
# create campaigns for the stub works
c1 = models.Campaign.objects.create(
name=e1.work.title,
2018-04-07 22:38:33 +00:00
work=e1.work,
description='Test Campaign 1',
deadline=now(),
target=D('1000.00'),
)
c2 = models.Campaign.objects.create(
name=e2.work.title,
2018-04-07 22:38:33 +00:00
work=e2.work,
description='Test Campaign 2',
deadline=now(),
target=D('1000.00'),
)
c2.managers.add(manager)
c2.save()
self.assertEqual(c2.pk, e2.work.last_campaign().pk)
2012-02-10 01:51:10 +00:00
# comment on the works
site = Site.objects.all()[0]
wct = ContentType.objects.get_for_model(models.Work)
comment1 = Comment(
content_type=wct,
object_pk=e1.work.pk,
comment="test comment1",
2018-04-07 22:38:33 +00:00
user=user,
2012-02-10 01:51:10 +00:00
site=site
)
comment1.save()
comment2 = Comment(
content_type=wct,
object_pk=e2.work.pk,
comment="test comment2",
2018-04-07 22:38:33 +00:00
user=user,
2012-02-10 01:51:10 +00:00
site=site
)
comment2.save()
comment3 = Comment(
content_type=wct,
object_pk=e2.work.pk,
comment="test comment3",
2018-04-07 22:38:33 +00:00
user=manager,
site=site
)
comment3.save()
2018-04-07 22:38:33 +00:00
# now add related edition to make sure Works get merged
bookloader.add_related(isbn1)
# non-zero
2018-04-07 22:38:33 +00:00
self.assertGreater(models.Work.objects.count(), 0)
w3 = models.Edition.get_by_isbn(isbn1).work
2018-04-07 22:38:33 +00:00
# and that relevant Campaigns and Wishlists are updated
2018-04-07 22:38:33 +00:00
c1 = Campaign.objects.get(pk=c1.pk)
c2 = Campaign.objects.get(pk=c2.pk)
2012-05-25 16:52:25 +00:00
self.assertEqual(c1.work, c2.work)
self.assertEqual(user.wishlist.works.all().count(), 1)
self.assertEqual(Comment.objects.for_model(w3).count(), 3)
2018-04-07 22:38:33 +00:00
anon_client = Client()
r = anon_client.get("/work/%s/" % w3.pk)
self.assertEqual(r.status_code, 200)
r = anon_client.get("/work/%s/" % e2.work.pk)
self.assertEqual(r.status_code, 200)
2018-04-07 22:38:33 +00:00
2011-11-06 21:33:04 +00:00
def test_ebook(self):
2016-12-30 22:24:20 +00:00
with requests_mock.Mocker(real_http=True) as m:
with open(os.path.join(TESTDIR, 'gb_latinlanguage.json')) as gb:
m.get('https://www.googleapis.com/books/v1/volumes', content=gb.read())
edition = bookloader.add_by_oclc('1246014')
2018-04-07 22:38:33 +00:00
# we've seen the public domain status of this book fluctuate -- and the OCLC
# number can disappear. So if the ebook count is 2 then test
2016-12-30 22:24:20 +00:00
#if edition is not None and edition.ebooks.count() == 2:
self.assertEqual(edition.ebooks.count(), 2)
#ebook_epub = edition.ebooks.all()[0]
ebook_epub = edition.ebooks.filter(format='epub')[0]
self.assertEqual(ebook_epub.format, 'epub')
self.assertEqual(parse_qs(urlparse(ebook_epub.url).query).get("id"), ['N1RfAAAAMAAJ'])
self.assertEqual(parse_qs(urlparse(ebook_epub.url).query).get("output"), ['epub'])
self.assertEqual(ebook_epub.provider, 'Google Books')
self.assertEqual(ebook_epub.set_provider(), 'Google Books')
ebook_pdf = edition.ebooks.filter(format='pdf')[0]
self.assertEqual(ebook_pdf.format, 'pdf')
self.assertEqual(parse_qs(urlparse(ebook_pdf.url).query).get("id"), ['N1RfAAAAMAAJ'])
self.assertEqual(parse_qs(urlparse(ebook_pdf.url).query).get("output"), ['pdf'])
2018-04-07 22:38:33 +00:00
self.assertEqual(ebook_pdf.provider, 'Google Books')
2016-12-30 22:24:20 +00:00
w = edition.work
self.assertEqual(w.first_epub().url, ebook_epub.url)
self.assertEqual(w.first_pdf().url, ebook_pdf.url)
self.assertEqual(w.first_epub_url(), ebook_epub.url)
self.assertEqual(w.first_pdf_url(), ebook_pdf.url)
2018-04-07 22:38:33 +00:00
ebook_pdf.url = 'https://en.wikisource.org/wiki/Frankenstein'
2016-12-30 22:24:20 +00:00
self.assertEqual(ebook_pdf.set_provider(), 'Wikisource')
2018-04-07 22:38:33 +00:00
self.user.wishlist.add_work(w, 'test')
2016-12-30 22:24:20 +00:00
tasks.report_new_ebooks(date_today())
2018-04-07 22:38:33 +00:00
r = self.client.get("/notification/")
2016-12-30 22:24:20 +00:00
self.assertEqual(r.status_code, 200)
2018-04-07 22:38:33 +00:00
2016-12-30 22:24:20 +00:00
ebook_pdf.increment()
updated_ebook = Ebook.objects.get(pk=ebook_pdf.pk)
self.assertEqual(int(updated_ebook.download_count), 1)
self.assertEqual(int(edition.work.download_count), 1)
2011-11-06 22:42:09 +00:00
def test_add_no_ebook(self):
# this edition lacks an ebook, but we should still be able to load it
# http://books.google.com/books?id=D-WjL_HRbNQC&printsec=frontcover#v=onepage&q&f=false
# Social Life of Information
2016-12-30 22:24:20 +00:00
with requests_mock.Mocker(real_http=True) as m:
with open(os.path.join(TESTDIR, 'gb_sociallife.json')) as gb:
m.get('https://www.googleapis.com/books/v1/volumes', content=gb.read())
e = bookloader.add_by_isbn('1578517087')
self.assertTrue(e)
def test_add_openlibrary(self):
work = bookloader.add_by_isbn('0441007465').work
bookloader.add_related('0441007465')
bookloader.add_openlibrary(work)
2011-12-19 07:34:29 +00:00
subjects = [s.name for s in work.subjects.all()]
self.assertTrue(len(subjects) > 10)
self.assertTrue('Science fiction' in subjects)
2018-04-07 22:38:33 +00:00
self.assertTrue('/works/OL27258W' in work.identifiers.filter(type='olwk').values_list('value', flat=True))
self.assertTrue('888628' in work.identifiers.filter(type='gdrd').values_list('value', flat=True))
self.assertTrue('609' in work.identifiers.filter(type='ltwk').values_list('value', flat=True))
def test_unicode_openlibrary(self):
2016-12-30 22:24:20 +00:00
with requests_mock.Mocker(real_http=True) as m:
with open(os.path.join(TESTDIR, 'gb_fightclub.json')) as gb:
m.get('https://www.googleapis.com/books/v1/volumes', content=gb.read())
work = bookloader.add_by_isbn('9783894808358').work #fight club
bookloader.add_openlibrary(work)
self.assertTrue(work.description.startswith('Sie sind jung,'))
2018-04-07 22:38:33 +00:00
2015-08-21 22:31:52 +00:00
def notest_load_gutenberg_edition(self):
"""Let's try this out for Moby Dick"""
2018-04-07 22:38:33 +00:00
title = "Moby Dick"
ol_work_id = "/works/OL102749W"
gutenberg_etext_id = 2701
2017-07-27 14:33:13 +00:00
epub_url = "https://www.gutenberg.org/cache/epub/2701/pg2701.epub"
license = 'https://www.gutenberg.org/license'
lang = 'en'
format = 'epub'
2018-04-07 22:38:33 +00:00
publication_date = datetime(2001, 7, 1)
# https://www.amazon.com/Moby-Dick-Whale-Penguin-Classics-Deluxe/dp/0142000086
seed_isbn = '9780142000083'
ebook = bookloader.load_gutenberg_edition(
title,
gutenberg_etext_id,
ol_work_id,
seed_isbn,
epub_url,
format,
license,
lang,
publication_date
)
self.assertEqual(ebook.url, epub_url)
2018-04-07 22:38:33 +00:00
def tearDown(self):
for ebf in EbookFile.objects.all():
ebf.file.delete()
2018-04-07 22:38:33 +00:00
class SearchTests(TestCase):
2016-12-31 03:26:16 +00:00
def test_search_mock(self):
with requests_mock.Mocker(real_http=True) as m:
2018-04-07 22:38:33 +00:00
with open(
os.path.join(TESTDIR, 'gb_melville.json')
) as gb, open(os.path.join(TESTDIR, 'gb_melville2.json')) as gb2:
m.get(
'https://www.googleapis.com/books/v1/volumes',
[{'content':gb2.read()}, {'content':gb.read()}]
)
2016-12-31 03:26:16 +00:00
self.test_pagination(mocking=True)
self.test_basic_search(mocking=True)
self.test_googlebooks_search(mocking=True)
2016-12-31 03:26:16 +00:00
def test_basic_search(self, mocking=False):
if not (mocking or settings.TEST_INTEGRATION):
return
2011-11-20 14:28:53 +00:00
results = search.gluejar_search('melville')
self.assertEqual(len(results), 10)
r = results[0]
self.assertTrue(r.has_key('title'))
self.assertTrue(r.has_key('author'))
self.assertTrue(r.has_key('description'))
self.assertTrue(r.has_key('cover_image_thumbnail'))
2018-04-07 22:38:33 +00:00
self.assertTrue(
r['cover_image_thumbnail'].startswith('https')
or r['cover_image_thumbnail'].startswith('http')
)
self.assertTrue(r.has_key('publisher'))
self.assertTrue(r.has_key('isbn_13'))
self.assertTrue(r.has_key('googlebooks_id'))
2016-12-31 03:26:16 +00:00
def test_pagination(self, mocking=False):
if not (mocking or settings.TEST_INTEGRATION):
return
2012-02-05 00:22:04 +00:00
r1 = search.gluejar_search('melville', page=1)
r2 = search.gluejar_search('melville', page=2)
isbns1 = set([r['isbn_13'] for r in r1])
isbns2 = set([r['isbn_13'] for r in r2])
self.assertTrue(isbns1 != isbns2)
2016-12-31 03:26:16 +00:00
def test_googlebooks_search(self, mocking=False):
if not (mocking or settings.TEST_INTEGRATION):
return
2012-02-05 00:22:04 +00:00
response = search.googlebooks_search('melville', '69.243.24.29', 1)
self.assertEqual(len(response['items']), 10)
2011-10-07 21:17:54 +00:00
2011-10-07 21:17:54 +00:00
class CampaignTests(TestCase):
2016-04-09 17:21:12 +00:00
fixtures = ['initial_data.json']
def test_b2u(self):
w = Work()
w.save()
this_year = datetime.now().year
c = Campaign(
2018-04-07 22:38:33 +00:00
target=D('12000.00'),
deadline=datetime(this_year, 1, 1),
work=w, type=2,
cc_date_initial=datetime(this_year + 100, 1, 1),
)
2018-04-07 22:38:33 +00:00
self.assertTrue(c.set_dollar_per_day() < 0.34)
self.assertTrue(c.dollar_per_day > 0.31)
t = Transaction(type=1, campaign=c, approved=True, amount=D(6000.1), status="Complete")
2014-03-02 03:42:26 +00:00
t.save()
2013-08-09 02:32:58 +00:00
c.status = 'ACTIVE'
c.save()
c.update_left()
#print(w.percent_of_goal())
2018-04-07 22:38:33 +00:00
self.assertEqual(w.percent_unglued(), 3)
self.assertTrue(w.percent_of_goal() > 49)
ofr = Offer.objects.create(work=w, price=D(10), active=True)
self.assertTrue(c.days_per_copy < D(32.26))
self.assertTrue(c.days_per_copy > D(29.41))
def test_required_fields(self):
# a campaign must have a target, deadline and a work
2017-07-27 14:33:13 +00:00
# see https://stackoverflow.com/questions/21458387/transactionmanagementerror-you-cant-execute-queries-until-the-end-of-the-atom
2016-04-09 17:24:44 +00:00
with transaction.atomic():
c = Campaign()
self.assertRaises(IntegrityError, c.save)
with transaction.atomic():
c = Campaign(target=D('1000.00'))
self.assertRaises(IntegrityError, c.save)
with transaction.atomic():
c = Campaign(target=D('1000.00'), deadline=datetime(2013, 1, 1))
self.assertRaises(IntegrityError, c.save)
w = Work()
w.save()
c = Campaign(target=D('1000.00'), deadline=datetime(2013, 1, 1), work=w)
c.license = 'CC BY-NC'
c.save()
2017-07-27 14:33:13 +00:00
self.assertEqual(c.license_url, 'https://creativecommons.org/licenses/by-nc/3.0/')
2014-11-10 18:14:59 +00:00
self.assertEqual(c.license_badge, '/static/images/ccbync.png')
2018-04-07 22:38:33 +00:00
def test_campaign_status(self):
2018-04-07 22:38:33 +00:00
# need a user to associate with a transaction
user = User.objects.create_user('test', 'test@example.org', 'testpass')
2018-04-07 22:38:33 +00:00
w = Work()
w.save()
w2 = Work()
w2.save()
# INITIALIZED
2018-04-07 22:38:33 +00:00
c1 = Campaign(target=D('1000.00'), deadline=Campaign.latest_ending(), work=w)
c1.save()
self.assertEqual(c1.status, 'INITIALIZED')
# ACTIVATED
2018-04-07 22:38:33 +00:00
c2 = Campaign(
target=D('1000.00'),
deadline=datetime(2013, 1, 1),
work=w,
description='dummy description'
)
c2.save()
self.assertEqual(c2.status, 'INITIALIZED')
u = User.objects.create_user('claimer', 'claimer@example.org', 'claimer')
u.save()
2018-04-07 22:38:33 +00:00
rh = RightsHolder(owner=u, rights_holder_name='rights holder name')
rh.save()
2018-04-07 22:38:33 +00:00
cl = Claim(rights_holder=rh, work=w, user=u, status='active')
cl.save()
2018-04-07 22:38:33 +00:00
cl2 = Claim(rights_holder=rh, work=w2, user=u, status='active')
cl2.save()
c2.activate()
self.assertEqual(c2.status, 'ACTIVE')
# SUSPENDED
c2.suspend(reason="for testing")
self.assertEqual(c2.status, 'SUSPENDED')
# RESUMING
c2.resume(reason="for testing")
#self.assertEqual(c2.suspended, None)
2018-04-07 22:38:33 +00:00
self.assertEqual(c2.status, 'ACTIVE')
# should not let me suspend a campaign that hasn't been initialized
self.assertRaises(UnglueitError, c1.suspend, "for testing")
# UNSUCCESSFUL
2018-04-07 22:38:33 +00:00
c3 = Campaign(
target=D('1000.00'),
deadline=now() - timedelta(days=1),
work=w2,
description='dummy description'
)
c3.save()
c3.activate()
self.assertEqual(c3.status, 'ACTIVE')
2018-04-07 22:38:33 +00:00
# at this point, since the deadline has passed,
# the status should change and be UNSUCCESSFUL
self.assertTrue(c3.update_status())
self.assertEqual(c3.status, 'UNSUCCESSFUL')
2018-04-07 22:38:33 +00:00
# premiums
2018-04-07 22:38:33 +00:00
pr1 = Premium(type='CU', campaign=c3, amount=10, description='botsnack', limit=1)
pr1.save()
2018-04-07 22:38:33 +00:00
self.assertEqual(pr1.premium_remaining, 1)
#cloning (note we changed c3 to w2 to make it clonable)
2018-04-07 22:38:33 +00:00
c7 = c3.clone()
self.assertEqual(c7.status, 'INITIALIZED')
2018-04-07 22:38:33 +00:00
self.assertEqual(c7.premiums.all()[0].description, 'botsnack')
# SUCCESSFUL
2018-04-07 22:38:33 +00:00
c4 = Campaign(
target=D('1000.00'),
deadline=now() - timedelta(days=1),
work=w,
description='dummy description'
)
c4.save()
c4.activate()
t = Transaction()
t.amount = D('1234.00')
t.type = PAYMENT_TYPE_AUTHORIZATION
t.status = 'ACTIVE'
t.approved = True
t.campaign = c4
t.user = user
2018-04-07 22:38:33 +00:00
t.save()
self.assertTrue(c4.update_status())
self.assertEqual(c4.status, 'SUCCESSFUL')
2018-04-07 22:38:33 +00:00
# WITHDRAWN
2018-04-07 22:38:33 +00:00
c5 = Campaign(
target=D('1000.00'),
deadline=datetime(2013, 1, 1),
work=w,
description='dummy description'
)
c5.save()
c5.activate().withdraw('testing')
2018-04-07 22:38:33 +00:00
self.assertEqual(c5.status, 'WITHDRAWN')
# testing percent-of-goal
w2 = Work()
w2.save()
2018-04-07 22:38:33 +00:00
c6 = Campaign(
target=D('1000.00'),
deadline=now() + timedelta(days=1),
work=w2,
description='dummy description'
)
c6.save()
2018-04-07 22:38:33 +00:00
cl = Claim(rights_holder=rh, work=w2, user=u, status='active')
cl.save()
c6.activate()
t = Transaction()
t.amount = D('234.00')
t.type = PAYMENT_TYPE_AUTHORIZATION
t.status = 'ACTIVE'
t.approved = True
t.campaign = c6
t.user = user
t.save()
self.assertEqual(w2.percent_of_goal(), 23)
2018-04-07 22:38:33 +00:00
self.assertEqual(c1.launchable, False)
2018-04-07 22:38:33 +00:00
c1.description = "description"
self.assertEqual(c1.launchable, True)
c1.work.create_offers()
self.assertEqual(c1.work.offers.count(), 2)
self.assertEqual(c1.work.offers.filter(license=2).count(), 1)
c1.type = 2
c1.save()
self.assertEqual(c1.launchable, False)
2018-04-07 22:38:33 +00:00
of1 = c1.work.offers.get(license=2)
of1.price = D(2)
of1.active = True
of1.save()
self.assertEqual(c1.launchable, False)
2018-04-07 22:38:33 +00:00
e1 = models.Edition(title="title", work=c1.work)
e1.save()
2018-04-07 22:38:33 +00:00
ebf1 = models.EbookFile(edition=e1, format=1)
ebf1.save()
c1.set_cc_date_initial()
self.assertEqual(c1.cc_date, settings.MAX_CC_DATE)
c1.target = D(settings.UNGLUEIT_MAXIMUM_TARGET)
c1.save()
self.assertEqual(c1.launchable, True)
class WishlistTest(TestCase):
2016-12-31 03:26:16 +00:00
fixtures = ['initial_data.json', 'neuromancer.json']
def test_add_remove(self):
# add a work to a user's wishlist
user = User.objects.create_user('test', 'test@example.org', 'testpass')
edition = bookloader.add_by_isbn('0441007465')
work = edition.work
2018-04-07 22:38:33 +00:00
num_wishes = work.num_wishes
user.wishlist.add_work(work, 'test')
self.assertEqual(user.wishlist.works.count(), 1)
self.assertEqual(work.num_wishes, num_wishes+1)
2016-01-20 01:53:55 +00:00
self.assertEqual(work.priority(),1)
user.wishlist.remove_work(work)
self.assertEqual(user.wishlist.works.count(), 0)
self.assertEqual(work.num_wishes, num_wishes)
2018-04-07 22:38:33 +00:00
class CeleryTaskTest(TestCase):
def test_single_fac(self):
n = 10
task = tasks.fac.delay(n)
result = task.get(timeout=10)
2018-04-07 22:38:33 +00:00
self.assertEqual(result, factorial(n))
def test_subtask(self):
n = 30
subtasks = [tasks.fac.subtask(args=(x,)) for x in range(n)]
job = TaskSet(tasks=subtasks)
result = job.apply_async()
while not result.ready():
sleep(0.2)
2018-04-07 22:38:33 +00:00
self.assertEqual(result.join(), [factorial(x) for x in range(n)])
class GoodreadsTest(TestCase):
2014-05-02 21:20:37 +00:00
@unittest.skip("Goodreads down at the moment")
def test_goodreads_shelves(self):
2016-12-31 03:26:16 +00:00
if not settings.GOODREADS_API_SECRET:
return
# test to see whether the core undeletable shelves are on the list
gr_uid = "767708" # for Raymond Yee
2018-04-07 22:38:33 +00:00
gc = goodreads.GoodreadsClient(
key=settings.GOODREADS_API_KEY,
secret=settings.GOODREADS_API_SECRET
)
shelves = gc.shelves_list(gr_uid)
shelf_names = [s['name'] for s in shelves['user_shelves']]
self.assertTrue('currently-reading' in shelf_names)
self.assertTrue('read' in shelf_names)
self.assertTrue('to-read' in shelf_names)
2014-05-02 21:20:37 +00:00
@unittest.skip("Goodreads down at the moment")
def test_review_list_unauth(self):
2016-12-31 03:26:16 +00:00
if not settings.GOODREADS_API_SECRET:
return
gr_uid = "767708" # for Raymond Yee
2018-04-07 22:38:33 +00:00
gc = goodreads.GoodreadsClient(
key=settings.GOODREADS_API_KEY,
secret=settings.GOODREADS_API_SECRET
)
reviews = gc.review_list_unauth(user_id=gr_uid, shelf='read')
# test to see whether there is a book field in each of the review
2017-07-27 14:33:13 +00:00
# url for test is https://www.goodreads.com/review/list.xml?id=767708&shelf=read&page=1&per_page=20&order=a&v=2&key=[key]
self.assertTrue(all([r.has_key("book") for r in reviews]))
class LibraryThingTest(TestCase):
def test_scrape_test_lib(self):
# account yujx : has one book: 0471925675
lt_username = 'yujx'
lt = librarything.LibraryThing(username=lt_username)
books = list(lt.parse_user_catalog(view_style=5))
2018-04-07 22:38:33 +00:00
self.assertEqual(len(books), 1)
self.assertEqual(books[0]['isbn'], '0471925675')
self.assertEqual(books[0]['work_id'], '80826')
self.assertEqual(books[0]['book_id'], '79883733')
class ISBNTest(TestCase):
def test_ISBN(self):
milosz_10 = '006019667X'
milosz_13 = '9780060196677'
python_10 = '0-672-32978-6'
2018-04-19 02:00:36 +00:00
funky = u'0672—329 78-6' # endash, mdash, space
python_10_wrong = '0-672-32978-7'
python_13 = '978-0-672-32978-4'
2018-04-07 22:38:33 +00:00
isbn_python_10 = isbn.ISBN(python_10)
isbn_python_13 = isbn.ISBN(python_13)
# return None for invalid characters
self.assertEqual(None, isbn.ISBN("978-0-M72-32978-X").to_string('13'))
self.assertEqual(isbn.ISBN("978-0-M72-32978-X").valid, False)
2017-08-03 21:09:42 +00:00
self.assertEqual(None, bookloader.valid_isbn("978-0-M72-32978-X"))
# check that only ISBN 13 starting with 978 or 979 are accepted
self.assertEqual(None, isbn.ISBN("111-0-M72-32978-X").to_string())
2017-08-03 21:09:42 +00:00
self.assertEqual(None, bookloader.valid_isbn("111-0-M72-32978-X"))
2017-08-03 21:31:12 +00:00
self.assertEqual(isbn_python_13.to_string(), bookloader.valid_isbn(funky))
2018-04-07 22:38:33 +00:00
# right type?
self.assertEqual(isbn_python_10.type, '10')
self.assertEqual(isbn_python_13.type, '13')
# valid?
self.assertEqual(isbn_python_10.valid, True)
self.assertEqual(isbn.ISBN(python_10_wrong).valid, False)
2017-08-03 21:31:12 +00:00
self.assertEqual(13, len(bookloader.valid_isbn(python_10)))
self.assertEqual(isbn_python_13.to_string(), bookloader.valid_isbn(python_10_wrong))
2018-04-07 22:38:33 +00:00
# do conversion -- first the outside methods
2018-04-07 22:38:33 +00:00
self.assertEqual(isbn.convert_10_to_13(isbn.strip(python_10)), isbn.strip(python_13))
self.assertEqual(isbn.convert_10_to_13(isbn.strip(python_10)), isbn.strip(python_13))
self.assertEqual(isbn.convert_13_to_10('xxxxxxxxxxxxx'), None)
self.assertEqual(isbn.convert_10_to_13('xxxxxxxxxx'), None)
2017-08-03 21:09:42 +00:00
self.assertEqual(None, bookloader.valid_isbn('xxxxxxxxxxxxx'))
self.assertEqual(None, bookloader.valid_isbn('xxxxxxxxxx'))
2018-04-07 22:38:33 +00:00
# check formatting
self.assertEqual(isbn.ISBN(python_13).to_string(type='13'), '9780672329784')
2018-04-07 22:38:33 +00:00
self.assertEqual(isbn.ISBN(python_13).to_string('13', True), '978-0-672-32978-4')
self.assertEqual(isbn.ISBN(python_13).to_string(type='10'), '0672329786')
self.assertEqual(isbn.ISBN(python_10).to_string(type='13'), '9780672329784')
2018-04-07 22:38:33 +00:00
self.assertEqual(isbn.ISBN(python_10).to_string(10, True), '0-672-32978-6')
# complain if one tries to get ISBN-10 for a 979 ISBN 13
# making up a 979 ISBN
isbn_979 = isbn.ISBN("979-1-234-56789-0").validate()
self.assertEqual(isbn_979.to_string('10'), None)
2018-04-07 22:38:33 +00:00
# check casting to string -- ISBN 13
self.assertEqual(str(isbn.ISBN(python_10)), '0672329786')
2018-04-07 22:38:33 +00:00
# test __eq__ and __ne__ and validate
self.assertTrue(isbn.ISBN(milosz_10) == isbn.ISBN(milosz_13))
self.assertTrue(isbn.ISBN(milosz_10) == milosz_13)
self.assertFalse(isbn.ISBN(milosz_10) == 'ddds')
self.assertFalse(isbn.ISBN(milosz_10) != milosz_10)
self.assertTrue(isbn.ISBN(python_10) != python_10_wrong)
self.assertEqual(isbn.ISBN(python_10_wrong).validate(), python_10)
self.assertEqual(isbn.ISBN(python_13).validate(), python_10)
2018-04-07 22:38:33 +00:00
# curious about set membership
2018-04-07 22:38:33 +00:00
self.assertEqual(len(set([isbn.ISBN(milosz_10), isbn.ISBN(milosz_13)])), 2)
self.assertEqual(len(set([str(isbn.ISBN(milosz_10)), str(isbn.ISBN(milosz_13))])), 2)
self.assertEqual(
len(set([isbn.ISBN(milosz_10).to_string(), isbn.ISBN(milosz_13).to_string()])),
1
)
class EncryptedKeyTest(TestCase):
def test_create_read_key(self):
name = "the great answer"
value = "42"
key = Key.objects.create(name=name, value=value)
key.save()
# do we get back the value?
self.assertEqual(Key.objects.filter(name=name)[0].value, value)
# just checking that the encrypted value is not the same as the value
self.assertNotEqual(key.encrypted_value, value) # is this always true?
2018-04-07 22:38:33 +00:00
class SafeGetWorkTest(TestCase):
def test_good_work(self):
w1 = models.Work()
w1.save()
w2 = models.Work()
w2.save()
w2_id = w2.id
bookloader.merge_works(w1, w2)
2018-04-07 22:38:33 +00:00
work = safe_get_work(w1.id)
self.assertEqual(work, w1)
work = safe_get_work(w2_id)
self.assertEqual(work, w1)
self.assertRaises(Http404, safe_get_work, 3)
2015-08-22 18:29:35 +00:00
2018-04-07 22:38:33 +00:00
class WorkTests(TestCase):
def setUp(self):
self.w1 = models.Work.objects.create()
self.w2 = models.Work.objects.create()
2015-08-22 18:29:35 +00:00
def test_preferred_edition(self):
2018-04-07 22:38:33 +00:00
ww = models.WasWork.objects.create(work=self.w1, was=self.w2.id)
e1 = models.Edition.objects.create(work=self.w1)
self.assertEqual(e1, self.w1.preferred_edition)
e2 = models.Edition.objects.create(work=self.w1)
2018-04-07 22:38:33 +00:00
self.w1.selected_edition = e2
self.w1.save()
self.assertEqual(e2, self.w1.preferred_edition)
self.assertEqual(e2, self.w2.preferred_edition)
2018-04-07 22:38:33 +00:00
def test_valid_subject(self):
self.assertTrue(valid_subject(u'A, valid, suj\xc3t'))
self.assertFalse(valid_subject(u'A, valid, suj\xc3t, '))
self.assertFalse(valid_subject(u'A valid suj\xc3t \x01'))
Subject.set_by_name(u'A, valid, suj\xc3t; A, valid, suj\xc3t, ', work=self.w1)
self.assertEqual(1, self.w1.subjects.count())
sub = Subject.set_by_name('nyt:hardcover_advice=2011-06-18', work=self.w1)
self.assertEqual(sub.name, 'NYT Bestseller - Hardcover advice')
self.assertEqual(2, self.w1.subjects.count())
sub2 = Subject.set_by_name('!lcsh: Something', work=self.w1)
self.assertEqual(sub2.name, 'Something')
self.assertEqual(3, self.w1.subjects.count())
sub3 = Subject.set_by_name('Something', work=self.w1)
self.assertEqual(sub3.name, 'Something')
self.assertEqual(3, self.w1.subjects.count())
self.assertEqual(sub3.authority, 'lcsh')
2018-04-07 22:38:33 +00:00
class DownloadPageTest(TestCase):
2016-04-09 17:21:12 +00:00
fixtures = ['initial_data.json']
def test_download_page(self):
w = models.Work()
w.save()
e1 = models.Edition()
e1.work = w
e1.unglued = True
e2 = models.Edition()
e2.work = w
e1.save()
e2.save()
2018-04-07 22:38:33 +00:00
eb1 = models.Ebook()
2017-07-27 14:33:13 +00:00
eb1.url = "https://example.org"
eb1.edition = e1
eb1.format = 'epub'
2018-04-07 22:38:33 +00:00
eb2 = models.Ebook()
2017-07-27 14:33:13 +00:00
eb2.url = "https://example2.com"
eb2.edition = e2
2012-09-25 16:27:39 +00:00
eb2.format = 'mobi'
2018-04-07 22:38:33 +00:00
eb1.save()
eb2.save()
2018-04-07 22:38:33 +00:00
anon_client = Client()
response = anon_client.get("/work/%s/download/" % w.id, follow=True)
2016-05-12 16:56:25 +00:00
self.assertContains(response, "/download_ebook/%s/"% eb1.id, count=11)
2018-06-05 18:48:22 +00:00
self.assertContains(response, "/download_ebook/%s/"% eb2.id, count=4)
2014-12-13 17:37:35 +00:00
self.assertTrue(eb1.edition.work.is_free)
eb1.delete()
self.assertTrue(eb2.edition.work.is_free)
eb2.delete()
self.assertFalse(eb2.edition.work.is_free)
2013-03-04 22:19:39 +00:00
class MailingListTests(TestCase):
#mostly to check that MailChimp account is setp correctly
2013-03-04 22:19:39 +00:00
def test_mailchimp(self):
from postmonkey import PostMonkey
pm = PostMonkey(settings.MAILCHIMP_API_KEY)
2016-12-31 03:26:16 +00:00
if settings.TEST_INTEGRATION:
2018-04-07 22:38:33 +00:00
self.assertEqual(pm.ping(), "Everything's Chimpy!")
2016-12-31 03:26:16 +00:00
self.user = User.objects.create_user('chimp_test', 'eric@gluejar.com', 'chimp_test')
self.assertTrue(self.user.profile.on_ml)
2013-03-04 22:19:39 +00:00
@override_settings(LOCAL_TEST=True)
2013-08-27 22:03:35 +00:00
class EbookFileTests(TestCase):
2016-04-09 17:21:12 +00:00
fixtures = ['initial_data.json']
2014-01-15 13:32:55 +00:00
def test_badepub_errors(self):
textfile = NamedTemporaryFile(delete=False)
textfile.write("bad text file")
textfile.seek(0)
self.assertTrue(test_epub(textfile))
2013-08-27 22:03:35 +00:00
def test_ebookfile(self):
"""
Read the test epub file
"""
2013-08-27 22:03:35 +00:00
w = Work.objects.create(title="Work 1")
2018-04-07 22:38:33 +00:00
e = Edition.objects.create(title=w.title, work=w)
2013-08-27 22:03:35 +00:00
u = User.objects.create_user('test', 'test@example.org', 'testpass')
2018-04-07 22:38:33 +00:00
rh = RightsHolder.objects.create(owner=u, rights_holder_name='rights holder name')
cl = Claim.objects.create(rights_holder=rh, work=w, user=u, status='active')
c = Campaign.objects.create(
work=w,
type=parameters.BUY2UNGLUE,
cc_date_initial=datetime(2020, 1, 1),
target=1000,
deadline=datetime(2020, 1, 1),
license='CC BY',
description="dummy description",
)
# download the test epub into a temp file
temp = NamedTemporaryFile(delete=False)
test_file_content = requests.get(settings.BOOXTREAM_TEST_EPUB_URL).content
2018-04-07 22:38:33 +00:00
temp.write(test_file_content)
temp.close()
2018-04-07 22:38:33 +00:00
try:
# now we can try putting the test epub file into Django storage
temp_file = open(temp.name)
2018-04-07 22:38:33 +00:00
dj_file = DjangoFile(temp_file)
2018-04-07 22:38:33 +00:00
ebf = EbookFile(format='epub', edition=e, file=dj_file)
ebf.save()
2018-04-07 22:38:33 +00:00
temp_file.close()
finally:
# make sure we get rid of temp file
os.remove(temp.name)
2018-04-07 22:38:33 +00:00
test_epub = EPUB(ebf.file, mode='a')
self.assertEqual(len(test_epub.opf), 4)
self.assertTrue(len(test_epub.opf[2]) < 30)
acq = Acq.objects.create(user=u,work=w,license=TESTING)
self.assertIsNot(acq.nonce, None)
2018-04-07 22:38:33 +00:00
url = acq.get_watermarked().download_link_epub
self.assertRegexpMatches(url, 'github.com/eshellman/42_ebook/blob/master/download/42')
#self.assertRegexpMatches(url, 'booxtream.com/')
2014-02-21 18:16:55 +00:00
with self.assertRaises(UnglueitError) as cm:
c.activate()
off = Offer(price=10.00, work=w, active=True)
off.save()
c.activate()
#flip the campaign to success
2018-04-07 22:38:33 +00:00
c.cc_date_initial = datetime(2012, 1, 1)
c.update_status()
2018-04-07 22:38:33 +00:00
self.assertEqual(c.work.ebooks().count(), 2)
c.do_watermark = False
c.save()
2018-04-07 22:38:33 +00:00
url = acq.get_watermarked().download_link_epub
2016-03-28 18:30:05 +00:00
def test_ebookfile_thanks(self):
2014-08-28 19:29:41 +00:00
w = Work.objects.create(title="Work 2")
2018-04-07 22:38:33 +00:00
e = Edition.objects.create(title=w.title, work=w)
2014-08-28 19:29:41 +00:00
u = User.objects.create_user('test2', 'test@example.org', 'testpass')
2018-04-07 22:38:33 +00:00
rh = RightsHolder.objects.create(owner=u, rights_holder_name='rights holder name 2')
cl = Claim.objects.create(rights_holder=rh, work=w, user=u, status='active')
c = Campaign.objects.create(work=w,
type=parameters.THANKS,
license='CC BY-NC',
description="Please send me money",
)
2014-08-28 19:29:41 +00:00
# download the test epub into a temp file
temp = NamedTemporaryFile(delete=False)
test_file_content = requests.get(settings.TEST_PDF_URL).content
2018-04-07 22:38:33 +00:00
2014-08-28 19:29:41 +00:00
temp.write(test_file_content)
temp.close()
try:
# now we can try putting the test pdf file into Django storage
temp_file = open(temp.name)
2018-04-07 22:38:33 +00:00
2014-08-28 19:29:41 +00:00
dj_file = DjangoFile(temp_file)
2018-04-07 22:38:33 +00:00
ebf = EbookFile(format='pdf', edition=e, file=dj_file)
2014-08-28 19:29:41 +00:00
ebf.save()
2018-04-07 22:38:33 +00:00
eb = Ebook(format='pdf', edition=e, url=ebf.file.url, provider='Unglue.it')
eb.save()
ebf.ebook = eb
ebf.save()
2018-04-07 22:38:33 +00:00
2014-08-28 19:29:41 +00:00
temp_file.close()
finally:
# make sure we get rid of temp file
os.remove(temp.name)
#test the ask-appender
c.add_ask_to_ebfs()
2016-12-31 03:26:16 +00:00
if settings.AWS_SECRET_ACCESS_KEY:
2018-06-09 19:00:45 +00:00
askingpdfurl = c.work.ebookfiles().filter(asking=True)[0].file.url
assert test_pdf(askingpdfurl)
2016-12-31 03:26:16 +00:00
else:
assert test_pdf(c.work.ebookfiles().filter(asking=True)[0].file)
2018-04-07 22:38:33 +00:00
2016-03-28 18:30:05 +00:00
#Now do the same with epub
temp = NamedTemporaryFile(delete=False)
test_file_content = requests.get(settings.BOOXTREAM_TEST_EPUB_URL).content
2018-04-07 22:38:33 +00:00
2016-03-28 18:30:05 +00:00
temp.write(test_file_content)
temp.close()
try:
# now we can try putting the test pdf file into Django storage
temp_file = open(temp.name)
2018-04-07 22:38:33 +00:00
2016-03-28 18:30:05 +00:00
dj_file = DjangoFile(temp_file)
2018-04-07 22:38:33 +00:00
ebf = EbookFile(format='epub', edition=e, file=dj_file)
2016-03-28 18:30:05 +00:00
ebf.save()
2018-04-07 22:38:33 +00:00
eb = Ebook(format='epub', edition=e, url=ebf.file.url, provider='Unglue.it')
eb.save()
ebf.ebook = eb
ebf.save()
2016-03-28 18:30:05 +00:00
temp_file.close()
ebf.make_mobi()
2016-03-28 18:30:05 +00:00
finally:
# make sure we get rid of temp file
os.remove(temp.name)
#test the ask-appender
c.add_ask_to_ebfs()
2018-04-07 22:38:33 +00:00
self.assertTrue(c.work.ebookfiles().filter(asking=True, format='epub').count() > 0)
2016-12-31 03:26:16 +00:00
if settings.MOBIGEN_URL:
2018-04-07 22:38:33 +00:00
self.assertTrue(c.work.ebookfiles().filter(asking=True, format='mobi').count() > 0)
self.assertTrue(c.work.ebookfiles().filter(asking=True, ebook__active=True).count() > 0)
self.assertTrue(c.work.ebookfiles().filter(asking=False, ebook__active=True).count() == 0)
#test the unasker
c.revert_asks()
2018-04-07 22:38:33 +00:00
self.assertTrue(c.work.ebookfiles().filter(asking=True, ebook__active=True).count() == 0)
self.assertTrue(c.work.ebookfiles().filter(asking=False, ebook__active=True).count() > 0)
2014-08-28 19:29:41 +00:00
2018-06-09 19:00:45 +00:00
def test_bad_ebookfile(self):
w = Work.objects.create(title="Work 3")
e = Edition.objects.create(title=w.title, work=w)
temp = NamedTemporaryFile(delete=False)
test_file_content = "bad text file"
temp.write(test_file_content)
temp.close()
try:
# put the bad file into Django storage
temp_file = open(temp.name)
dj_file = DjangoFile(temp_file)
ebf = EbookFile(format='epub', edition=e, file=dj_file)
ebf.save()
temp_file.close()
ebf.make_mobi()
finally:
# make sure we get rid of temp file
os.remove(temp.name)
self.assertTrue(ebf.mobied < 0)
2015-01-26 17:48:16 +00:00
class MobigenTests(TestCase):
def test_convert_to_mobi(self):
"""
2018-04-07 22:38:33 +00:00
check the size of the mobi output of a Moby Dick epub
2015-01-26 17:48:16 +00:00
"""
from regluit.core.mobigen import convert_to_mobi
2016-12-30 22:24:20 +00:00
if settings.TEST_INTEGRATION:
2018-04-07 22:38:33 +00:00
output = convert_to_mobi(
"https://github.com/GITenberg/Moby-Dick--Or-The-Whale_2701/releases/download/0.2.0/Moby-Dick-Or-The-Whale.epub"
)
self.assertTrue(len(output) > 2207877)
2015-01-26 17:48:16 +00:00
@override_settings(LOCAL_TEST=True)
class LibTests(TestCase):
2016-04-09 17:21:12 +00:00
fixtures = ['initial_data.json']
class transaction:
pass
2018-04-07 22:38:33 +00:00
def test_purchase(self):
w = Work.objects.create(title="Work 1")
2018-04-07 22:38:33 +00:00
e = Edition.objects.create(title=w.title, work=w)
u = User.objects.create_user('test', 'test@example.org', 'testpass')
lu = User.objects.create_user('library', 'testu@example.org', 'testpass')
2018-04-07 22:38:33 +00:00
lib = Library.objects.create(user=lu, owner=u)
c = Campaign.objects.create(
work=w,
type=parameters.BUY2UNGLUE,
cc_date_initial=datetime(2020, 1, 1),
target=1000,
deadline=datetime(2020, 1, 1)
)
new_acq = Acq.objects.create(user=lib.user, work=c.work, license=LIBRARY)
self.assertTrue(new_acq.borrowable)
2018-04-07 22:38:33 +00:00
reserve_acq = Acq.objects.create(user=u, work=c.work, license=RESERVE, lib_acq=new_acq)
self.assertTrue(reserve_acq.borrowable)
self.assertFalse(new_acq.borrowable)
2018-04-07 22:38:33 +00:00
self.assertTrue(reserve_acq.expires < now() + timedelta(hours=25))
reserve_acq.borrow()
2018-04-07 22:38:33 +00:00
self.assertTrue(reserve_acq.expires > now() + timedelta(hours=25))
u2 = User.objects.create_user('user2', 'test2@example.org', 'testpass')
2018-04-07 22:38:33 +00:00
Hold.objects.get_or_create(library=lib, work=w, user=u2)
reserve_acq.expire_in(timedelta(seconds=0))
tasks.refresh_acqs()
2018-04-07 22:38:33 +00:00
self.assertEqual(reserve_acq.holds.count(), 0)
class GitHubTests(TestCase):
def test_ebooks_in_github_release(self):
2018-04-07 22:38:33 +00:00
(repo_owner, repo_name, repo_tag) = (
'GITenberg',
'Adventures-of-Huckleberry-Finn_76',
'0.0.50'
)
ebooks = bookloader.ebooks_in_github_release(
repo_owner,
repo_name,
tag=repo_tag,
token=settings.GITHUB_PUBLIC_TOKEN
)
expected_set = set([
('epub', u'Adventures-of-Huckleberry-Finn.epub'),
('mobi', u'Adventures-of-Huckleberry-Finn.mobi'),
('pdf', u'Adventures-of-Huckleberry-Finn.pdf')
])
self.assertEqual(set(ebooks), expected_set)
class OnixLoaderTests(TestCase):
fixtures = ['initial_data.json']
def test_load(self):
2018-04-07 22:38:33 +00:00
TEST_BOOKS = [{
'': u'',
'Author1First': u'Joseph',
'Author1Last': u'Necvatal',
'Author1Role': u'',
'Author2First': u'',
'Author2Last': u'',
'Author2Role': u'',
'Author3First': u'',
'Author3Last': u'',
'Author3Role': u'',
'AuthorBio': u'',
'AuthorsList': u'Joseph Nechvatal',
'BISACCode1': u'',
'BISACCode2': u'',
'BISACCode3': u'',
'Book-level DOI': u'10.3998/ohp.9618970.0001.001',
'ClothISBN': u'N/A',
'CopyrightYear': u'2011',
'DescriptionBrief': u'',
'DescriptionLong': u'',
'Excerpt': u'',
'FullTitle': u'Immersion into Noise',
'License': u'CC BY-SA',
'List Price in USD (paper ISBN)': u'23.99',
'ListPriceCurrencyType': u'',
'PaperISBN': u'9781607852414',
'Publisher': u'Open Humanities Press',
'SubjectListMARC': u'',
'Subtitle': u'',
'TableOfContents': u'',
'Title': u'Immersion into Noise',
'URL': u'https://doi.org/10.3998/ohp.9618970.0001.001',
'eISBN': u'N/A',
'eListPrice': u'N/A',
'ePublicationDate': u'',
'eTerritoryRights': u''
}, {
'': u'',
'CAD price eub': u'9.95',
'Title': u'That Greece Might Still Be Free',
'USD price epub': u'9.95',
'ISBN 2 with dashes': u'978-1-906924-01-0',
'Plain Text Blurb': u'When in 1821, the Greeks rose in violent revolution against the rule of the Ottoman Turks, waves of sympathy spread across Western Europe and the United States. More than a thousand volunteers set out to fight for the cause. The Philhellenes, whether they set out to recreate the Athens of Pericles, start a new crusade, or make money out of a war, all felt that Greece had unique claim on the sympathy of the world. As Lord Byron wrote, "I dreamed that Greece might still be Free"; and he died at Missolonghi trying to translate that dream into reality. William St Clair\'s meticulously researched and highly readable account of their aspirations and experiences was hailed as definitive when it was first published. Long out of print, it remains the standard account of the Philhellenic movement and essential reading for any students of the Greek War of Independence, Byron, and European Romanticism. Its relevance to more modern ethnic and religious conflicts is becoming increasingly appreciated by scholars worldwide. This revised edition includes a new introduction by Roderick Beaton, an updated bibliography and many new illustrations.',
'Cover URL': u'http://www.openbookpublishers.com/shopimages/products/cover/3',
'keywords': u'Greece; Greek History; Lord Byron; War of Independence; Philhellenes; war; history; Romanticism',
'Publication type': u'Monograph', 'GBP price epub': u'5.95', 'publication month': u'11', 'no of tables': u'',
'GBP price paperback': u'15.95', 'AUD price epub': u'9.95', 'ISBN 4 with dashes': u'978-1-906924-02-7-epub',
'DOI prefix': u'10.11647', 'License URL (human-readable summary)': u'https://creativecommons.org/licenses/by-nc-nd/2.0/',
'Contributor 5 surname': u'', 'Contributor 1 first name': u'William', 'Contributor 6 first name': u'',
'ONIX Role Code (List 17)6': u'', 'ONIX Role Code (List 17)5': u'', 'ONIX Role Code (List 17)4': u'',
'ONIX Role Code (List 17)3': u'', 'ONIX Role Code (List 17)2': u'A24', 'ONIX Role Code (List 17)1': u'A01',
'GBP price hardback': u'29.95', 'Subtitle': u'The Philhellenes in the War of Independence', 'ONIX tag6': u'',
'ISBN 3 with dashes': u'978-1-906924-02-7', 'Countries excluded': u'None', 'first edition publication date': u'39753',
'Original Language': u'English', 'ISBN 1 with dashes': u'978-1-906924-00-3', 'Contributor 4 first name': u'',
'ISBN 5 with dashes': u'978-1-906924-02-7-mobi', 'Contributor 2 surname': u'Beaton',
'License URL (our copyright tab)': u'https://www.openbookpublishers.com/isbn/9781906924003#copyright',
'BISAC subject code 1': u'HIS042000', 'BISAC subject code 3': u'HIS037060',
'BISAC subject code 2': u'HIS054000', 'BISAC subject code 5': u'',
'BISAC subject code 4': u'', 'Status': u'Active', 'Geographic rights': u'Worldwide',
'Series Name': u'', 'Contributor 5 first name': u'', 'ISSN Print with dashes': u'',
'ISBN 5': u'9781906924027mobi', 'Contributor 1 surname': u'St Clair',
'Contributor 2 first name': u'Roderick',
'Book-page permanent URL': u'http://www.openbookpublishers.com/isbn/9781906924003',
'EUR price hardback': u'36.95', 'EUR price epub': u'7.95', 'Contributor 6 surname': u'',
'current edition number (integers only)': u'1',
'Table of Content': u"Introduction by Roderick Beaton\n1. The Outbreak\n2. The Return of the Ancient Hellenes\n3. The Regiment\n4. Two Kinds of War\n5. The Cause of Greece, the Cause of Europe\n6. The Road to Marseilles\n7. Chios\n8. The Battalion of Philhellenes\n9. The Battle of Peta\n10. The Triumph of the Captains\n11. The Return Home\n12. The German Legion\n13. Knights and Crusaders\n14. Secrets of State\n15. Enter the British\n16. Lord Byron joins the Cause\n17. 'To bring Freedom and Knowledge to Greece'\n18. Arrivals at Missolonghi\n19. The Byron Brigade\n20. Essays in Regeneration\n21. The New Apostles\n22. The English Gold\n23. The Coming of the Arabs\n24. The Shade of Napoleon\n25. 'No freedom to fight for at home'\n26. French Idealism and French Cynicism\n27. Regulars Again\n28. A New Fleet\n29. Athens and Navarino\n30. America to the Rescue\n31. Later\nAppendix I: Remarks on Numbers\nAppendix II: The Principal Philhellenic Expeditions\nNotes on the Select Bibliography\nSelect Bibliography\nBibliography of Primary and Secondary Material Since 1972\nNotes\nIndex",
'no of illustrations': u'41', 'OBP Role Name6': u'', 'GBP price PDF': u'5.95',
'OBP Role Name4': u'', 'OBP Role Name5': u'', 'OBP Role Name2': u'Introduction',
'OBP Role Name3': u'', 'OBP Role Name1': u'Author', 'ONIX Status code': u'04',
'LLC (Library of Congress Codes)': u'', 'publication day': u'1',
'Copyright holder 2': u'', 'Language': u'English', 'Contributor 3 first name': u'',
'CAD price hardback': u'54.95', 'USD price paperback': u'26.95', 'ONIX tag1': u'By (author)',
'ONIX tag3': u'', 'ONIX tag2': u'Introduction by', 'ONIX tag5': u'',
'ONIX tag4': u'', 'no of audio/video': u'0', 'EUR price mobi': u'7.95',
'Version of the license': u'2.0', 'publication year': u'2008',
'CAD price paperback': u'29.95', 'Full-text URL - PDF': u'http://www.openbookpublishers.com/reader/3',
'Copyright holder 1': u'William St Clair', 'Copyright holder 3': u'',
'Short Blurb (less than 100 words)': u'When in 1821, the Greeks rose in violent revolution against Ottoman rule, waves of sympathy spread across western Europe and the USA. Inspired by a belief that Greece had a unique claim on the sympathy of the world, more than a thousand Philhellenes set out to fight for the cause. This meticulously researched and highly readable account of their aspirations and experiences has long been the standard account of the Philhellenic movement and essential reading for students of the Greek War of Independence, Byron and European Romanticism. Its relevance to more modern conflicts is also becoming increasingly appreciated.',
'BIC subject code 3': u'3JH', 'BIC subject code 2': u'1DVG', 'BIC subject code 1': u'HBJD',
'ISSN Digital with dashes': u'', 'USD price mobi': u'9.95', 'BIC subject code 5': u'',
'BIC subject code 4': u'', 'ONIX Language Code': u'eng', 'AUD price paperback': u'29.95',
'AUD price mobi': u'9.95', 'No. in the Series': u'', 'CAD price PDF': u'9.95',
'CAD price mobi': u'9.95', 'DOI suffix': u'OBP.0001', 'USD price PDF': u'9.95',
'Book-page URL': u'http://www.openbookpublishers.com/product/3',
'Academic discipline (OBP)': u'History and Biography', 'EUR price paperback': u'19.95',
'License': u'CC BY-NC-ND', 'AUD price PDF': u'9.95', 'Contributor 3 surname': u'',
'AUD price hardback': u'54.95', 'ISBN 4': u'9781906924027epub', 'no of pages': u'440',
'ISBN 2': u'9781906924010', 'ISBN 3': u'9781906924027', 'ISBN 1': u'9781906924003',
'pages': u'xxi + 419', 'Contributor 4 surname': u'', 'USD price hardback': u'48.95',
'Full-text URL - HTML': u'http://www.openbookpublishers.com/htmlreader/978-1-906924-00-3/main.html',
'GBP price mobi': u'5.95', 'Format 1': u'Paperback ', 'EUR price PDF': u'7.95', 'Format 3': u'pdf',
'Format 2': u'Hardback', 'Format 5': u'mobi', 'Format 4': u'epub', 'MARC Code1': u'aut',
'MARC Code2': u'aui', 'MARC Code3': u'', 'MARC Code4': u'', 'MARC Code5': u'',
'MARC Code6': u'', 'ISO Language Code': u'en'
}]
results = load_from_books(TEST_BOOKS)
for (book, work, edition) in results:
2018-04-07 22:38:33 +00:00
assert loaded_book_ok(book, work, edition)