Exemplo n.º 1
0
class MainTestCase(unittest.TestCase):
    def setUp(self):
        self.url = "http://www.example.com"
        self.output_file = "/tmp/example.pdf"
        self.wkhtmltopdf = WKhtmlToPdf(self.url, self.output_file)
    
    def test_wkhtmltopdf_options(self):
        # test default options
        self.assertEqual(self.wkhtmltopdf._create_option_list(), ['--enable-plugins', '--orientation Portrait', '--dpi 100'])
        
        # test flash_plugin option
        self.wkhtmltopdf.flash_plugin = False
        self.assertEqual(self.wkhtmltopdf._create_option_list(), ['--orientation Portrait', '--dpi 100'])
        
        # test orientation option
        self.wkhtmltopdf.orientation = 'Landscape'
        self.assertEqual(self.wkhtmltopdf._create_option_list(), ['--orientation Landscape', '--dpi 100'])
        
        # test dpi option
        self.wkhtmltopdf.dpi = 300
        self.assertEqual(self.wkhtmltopdf._create_option_list(), ['--orientation Landscape', '--dpi 300'])
        
        # test disable_javascript option
        self.wkhtmltopdf.disable_javascript = True
        self.assertEqual(self.wkhtmltopdf._create_option_list(), ['--disable-javascript', '--orientation Landscape', '--dpi 300'])
        
        # test delay option
        self.wkhtmltopdf.delay = 1
        self.assertEqual(self.wkhtmltopdf._create_option_list(), ['--disable-javascript', '--redirect-delay 1', '--orientation Landscape', '--dpi 300'])
        
        # test no_background option
        self.wkhtmltopdf.no_background = True
        self.assertEqual(self.wkhtmltopdf._create_option_list(), ['--disable-javascript', '--no-background', '--redirect-delay 1', '--orientation Landscape', '--dpi 300'])
        
        # test grayscale option
        self.wkhtmltopdf.grayscale = True
        self.assertEqual(self.wkhtmltopdf._create_option_list(), ['--disable-javascript', '--no-background', '--grayscale', '--redirect-delay 1', '--orientation Landscape', '--dpi 300'])
    
    def test_wkhtmltopdf_callable(self):
        wkhtmltopdf(self.url, self.output_file)
        self.assertTrue(os.path.exists(self.output_file))
    
    def tearDown(self):
        try:
            os.remove(self.output_file)
        except OSError:
            pass
Exemplo n.º 2
0
class MainTestCase(unittest.TestCase):
    def setUp(self):
        self.url = "http://www.example.com"
        self.output_file = "/tmp/example.pdf"
        self.wkhtmltopdf = WKhtmlToPdf(self.url, self.output_file)

    def test_wkhtmltopdf_options(self):
        # test default options
        self.assertEqual(
            self.wkhtmltopdf._create_option_list(),
            ['--enable-plugins', '--orientation Portrait', '--dpi 100'])

        # test flash_plugin option
        self.wkhtmltopdf.flash_plugin = False
        self.assertEqual(self.wkhtmltopdf._create_option_list(),
                         ['--orientation Portrait', '--dpi 100'])

        # test orientation option
        self.wkhtmltopdf.orientation = 'Landscape'
        self.assertEqual(self.wkhtmltopdf._create_option_list(),
                         ['--orientation Landscape', '--dpi 100'])

        # test dpi option
        self.wkhtmltopdf.dpi = 300
        self.assertEqual(self.wkhtmltopdf._create_option_list(),
                         ['--orientation Landscape', '--dpi 300'])

        # test disable_javascript option
        self.wkhtmltopdf.disable_javascript = True
        self.assertEqual(
            self.wkhtmltopdf._create_option_list(),
            ['--disable-javascript', '--orientation Landscape', '--dpi 300'])

        # test delay option
        self.wkhtmltopdf.delay = 1
        self.assertEqual(self.wkhtmltopdf._create_option_list(), [
            '--disable-javascript', '--redirect-delay 1',
            '--orientation Landscape', '--dpi 300'
        ])

        # test no_background option
        self.wkhtmltopdf.no_background = True
        self.assertEqual(self.wkhtmltopdf._create_option_list(), [
            '--disable-javascript', '--no-background', '--redirect-delay 1',
            '--orientation Landscape', '--dpi 300'
        ])

        # test grayscale option
        self.wkhtmltopdf.grayscale = True
        self.assertEqual(self.wkhtmltopdf._create_option_list(), [
            '--disable-javascript', '--no-background', '--grayscale',
            '--redirect-delay 1', '--orientation Landscape', '--dpi 300'
        ])

    def test_wkhtmltopdf_callable(self):
        wkhtmltopdf(self.url, self.output_file)
        self.assertTrue(os.path.exists(self.output_file))

    def tearDown(self):
        try:
            os.remove(self.output_file)
        except OSError:
            pass