1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
# -*- coding: utf-8 -*-
"""
Command line test
~~~~~~~~~~~~~~~~~
:copyright: Copyright 2006-2014 by the Pygments team, see AUTHORS.
:license: BSD, see LICENSE for details.
"""
from __future__ import print_function
import io
import os
import sys
import tempfile
import unittest
from pygments import highlight
from pygments.util import StringIO, BytesIO
from pygments.cmdline import main as cmdline_main
import support
TESTFILE, TESTDIR = support.location(__file__)
def run_cmdline(*args):
saved_stdout = sys.stdout
saved_stderr = sys.stderr
if sys.version_info > (3,):
stdout_buffer = BytesIO()
stderr_buffer = BytesIO()
new_stdout = sys.stdout = io.TextIOWrapper(stdout_buffer, 'utf-8')
new_stderr = sys.stderr = io.TextIOWrapper(stderr_buffer, 'utf-8')
else:
stdout_buffer = new_stdout = sys.stdout = StringIO()
stderr_buffer = new_stderr = sys.stderr = StringIO()
try:
ret = cmdline_main(["pygmentize"] + list(args))
finally:
sys.stdout = saved_stdout
sys.stderr = saved_stderr
new_stdout.flush()
new_stderr.flush()
out, err = stdout_buffer.getvalue().decode('utf-8'), \
stderr_buffer.getvalue().decode('utf-8')
return (ret, out, err)
class CmdLineTest(unittest.TestCase):
def test_L_opt(self):
c, o, e = run_cmdline("-L")
self.assertEqual(c, 0)
self.assertTrue("Lexers" in o and "Formatters" in o and
"Filters" in o and "Styles" in o)
c, o, e = run_cmdline("-L", "lexer")
self.assertEqual(c, 0)
self.assertTrue("Lexers" in o and "Formatters" not in o)
c, o, e = run_cmdline("-L", "lexers")
self.assertEqual(c, 0)
def test_O_opt(self):
filename = TESTFILE
c, o, e = run_cmdline("-Ofull=1,linenos=true,foo=bar",
"-fhtml", filename)
self.assertEqual(c, 0)
self.assertTrue("<html" in o)
self.assertTrue('class="linenos"' in o)
def test_P_opt(self):
filename = TESTFILE
c, o, e = run_cmdline("-Pfull", "-Ptitle=foo, bar=baz=,",
"-fhtml", filename)
self.assertEqual(c, 0)
self.assertTrue("<title>foo, bar=baz=,</title>" in o)
def test_F_opt(self):
filename = TESTFILE
c, o, e = run_cmdline("-Fhighlight:tokentype=Name.Blubb,"
"names=TESTFILE filename",
"-fhtml", filename)
self.assertEqual(c, 0)
self.assertTrue('<span class="n-Blubb' in o)
def test_H_opt(self):
c, o, e = run_cmdline("-H", "formatter", "html")
self.assertEqual(c, 0)
self.assertTrue('HTML' in o)
def test_S_opt(self):
c, o, e = run_cmdline("-S", "default", "-f", "html", "-O", "linenos=1")
self.assertEqual(c, 0)
def test_invalid_opts(self):
for opts in [("-L", "-lpy"), ("-L", "-fhtml"), ("-L", "-Ox"),
("-a",), ("-Sst", "-lpy"), ("-H",),
("-H", "formatter")]:
self.assertTrue(run_cmdline(*opts)[0] == 2)
def test_normal(self):
# test that cmdline gives the same output as library api
from pygments.lexers import PythonLexer
from pygments.formatters import HtmlFormatter
filename = TESTFILE
with open(filename, 'rb') as fp:
code = fp.read()
output = highlight(code, PythonLexer(), HtmlFormatter())
c, o, e = run_cmdline("-lpython", "-fhtml", filename)
self.assertEqual(o, output)
self.assertEqual(e, "")
self.assertEqual(c, 0)
def test_outfile(self):
# test that output file works with and without encoding
fd, name = tempfile.mkstemp()
os.close(fd)
for opts in [['-fhtml', '-o', name, TESTFILE],
['-flatex', '-o', name, TESTFILE],
['-fhtml', '-o', name, '-O', 'encoding=utf-8', TESTFILE]]:
try:
self.assertEqual(run_cmdline(*opts)[0], 0)
finally:
os.unlink(name)
|
