Conservancy/symposion_app
6
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add test

Browse source 
Signed-off-by: Hiroshi Miura <miurahr@linux.com>
This commit is contained in:
Hiroshi Miura 2015-06-21 13:57:05 +09:00
parent fbf356b335
commit 999c458c1b
1 changed files with 307 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

307
symposion/sponsorship/tests.py Normal file
View file

@ -0,0 +1,307 @@
from cStringIO import StringIO
import os
import shutil
import tempfile
from zipfile import ZipFile
from django.conf import settings
from django.contrib.auth.models import User
from django.core.exceptions import ValidationError
from django.core.urlresolvers import reverse
from django.test import TestCase
from django.test.utils import override_settings
from pycon.sponsorship.models import Benefit, Sponsor, SponsorBenefit,\
SponsorLevel
from symposion.conference.models import current_conference
class TestSponsorZipDownload(TestCase):
def setUp(self):
self.user = User.objects.create_user(username='joe',
email='joe@example.com',
password='joe')
self.user.is_staff = True
self.user.save()
self.url = reverse("sponsor_zip_logos")
self.assertTrue(self.client.login(username='joe@example.com',
password='joe'))
# we need a sponsor
conference = current_conference()
self.sponsor_level = SponsorLevel.objects.create(
conference=conference, name="Lead", cost=1)
self.sponsor = Sponsor.objects.create(
name="Big Daddy",
level=self.sponsor_level,
active=True,
)
# Create our benefits, of various types
self.text_benefit = Benefit.objects.create(name="text", type="text")
self.file_benefit = Benefit.objects.create(name="file", type="file")
# These names must be spelled exactly this way:
self.weblogo_benefit = Benefit.objects.create(name="Web logo", type="weblogo")
self.printlogo_benefit = Benefit.objects.create(name="Print logo", type="file")
self.advertisement_benefit = Benefit.objects.create(name="Advertisement", type="file")
def validate_response(self, rsp, names_and_sizes):
# Ensure a response from the view looks right, contains a valid
# zip archive, has files with the right names and sizes.
self.assertEqual("application/zip", rsp['Content-type'])
prefix = settings.CONFERENCE_URL_PREFIXES[settings.CONFERENCE_ID]
self.assertEqual(
'attachment; filename="pycon_%s_sponsorlogos.zip"' % prefix,
rsp['Content-Disposition'])
zipfile = ZipFile(StringIO(rsp.content), "r")
# Check out the zip - testzip() returns None if no errors found
self.assertIsNone(zipfile.testzip())
# Compare contents to what is expected
infolist = zipfile.infolist()
self.assertEqual(len(names_and_sizes), len(infolist))
for info, name_and_size in zip(infolist, names_and_sizes):
name, size = name_and_size
self.assertEqual(name, info.filename)
self.assertEqual(size, info.file_size)
def make_temp_file(self, name, size=0):
# Create a temp file with the given name and size under self.temp_dir
path = os.path.join(self.temp_dir, name)
with open(path, "wb") as f:
f.write(size * "x")
def test_must_be_logged_in(self):
# Must be logged in to use the view
# If not logged in, doesn't redirect, just serves up a login view
self.client.logout()
rsp = self.client.get(self.url)
self.assertEqual(200, rsp.status_code)
self.assertIn("""<body class="login">""", rsp.content)
def test_must_be_staff(self):
# Only staff can use the view
# If not staff, doesn't show error, just serves up a login view
# Also, the dashboard doesn't show the download button
self.user.is_staff = False
self.user.save()
rsp = self.client.get(self.url)
self.assertEqual(200, rsp.status_code)
self.assertIn("""<body class="login">""", rsp.content)
rsp = self.client.get(reverse('dashboard'))
self.assertNotIn(self.url, rsp.content)
def test_no_files(self):
# If there are no sponsor files, we still work
# And the dashboard shows our download button
rsp = self.client.get(self.url)
self.validate_response(rsp, [])
rsp = self.client.get(reverse('dashboard'))
self.assertIn(self.url, rsp.content)
def test_different_benefit_types(self):
# We only get files from the benefits named "Print logo" and "Web logo"
# And we ignore any non-existent files
try:
# Create a temp dir for media files
self.temp_dir = tempfile.mkdtemp()
with override_settings(MEDIA_ROOT=self.temp_dir):
# Give our sponsor some benefits
SponsorBenefit.objects.create(
sponsor=self.sponsor,
benefit=self.text_benefit,
text="Foo!"
)
self.make_temp_file("file1", 10)
SponsorBenefit.objects.create(
sponsor=self.sponsor,
benefit=self.file_benefit,
upload="file1"
)
self.make_temp_file("file2", 20)
SponsorBenefit.objects.create(
sponsor=self.sponsor,
benefit=self.weblogo_benefit,
upload="file2"
)
# Benefit whose file is missing from the disk
SponsorBenefit.objects.create(
sponsor=self.sponsor,
benefit=self.weblogo_benefit,
upload="file3"
)
# print logo benefit
self.make_temp_file("file4", 40)
SponsorBenefit.objects.create(
sponsor=self.sponsor,
benefit=self.printlogo_benefit,
upload="file4"
)
self.make_temp_file("file5", 50)
SponsorBenefit.objects.create(
sponsor=self.sponsor,
benefit=self.advertisement_benefit,
upload="file5"
)
rsp = self.client.get(self.url)
expected = [
('web_logos/lead/big_daddy/file2', 20),
('print_logos/lead/big_daddy/file4', 40),
('advertisement/lead/big_daddy/file5', 50)
]
self.validate_response(rsp, expected)
finally:
if hasattr(self, 'temp_dir'):
# Clean up any temp media files
shutil.rmtree(self.temp_dir)
def test_file_org(self):
# The zip file is organized into directories:
# {print_logos,web_logos,advertisement}/<sponsor_level>/<sponsor_name>/<filename>
# Add another sponsor at a different sponsor level
conference = current_conference()
self.sponsor_level2 = SponsorLevel.objects.create(
conference=conference, name="Silly putty", cost=1)
self.sponsor2 = Sponsor.objects.create(
name="Big Mama",
level=self.sponsor_level2,
active=True,
)
#
try:
# Create a temp dir for media files
self.temp_dir = tempfile.mkdtemp()
with override_settings(MEDIA_ROOT=self.temp_dir):
# Give our sponsors some benefits
self.make_temp_file("file1", 10)
SponsorBenefit.objects.create(
sponsor=self.sponsor,
benefit=self.weblogo_benefit,
upload="file1"
)
# print logo benefit
self.make_temp_file("file2", 20)
SponsorBenefit.objects.create(
sponsor=self.sponsor,
benefit=self.printlogo_benefit,
upload="file2"
)
# Sponsor 2
self.make_temp_file("file3", 30)
SponsorBenefit.objects.create(
sponsor=self.sponsor2,
benefit=self.weblogo_benefit,
upload="file3"
)
# print logo benefit
self.make_temp_file("file4", 42)
SponsorBenefit.objects.create(
sponsor=self.sponsor2,
benefit=self.printlogo_benefit,
upload="file4"
)
# ad benefit
self.make_temp_file("file5", 55)
SponsorBenefit.objects.create(
sponsor=self.sponsor2,
benefit=self.advertisement_benefit,
upload="file5"
)
rsp = self.client.get(self.url)
expected = [
('web_logos/lead/big_daddy/file1', 10),
('web_logos/silly_putty/big_mama/file3', 30),
('print_logos/lead/big_daddy/file2', 20),
('print_logos/silly_putty/big_mama/file4', 42),
('advertisement/silly_putty/big_mama/file5', 55),
]
self.validate_response(rsp, expected)
finally:
if hasattr(self, 'temp_dir'):
# Clean up any temp media files
shutil.rmtree(self.temp_dir)
class TestBenefitValidation(TestCase):
"""
It should not be possible to save a SponsorBenefit if it has the
wrong kind of data in it - e.g. a text-type benefit cannot have
an uploaded file, and vice-versa.
"""
def setUp(self):
# we need a sponsor
conference = current_conference()
self.sponsor_level = SponsorLevel.objects.create(
conference=conference, name="Lead", cost=1)
self.sponsor = Sponsor.objects.create(
name="Big Daddy",
level=self.sponsor_level,
)
# Create our benefit types
self.text_type = Benefit.objects.create(name="text", type="text")
self.file_type = Benefit.objects.create(name="file", type="file")
self.weblogo_type = Benefit.objects.create(name="log", type="weblogo")
self.simple_type = Benefit.objects.create(name="simple", type="simple")
def validate(self, should_work, benefit_type, upload, text):
obj = SponsorBenefit(
benefit=benefit_type,
sponsor=self.sponsor,
upload=upload,
text=text
)
if should_work:
obj.save()
else:
with self.assertRaises(ValidationError):
obj.save()
def test_text_has_text(self):
self.validate(True, self.text_type, upload=None, text="Some text")
def test_text_has_upload(self):
self.validate(False, self.text_type, upload="filename", text='')
def test_text_has_both(self):
self.validate(False, self.text_type, upload="filename", text="Text")
def test_file_has_text(self):
self.validate(False, self.file_type, upload=None, text="Some text")
def test_file_has_upload(self):
self.validate(True, self.file_type, upload="filename", text='')
def test_file_has_both(self):
self.validate(False, self.file_type, upload="filename", text="Text")
def test_weblogo_has_text(self):
self.validate(False, self.weblogo_type, upload=None, text="Some text")
def test_weblogo_has_upload(self):
self.validate(True, self.weblogo_type, upload="filename", text='')
def test_weblogo_has_both(self):
self.validate(False, self.weblogo_type, upload="filename", text="Text")
def test_simple_has_neither(self):
self.validate(True, self.simple_type, upload=None, text='')
def test_simple_has_text(self):
self.validate(True, self.simple_type, upload=None, text="Some text")
def test_simple_has_upload(self):
self.validate(False, self.simple_type, upload="filename", text='')
def test_simple_has_both(self):
self.validate(False, self.simple_type, upload="filename", text="Text")