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
|
# -*- coding: utf-8 -*-
"""
Command line test
~~~~~~~~~~~~~~~~~
:copyright: 2006 by Georg Brandl.
:license: GNU LGPL, 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.assert_(c == 0)
self.assert_(o.find("Lexers") and o.find("Formatters"))
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()
|