message / chat: output improvements

2023-09-04 08:49:43 +02:00
parent 6e2d5009c1
commit 63040b3688
4 changed files with 64 additions and 16 deletions
@@ -66,16 +66,20 @@ class TestChat(CmmTestCase):
    def test_print(self, mock_stdout: StringIO) -> None:
        self.chat.add_messages([self.message1, self.message2])
        self.chat.print(paged=False)
-        expected_output = f"""{'-'*terminal_width()}
-{Question.txt_header}
+        expected_output = f"""{Question.txt_header}
 Question 1
 {Answer.txt_header}
 Answer 1
+
 {'-'*terminal_width()}
+
 {Question.txt_header}
 Question 2
 {Answer.txt_header}
 Answer 2
+
+{'-'*terminal_width()}
+
 """
        self.assertEqual(mock_stdout.getvalue(), expected_output)

@@ -83,20 +87,24 @@ Answer 2
    def test_print_with_tags_and_file(self, mock_stdout: StringIO) -> None:
        self.chat.add_messages([self.message1, self.message2])
        self.chat.print(paged=False, with_tags=True, with_files=True)
-        expected_output = f"""{'-'*terminal_width()}
-{TagLine.prefix} atag1 btag2
+        expected_output = f"""{TagLine.prefix} atag1 btag2
 FILE: 0001.txt
 {Question.txt_header}
 Question 1
 {Answer.txt_header}
 Answer 1
+
 {'-'*terminal_width()}
+
 {TagLine.prefix} btag2
 FILE: 0002.txt
 {Question.txt_header}
 Question 2
 {Answer.txt_header}
 Answer 2
+
+{'-'*terminal_width()}
+
 """
        self.assertEqual(mock_stdout.getvalue(), expected_output)