From a377496b5a9bfab824b52a7f2e69a64f194930b8 Mon Sep 17 00:00:00 2001
From: Simon Law <simon.law@ecometrica.com>
Date: Fri, 20 Jul 2012 15:52:03 -0400
Subject: Reliable command-line argument parsing for wkhtmltopdf().

The API for wkhtmltopdf has changed. Long arguments that take no
parameters now use True and not the empty string. In addition,
argument-parameters may now be Unicode.
---
 wkhtmltopdf/tests.py | 38 ++++++++++++++++++++++++++++++++++++--
 1 file changed, 36 insertions(+), 2 deletions(-)

(limited to 'wkhtmltopdf/tests.py')

diff --git a/wkhtmltopdf/tests.py b/wkhtmltopdf/tests.py
index f50d8f2..3e8359f 100644
--- a/wkhtmltopdf/tests.py
+++ b/wkhtmltopdf/tests.py
@@ -1,24 +1,58 @@
+# -*- coding: utf-8 -*-
+
 from __future__ import absolute_import
 
+from StringIO import StringIO
 import os
+import sys
 
 from django.test import TestCase
 
-from .utils import template_to_temp_file, wkhtmltopdf
+from .subprocess import CalledProcessError
+from .utils import _options_to_args, template_to_temp_file, wkhtmltopdf
 
 
 class TestUtils(TestCase):
+    def setUp(self):
+        # Clear standard error
+        self._stderr = sys.stderr
+        sys.stderr = open(os.devnull, 'w')
+
+    def tearDown(self):
+        sys.stderr = self._stderr
+
+    def test_options_to_args(self):
+        self.assertEqual(_options_to_args(), [])
+        self.assertEqual(_options_to_args(heart=u'♥', verbose=True,
+                                          file_name='file-name'),
+                         ['--file-name', 'file-name',
+                          '--heart', u'♥',
+                          '--verbose'])
+
     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:
+            # Standard call
             pdf_output = wkhtmltopdf(pages=[temp_file])
+            self.assertTrue(pdf_output.startswith('%PDF'), pdf_output)
+
+            # Single page
+            pdf_output = wkhtmltopdf(pages=temp_file)
+            self.assertTrue(pdf_output.startswith('%PDF'), pdf_output)
+
+            # Unicode
+            pdf_output = wkhtmltopdf(pages=[temp_file], title=u'♥')
+            self.assertTrue(pdf_output.startswith('%PDF'), pdf_output)
+
+            # Invalid arguments
+            self.assertRaises(CalledProcessError,
+                              wkhtmltopdf, pages=[])
         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."""
-- 
cgit v1.2.3