259 lines
9.2 KiB
Python
Executable File
259 lines
9.2 KiB
Python
Executable File
from decimal import Decimal as D
|
|
from datetime import datetime, timedelta
|
|
|
|
from django.test import TestCase
|
|
from django.utils import unittest
|
|
from django.db import IntegrityError
|
|
from django.contrib.auth.models import User
|
|
|
|
from regluit.payment.models import Transaction
|
|
from regluit.core.models import Campaign, Work, UnglueitError
|
|
from regluit.core import bookloader, models, search
|
|
from regluit.payment.parameters import PAYMENT_TYPE_AUTHORIZATION
|
|
|
|
|
|
class TestBookLoader(TestCase):
|
|
|
|
def test_add_by_isbn(self):
|
|
# edition
|
|
edition = bookloader.add_by_isbn('0441012035')
|
|
self.assertEqual(edition.title, 'Neuromancer')
|
|
self.assertEqual(edition.publication_date, '2004')
|
|
self.assertEqual(edition.publisher, 'Ace Books')
|
|
self.assertEqual(edition.isbn_10, '0441012035')
|
|
self.assertEqual(edition.isbn_13, '9780441012039')
|
|
self.assertEqual(edition.googlebooks_id, "2NyiPwAACAAJ")
|
|
self.assertEqual(edition.language, "en")
|
|
|
|
# subjects
|
|
subject_names = [subject.name for subject in edition.subjects.all()]
|
|
self.assertEqual(len(subject_names), 11)
|
|
self.assertTrue('Japan' in subject_names)
|
|
|
|
# authors
|
|
self.assertEqual(edition.authors.all().count(), 1)
|
|
self.assertEqual(edition.authors.all()[0].name, 'William Gibson')
|
|
|
|
# work
|
|
self.assertTrue(edition.work)
|
|
|
|
def test_double_add(self):
|
|
bookloader.add_by_isbn('0441012035')
|
|
bookloader.add_by_isbn('0441012035')
|
|
self.assertEqual(models.Edition.objects.all().count(), 1)
|
|
self.assertEqual(models.Author.objects.all().count(), 1)
|
|
self.assertEqual(models.Work.objects.all().count(), 1)
|
|
self.assertEqual(models.Subject.objects.all().count(), 11)
|
|
|
|
def test_missing_isbn(self):
|
|
e = bookloader.add_by_isbn('0139391401')
|
|
self.assertEqual(e, None)
|
|
|
|
def test_thingisbn(self):
|
|
isbns = bookloader.thingisbn('0441012035')
|
|
self.assertTrue(len(isbns) > 20)
|
|
self.assertTrue('0441012035' in isbns)
|
|
self.assertTrue('3453313895' in isbns)
|
|
|
|
def test_add_related(self):
|
|
# add one edition
|
|
edition = bookloader.add_by_isbn('0441012035')
|
|
self.assertEqual(models.Edition.objects.count(), 1)
|
|
self.assertEqual(models.Work.objects.count(), 1)
|
|
|
|
# ask for related editions to be added using the work we just created
|
|
bookloader.add_related('0441012035')
|
|
self.assertTrue(models.Edition.objects.count() > 20)
|
|
self.assertEqual(models.Work.objects.count(), 1)
|
|
self.assertTrue(edition.work.editions.count() > 20)
|
|
|
|
def test_merge_works(self):
|
|
# add two editions and see that there are two stub works
|
|
e1 = bookloader.add_by_isbn('0465019358')
|
|
e2 = bookloader.add_by_isbn('1458776204')
|
|
self.assertTrue(e1)
|
|
self.assertTrue(e2)
|
|
self.assertEqual(models.Work.objects.count(), 2)
|
|
|
|
# add the stub works to a wishlist
|
|
user = User.objects.create_user('test', 'test@example.com', 'testpass')
|
|
user.wishlist.works.add(e1.work)
|
|
user.wishlist.works.add(e2.work)
|
|
|
|
# create campaigns for the stub works
|
|
c1 = models.Campaign.objects.create(
|
|
name=e1.work.title,
|
|
work=e2.work,
|
|
description='Test Campaign 1',
|
|
deadline=datetime.now(),
|
|
target=D('1000.00'),
|
|
)
|
|
c2 = models.Campaign.objects.create(
|
|
name=e2.work.title,
|
|
work=e2.work,
|
|
description='Test Campaign 2',
|
|
deadline=datetime.now(),
|
|
target=D('1000.00'),
|
|
)
|
|
|
|
# now add related edition to make sure Works get merged
|
|
bookloader.add_related('1458776204')
|
|
self.assertEqual(models.Work.objects.count(), 1)
|
|
|
|
# and that relevant Campaigns and Wishlists are updated
|
|
self.assertEqual(c1.work, c2.work)
|
|
self.assertEqual(user.wishlist.works.all().count(), 1)
|
|
|
|
def test_oclc(self):
|
|
edition = bookloader.add_by_oclc('1246014')
|
|
self.assertEqual(edition.title, 'The Latin language')
|
|
|
|
def test_ebook(self):
|
|
edition = bookloader.add_by_oclc('1246014')
|
|
self.assertEqual(edition.ebooks.count(), 1)
|
|
ebook = edition.ebooks.all()[0]
|
|
self.assertEqual(ebook.url, 'http://books.google.com/books/download/The_Latin_language.epub?id=U3FXAAAAYAAJ&ie=ISO-8859-1&output=epub&source=gbs_api')
|
|
self.assertEqual(ebook.provider, 'google')
|
|
|
|
def test_add_no_ebook(self):
|
|
# this edition lacks an ebook, but we should still be able to load it
|
|
e = bookloader.add_by_isbn('0465019358')
|
|
self.assertTrue(e)
|
|
|
|
class SearchTests(TestCase):
|
|
|
|
def test_basic_search(self):
|
|
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('image'))
|
|
self.assertTrue(r.has_key('publisher'))
|
|
self.assertTrue(r.has_key('isbn_10'))
|
|
self.assertTrue(r.has_key('googlebooks_id'))
|
|
|
|
def test_googlebooks_search(self):
|
|
response = search.googlebooks_search('melville')
|
|
self.assertEqual(len(response['items']), 10)
|
|
|
|
|
|
class CampaignTests(TestCase):
|
|
|
|
def test_required_fields(self):
|
|
# a campaign must have a target, deadline and a work
|
|
|
|
c = Campaign()
|
|
self.assertRaises(IntegrityError, c.save)
|
|
|
|
c = Campaign(target=D('1000.00'))
|
|
self.assertRaises(IntegrityError, c.save)
|
|
|
|
c = Campaign(target=D('1000.00'), deadline=datetime(2012, 1, 1))
|
|
self.assertRaises(IntegrityError, c.save)
|
|
|
|
w = Work()
|
|
w.save()
|
|
c = Campaign(target=D('1000.00'), deadline=datetime(2012, 1, 1), work=w)
|
|
c.save()
|
|
|
|
def test_campaign_status(self):
|
|
w = Work()
|
|
w.save()
|
|
# INITIALIZED
|
|
c1 = Campaign(target=D('1000.00'),deadline=datetime(2012,1,1),work=w)
|
|
c1.save()
|
|
self.assertEqual(c1.status, 'INITIALIZED')
|
|
# ACTIVATED
|
|
c2 = Campaign(target=D('1000.00'),deadline=datetime(2012,1,1),work=w)
|
|
c2.save()
|
|
self.assertEqual(c2.status, 'INITIALIZED')
|
|
c2.activate()
|
|
self.assertEqual(c2.status, 'ACTIVE')
|
|
# SUSPENDED
|
|
c2.suspend(reason="for testing")
|
|
self.assertEqual(c2.status, 'SUSPENDED')
|
|
# RESUMING
|
|
c2.resume()
|
|
self.assertEqual(c2.suspended, None)
|
|
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
|
|
c3 = Campaign(target=D('1000.00'),deadline=datetime.utcnow() - timedelta(days=1),work=w)
|
|
c3.save()
|
|
c3.activate()
|
|
self.assertEqual(c3.status, 'UNSUCCESSFUL')
|
|
# SUCCESSFUL
|
|
c4 = Campaign(target=D('1000.00'),deadline=datetime.utcnow() - timedelta(days=1),work=w)
|
|
c4.save()
|
|
c4.activate()
|
|
|
|
t = Transaction()
|
|
t.amount = D('1234.00')
|
|
t.type = PAYMENT_TYPE_AUTHORIZATION
|
|
t.status = 'ACTIVE'
|
|
t.campaign = c4
|
|
t.save()
|
|
self.assertEqual(c4.status, 'SUCCESSFUL')
|
|
|
|
# ACTIVE
|
|
c4.deadline = datetime.utcnow() + timedelta(days=1)
|
|
c4.save()
|
|
self.assertEqual(c4.status, 'ACTIVE')
|
|
|
|
# WITHDRAWN
|
|
c5 = Campaign(target=D('1000.00'),deadline=datetime(2012,1,1),work=w)
|
|
c5.save()
|
|
c5.activate().withdraw('testing')
|
|
self.assertEqual(c5.status, 'WITHDRAWN')
|
|
|
|
|
|
class SettingsTest(TestCase):
|
|
|
|
def test_dev_me_alignment(self):
|
|
from regluit.settings import me, dev
|
|
self.assertEqual(set(me.__dict__.keys()) ^ set(dev.__dict__.keys()), set([]))
|
|
|
|
def test_prod_me_alignment(self):
|
|
from regluit.settings import me, prod
|
|
self.assertEqual(set(me.__dict__.keys()) ^ set(prod.__dict__.keys()), set([]))
|
|
|
|
|
|
class WishlistTest(TestCase):
|
|
|
|
def test_add_remove(self):
|
|
# add a work to a user's wishlist
|
|
user = User.objects.create_user('test', 'test@example.com', 'testpass')
|
|
edition = bookloader.add_by_isbn('0441012035')
|
|
work = edition.work
|
|
user.wishlist.works.add(work)
|
|
self.assertEqual(user.wishlist.works.count(), 1)
|
|
user.wishlist.works.remove(work)
|
|
self.assertEqual(user.wishlist.works.count(), 0)
|
|
|
|
class SettingsTest(TestCase):
|
|
|
|
def test_dev_me_alignment(self):
|
|
try:
|
|
from regluit.settings import me, dev
|
|
except:
|
|
return
|
|
|
|
self.assertEqual(set(me.__dict__.keys()) ^ set(dev.__dict__.keys()), set([]))
|
|
|
|
|
|
def test_prod_me_alignment(self):
|
|
from regluit.settings import me, prod
|
|
self.assertEqual(set(me.__dict__.keys()) ^ set(prod.__dict__.keys()), set([]))
|
|
|
|
def suite():
|
|
|
|
testcases = [TestBookLoader, SearchTests, CampaignTests, WishlistTest]
|
|
suites = unittest.TestSuite([unittest.TestLoader().loadTestsFromTestCase(testcase) for testcase in testcases])
|
|
suites.addTest(SettingsTest('test_dev_me_alignment')) # leave out alignment with prod test right now
|
|
return suites
|
|
|