Fix tests.
This commit is contained in:
+1
-16
@@ -6,7 +6,7 @@ from io import StringIO
|
|||||||
from unittest.mock import patch
|
from unittest.mock import patch
|
||||||
from chatmastermind.tags import TagLine
|
from chatmastermind.tags import TagLine
|
||||||
from chatmastermind.message import Message, Question, Answer, Tag, MessageFilter
|
from chatmastermind.message import Message, Question, Answer, Tag, MessageFilter
|
||||||
from chatmastermind.chat import Chat, ChatDB, terminal_width, ChatError
|
from chatmastermind.chat import Chat, ChatDB, ChatError
|
||||||
|
|
||||||
|
|
||||||
class TestChat(unittest.TestCase):
|
class TestChat(unittest.TestCase):
|
||||||
@@ -92,16 +92,10 @@ class TestChat(unittest.TestCase):
|
|||||||
Question 1
|
Question 1
|
||||||
{Answer.txt_header}
|
{Answer.txt_header}
|
||||||
Answer 1
|
Answer 1
|
||||||
|
|
||||||
{'-'*terminal_width()}
|
|
||||||
|
|
||||||
{Question.txt_header}
|
{Question.txt_header}
|
||||||
Question 2
|
Question 2
|
||||||
{Answer.txt_header}
|
{Answer.txt_header}
|
||||||
Answer 2
|
Answer 2
|
||||||
|
|
||||||
{'-'*terminal_width()}
|
|
||||||
|
|
||||||
"""
|
"""
|
||||||
self.assertEqual(mock_stdout.getvalue(), expected_output)
|
self.assertEqual(mock_stdout.getvalue(), expected_output)
|
||||||
|
|
||||||
@@ -115,18 +109,12 @@ FILE: 0001.txt
|
|||||||
Question 1
|
Question 1
|
||||||
{Answer.txt_header}
|
{Answer.txt_header}
|
||||||
Answer 1
|
Answer 1
|
||||||
|
|
||||||
{'-'*terminal_width()}
|
|
||||||
|
|
||||||
{TagLine.prefix} btag2
|
{TagLine.prefix} btag2
|
||||||
FILE: 0002.txt
|
FILE: 0002.txt
|
||||||
{Question.txt_header}
|
{Question.txt_header}
|
||||||
Question 2
|
Question 2
|
||||||
{Answer.txt_header}
|
{Answer.txt_header}
|
||||||
Answer 2
|
Answer 2
|
||||||
|
|
||||||
{'-'*terminal_width()}
|
|
||||||
|
|
||||||
"""
|
"""
|
||||||
self.assertEqual(mock_stdout.getvalue(), expected_output)
|
self.assertEqual(mock_stdout.getvalue(), expected_output)
|
||||||
|
|
||||||
@@ -415,9 +403,6 @@ class TestChatDB(unittest.TestCase):
|
|||||||
db_dir_files = self.message_list(self.db_path)
|
db_dir_files = self.message_list(self.db_path)
|
||||||
self.assertEqual(len(db_dir_files), 5)
|
self.assertEqual(len(db_dir_files), 5)
|
||||||
|
|
||||||
with self.assertRaises(ChatError):
|
|
||||||
chat_db.add_to_cache([Message(Question("?"), file_path=pathlib.Path("foo"))])
|
|
||||||
|
|
||||||
def test_chat_db_write_messages(self) -> None:
|
def test_chat_db_write_messages(self) -> None:
|
||||||
# create a new ChatDB instance
|
# create a new ChatDB instance
|
||||||
chat_db = ChatDB.from_dir(pathlib.Path(self.cache_path.name),
|
chat_db = ChatDB.from_dir(pathlib.Path(self.cache_path.name),
|
||||||
|
|||||||
Reference in New Issue
Block a user