aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorSimon Law <simon.law@ecometrica.com>2012-07-20 14:30:13 -0400
committerSimon Law <simon.law@ecometrica.com>2012-07-20 14:30:13 -0400
commitdf83b5a8045d1fe69b3ae4d5ffdc5f98069862dc (patch)
tree4fd04e0396effa6e9dfb6716728e35fd8f88674b
parent10680259b0843d3688dbae241f8abb5c366ea4c5 (diff)
downloaddjango-wkhtmltopdf-df83b5a8045d1fe69b3ae4d5ffdc5f98069862dc.tar.gz
django-wkhtmltopdf-df83b5a8045d1fe69b3ae4d5ffdc5f98069862dc.tar.bz2
django-wkhtmltopdf-df83b5a8045d1fe69b3ae4d5ffdc5f98069862dc.zip
test_wkhtmltopdf tests that wkhtmltopdf generates a PDF.
-rw-r--r--wkhtmltopdf/tests.py14
1 files changed, 13 insertions, 1 deletions
diff --git a/wkhtmltopdf/tests.py b/wkhtmltopdf/tests.py
index 0249cb5..f50d8f2 100644
--- a/wkhtmltopdf/tests.py
+++ b/wkhtmltopdf/tests.py
@@ -4,10 +4,22 @@ import os
from django.test import TestCase
-from .utils import template_to_temp_file
+from .utils import template_to_temp_file, wkhtmltopdf
class TestUtils(TestCase):
+ def test_wkhtmltopdf(self):
+ """Should run wkhtmltopdf to generate a PDF"""
+ title = 'A test template.'
+ temp_file = template_to_temp_file('sample.html', {'title': title})
+ pdf_output = None
+ try:
+ pdf_output = wkhtmltopdf(pages=[temp_file])
+ finally:
+ if os.path.exists(temp_file):
+ os.remove(temp_file)
+ self.assertTrue(pdf_output.startswith('%PDF'), pdf_output)
+
def test_template_to_temp_file(self):
"""Should render a template to a temporary file."""
title = 'A test template.'