From 270ab316ed61b17e5cefebdd2081e469f06317e8 Mon Sep 17 00:00:00 2001
From: Mathias Ertl <mati@er.tl>
Date: Fri, 25 Dec 2020 13:34:02 +0100
Subject: consider trailing whitespace a part of the prompt, making copy/paste
 more straight forward

---
 tests/test_shell.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

(limited to 'tests/test_shell.py')

diff --git a/tests/test_shell.py b/tests/test_shell.py
index b3600a4e..b43768d4 100644
--- a/tests/test_shell.py
+++ b/tests/test_shell.py
@@ -142,9 +142,8 @@ def test_end_of_line_nums(lexer_bash):
 def test_newline_in_echo(lexer_session):
     fragment = '$ echo \\\nhi\nhi\n'
     tokens = [
-        (Token.Text, ''),
-        (Token.Generic.Prompt, '$'),
-        (Token.Text, ' '),
+        (Token.Name.Builtin, ''),
+        (Token.Generic.Prompt, '$ '),
         (Token.Name.Builtin, 'echo'),
         (Token.Text, ' '),
         (Token.Literal.String.Escape, '\\\n'),
@@ -217,8 +216,7 @@ def test_virtualenv(lexer_session):
         (Token.Text, ''),
         (Token.Text, ' '),
         (Token.Text, ''),
-        (Token.Generic.Prompt, '[~/project]$'),
-        (Token.Text, ' '),
+        (Token.Generic.Prompt, '[~/project]$ '),
         (Token.Text, 'foo'),
         (Token.Text, ' '),
         (Token.Text, '-h'),
-- 
cgit v1.2.1


From 456ac59d8481c09c29ea9f771550a933ba70edb8 Mon Sep 17 00:00:00 2001
From: Mathias Ertl <mati@er.tl>
Date: Sat, 26 Dec 2020 16:17:25 +0100
Subject: add tests to illustrate problem discussed in PR #1645

---
 tests/test_shell.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

(limited to 'tests/test_shell.py')

diff --git a/tests/test_shell.py b/tests/test_shell.py
index b43768d4..3cf7e204 100644
--- a/tests/test_shell.py
+++ b/tests/test_shell.py
@@ -154,6 +154,31 @@ def test_newline_in_echo(lexer_session):
     assert list(lexer_session.get_tokens(fragment)) == tokens
 
 
+def test_newline_in_ls(lexer_session):
+    fragment = '$ ls \\\nhi\nhi\n'
+    tokens = [
+        (Token.Text, ''),
+        (Token.Generic.Prompt, '$ '),
+        (Token.Text, 'ls'),
+        (Token.Text, ' '),
+        (Token.Literal.String.Escape, '\\\n'),
+        (Token.Text, 'hi'),
+        (Token.Text, '\n'),
+        (Token.Generic.Output, 'hi\n'),
+    ]
+    assert list(lexer_session.get_tokens(fragment)) == tokens
+
+
+def test_comment_after_prompt(lexer_session):
+    fragment = '$# comment'
+    tokens = [
+        (Token.Comment.Single, ''),
+        (Token.Generic.Prompt, '$'),
+        (Token.Comment.Single, '# comment\n'),
+    ]
+    assert list(lexer_session.get_tokens(fragment)) == tokens
+
+
 def test_msdos_gt_only(lexer_msdos):
     fragment = '> py\nhi\n'
     tokens = [
-- 
cgit v1.2.1