class PdfTest(TestCase):
def setUp(self):
self.pdf = MediaFile(file=File(open('testapp/fixtures/rst-cheatsheet.pdf')))
self.pdf.save()
def tearDown(self):
pdf_variations = Variation.objects.filter(
content_type = ContentType.objects.get_for_model(self.pdf),
object_id = self.pdf.pk
)
for variation in pdf_variations:
variation.delete()
self.pdf.delete()
def test_page_range(self):
from pyPdf import PdfFileReader
self.variation = Variation(
content_object = self.pdf,
spec = 'mediavariations.contrib.pypdf.specs.PageRange',
options = simplejson.dumps({
'start' : 0,
'stop' : 1
})
)
self.variation.save()
reader = PdfFileReader(self.variation.file.file)
self.assertEqual(reader.getNumPages(), 1)
class BlitlineTest(TestCase):
def setUp(self):
self.mediafile = MediaFile(file=File(open('testapp/fixtures/elephant_test_image.jpeg')))
self.mediafile.save()
def tearDown(self):
"""
delete the mediafile
"""
mediafile_variations = Variation.objects.filter(
content_type = ContentType.objects.get_for_model(self.mediafile),
object_id = self.mediafile.pk
)
for variation in mediafile_variations:
variation.delete()
self.mediafile.delete()
def test_setup(self):
self.assertEqual(self.mediafile.type, 'image')
self.assertEqual(get_image_dimensions(self.mediafile.file), (404, 346))
def test_blitline(self):
self.variation = Variation(
content_object=self.mediafile,
spec = 'mediavariations.contrib.blitline.specs.Generic',
options = '''
{"functions":
[{ "name": "crop",
"params" : { "x": 5, "y": 5, "width": 40,"height": 40 }
}]
}
'''
)
self.variation.save()
# auto medialfield discovery
self.assertEqual(self.variation.field, 'file')
# wait for image to be processed
while (not self.variation.processed):
self.variation.get_progress()
sleep(1)
# test image
self.assertEqual(get_image_dimensions(self.variation.file), (40, 40))
def test_templatetag(self):
from mediavariations.templatetags.mediavariations import mediavariation
self.variated_url = mediavariation(self.mediafile, 'blitline')
path, filename = os.path.split(self.mediafile.file.name)
basename, ext = os.path.splitext(filename)
self.assertTrue(basename in self.variated_url)