Merged in hroncok/pygments-main/doscon_prompt (pull request #734)

Recognize single > as a prompt in doscon

1 files changed, 18 insertions, 1 deletions

diff --git a/tests/test_shell.py b/tests/test_shell.py
index e283793e..1121240a 100644
--- a/tests/test_shell.py
+++ b/tests/test_shell.py
@@ -10,7 +10,7 @@
 import unittest
 
 from pygments.token import Token
-from pygments.lexers import BashLexer, BashSessionLexer
+from pygments.lexers import BashLexer, BashSessionLexer, MSDOSSessionLexer
 
 
 class BashTest(unittest.TestCase):
@@ -140,3 +140,20 @@ class BashSessionTest(unittest.TestCase):
         ]
         self.assertEqual(tokens, list(self.lexer.get_tokens(fragment)))
 
+class MSDOSSessionTest(unittest.TestCase):
+
+    def setUp(self):
+        self.lexer = MSDOSSessionLexer()
+
+    def testGtOnlyPrompt(self):
+        fragment = u'> py\nhi\n'
+        tokens = [
+            (Token.Text, u''),
+            (Token.Generic.Prompt, u'>'),
+            (Token.Text, u' '),
+            (Token.Text, u'py'),
+            (Token.Text, u''),
+            (Token.Text, u'\n'),
+            (Token.Generic.Output, u'hi\n'),
+        ]
+        self.assertEqual(tokens, list(self.lexer.get_tokens(fragment)))