[svn] Add tests for filters.

2 files changed, 27 insertions, 3 deletions

diff --git a/tests/test_basic_api.py b/tests/test_basic_api.py
index a5abd2b8..f07c00e7 100644
--- a/tests/test_basic_api.py
+++ b/tests/test_basic_api.py
@@ -7,11 +7,12 @@
     :license: BSD, see LICENSE for more details.
 """
 
+import os
 import unittest
 import StringIO
 import random
 
-from pygments import lexers, formatters, format
+from pygments import lexers, formatters, filters, format
 from pygments.token import _TokenType, Text
 from pygments.lexer import RegexLexer
 
@@ -70,6 +71,16 @@ class LexersTest(unittest.TestCase):
             a(isinstance(x, lexers.PythonLexer))
             ae(x.options["opt"], "val")
 
+    def test_filters(self):
+        for x in filters.FILTERS.keys():
+            lx = lexers.PythonLexer()
+            lx.add_filter(x)
+            text = file(os.path.join(testdir, testfile)).read().decode('utf-8')
+            tokens = list(lx.get_tokens(text))
+            roundtext = ''.join([t[1] for t in tokens])
+            self.assertEquals(roundtext, text,
+                              "lexer roundtrip with %s filter failed" % x) 
+
 
 class FormattersTest(unittest.TestCase):
 
diff --git a/tests/test_cmdline.py b/tests/test_cmdline.py
index 9a068d53..e82eb247 100644
--- a/tests/test_cmdline.py
+++ b/tests/test_cmdline.py
@@ -33,8 +33,14 @@ 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"))
+        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 not "Formatters" in o)
+        c, o, e = run_cmdline("-L", "lexers")
+        self.assertEquals(c, 0)
 
     def test_O_opt(self):
         filename = os.path.join(testdir, testfile)
@@ -43,6 +49,13 @@ class CmdLineTest(unittest.TestCase):
         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")]: