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
|
# -*- coding: utf-8 -*-
"""
Command line test
~~~~~~~~~~~~~~~~~
:copyright: 2006-2007 by Georg Brandl.
:license: BSD, see LICENSE for more details.
"""
# Test the command line interface
import sys, os
import unittest
import StringIO
from pygments import cmdline_main, highlight
def run_cmdline(*args):
saved_stdout = sys.stdout
saved_stderr = sys.stderr
new_stdout = sys.stdout = StringIO.StringIO()
new_stderr = sys.stderr = StringIO.StringIO()
try:
ret = cmdline_main(["pygmentize"] + list(args))
finally:
sys.stdout = saved_stdout
sys.stderr = saved_stderr
return (ret, new_stdout.getvalue(), new_stderr.getvalue())
class CmdLineTest(unittest.TestCase):
def test_L_opt(self):
c, o, e = run_cmdline("-L")
self.assertEquals(c, 0)
self.assert_("Lexers" in o and "Formatters" in o and
"Filters" in o and "Styles" in o)
c, o, e = run_cmdline("-L", "lexer")
self.assertEquals(c, 0)
self.assert_("Lexers" in o and "Formatters" not in o)
c, o, e = run_cmdline("-L", "lexers")
self.assertEquals(c, 0)
def test_O_opt(self):
filename = os.path.join(testdir, testfile)
c, o, e = run_cmdline("-Ofull=1,linenos=true", "-fhtml", filename)
self.assertEquals(c, 0)
self.assert_("<html" in o)
self.assert_('class="linenos"' in o)
def test_F_opt(self):
filename = os.path.join(testdir, testfile)
c, o, e = run_cmdline("-Fhighlight:tokentype=Name.Blubb,names=testfile testdir",
"-fhtml", filename)
self.assertEquals(c, 0)
self.assert_('<span class="n-Blubb' in o)
def test_invalid_opts(self):
for opts in [("-L", "-lpy"), ("-L", "-fhtml"), ("-L", "-Ox"),
("-a",), ("-Sst", "-lpy")]:
self.assert_(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 = os.path.join(testdir, testfile)
code = file(filename).read()
output = highlight(code, PythonLexer(), HtmlFormatter())
c, o, e = run_cmdline("-lpython", "-fhtml", filename)
self.assertEquals(o, output)
self.assertEquals(e, "")
self.assertEquals(c, 0)
if __name__ == '__main__':
unittest.main()
|
