121 lines
4.4 KiB
Python
121 lines
4.4 KiB
Python
from unittest import mock
|
|
import pathlib
|
|
from typing import cast
|
|
from .test_main import CmmTestCase
|
|
from chatmastermind.message import source_code, Message, MessageError, Question, Answer
|
|
from chatmastermind.tags import Tag
|
|
|
|
|
|
class SourceCodeTestCase(CmmTestCase):
|
|
def test_source_code_with_include_delims(self) -> None:
|
|
text = """
|
|
Some text before the code block
|
|
```python
|
|
print("Hello, World!")
|
|
```
|
|
Some text after the code block
|
|
```python
|
|
x = 10
|
|
y = 20
|
|
print(x + y)
|
|
```
|
|
"""
|
|
expected_result = [
|
|
" ```python\n print(\"Hello, World!\")\n ```\n",
|
|
" ```python\n x = 10\n y = 20\n print(x + y)\n ```\n"
|
|
]
|
|
result = source_code(text, include_delims=True)
|
|
self.assertEqual(result, expected_result)
|
|
|
|
def test_source_code_without_include_delims(self) -> None:
|
|
text = """
|
|
Some text before the code block
|
|
```python
|
|
print("Hello, World!")
|
|
```
|
|
Some text after the code block
|
|
```python
|
|
x = 10
|
|
y = 20
|
|
print(x + y)
|
|
```
|
|
"""
|
|
expected_result = [
|
|
" print(\"Hello, World!\")\n",
|
|
" x = 10\n y = 20\n print(x + y)\n"
|
|
]
|
|
result = source_code(text, include_delims=False)
|
|
self.assertEqual(result, expected_result)
|
|
|
|
def test_source_code_with_single_code_block(self) -> None:
|
|
text = "```python\nprint(\"Hello, World!\")\n```"
|
|
expected_result = ["```python\nprint(\"Hello, World!\")\n```\n"]
|
|
result = source_code(text, include_delims=True)
|
|
self.assertEqual(result, expected_result)
|
|
|
|
def test_source_code_with_no_code_blocks(self) -> None:
|
|
text = "Some text without any code blocks"
|
|
expected_result: list[str] = []
|
|
result = source_code(text, include_delims=True)
|
|
self.assertEqual(result, expected_result)
|
|
|
|
|
|
class QuestionTestCase(CmmTestCase):
|
|
def test_question_with_prefix(self) -> None:
|
|
with self.assertRaises(MessageError):
|
|
Question("=== QUESTION === What is your name?")
|
|
|
|
def test_question_without_prefix(self) -> None:
|
|
question = Question("What is your favorite color?")
|
|
self.assertIsInstance(question, Question)
|
|
self.assertEqual(question, "What is your favorite color?")
|
|
|
|
|
|
class AnswerTestCase(CmmTestCase):
|
|
def test_answer_with_prefix(self) -> None:
|
|
with self.assertRaises(MessageError):
|
|
Answer("=== ANSWER === Yes")
|
|
|
|
def test_answer_without_prefix(self) -> None:
|
|
answer = Answer("No")
|
|
self.assertIsInstance(answer, Answer)
|
|
self.assertEqual(answer, "No")
|
|
|
|
|
|
class TestMessage(CmmTestCase):
|
|
|
|
def test_from_file_txt(self) -> None:
|
|
file_path = pathlib.Path('message.txt')
|
|
with mock.patch('chatmastermind.message.pathlib.Path.exists', return_value=True), \
|
|
mock.patch('builtins.open',
|
|
mock.mock_open(
|
|
read_data="TAGS: tag1 tag2\n=== QUESTION ===\nThis is a question\n=== ANSWER ===\nThis is an answer")):
|
|
message = Message.from_file(file_path)
|
|
|
|
self.assertIsInstance(message, Message)
|
|
self.assertEqual(message.question, 'This is a question')
|
|
self.assertEqual(message.answer, 'This is an answer')
|
|
self.assertSetEqual(cast(set[Tag], message.tags), {Tag('tag1'), Tag('tag2')})
|
|
self.assertEqual(message.file_path, file_path)
|
|
|
|
def test_from_file_yaml(self) -> None:
|
|
# TODO: Create a test yaml file
|
|
pass
|
|
|
|
def test_to_file_txt(self) -> None:
|
|
file_path = pathlib.Path('message.txt')
|
|
message = Message(Question('This is a question'),
|
|
Answer('This is an answer'),
|
|
{Tag('tag1'), Tag('tag2')},
|
|
file_path)
|
|
expected_content = "TAGS: tag1 tag2\n=== QUESTION ===\nThis is a question\n=== ANSWER ===\nThis is an answer\n"
|
|
# fake write
|
|
with mock.patch("builtins.open", mock.mock_open) as mock_fd:
|
|
message.to_file(None)
|
|
mock_fd().assert_called_once_with(file_path, 'w')
|
|
mock_fd().write.assert_called_once_with(expected_content)
|
|
|
|
def test_to_file_yaml(self) -> None:
|
|
# TODO: Create a test yaml file
|
|
pass
|