message: added function 'rm_file()' and test
This commit is contained in:
parent
589b92c9b6
commit
5e392e782e
@ -517,6 +517,13 @@ class Message():
|
|||||||
yaml.dump(data, temp_fd, sort_keys=False)
|
yaml.dump(data, temp_fd, sort_keys=False)
|
||||||
shutil.move(temp_file_path, file_path)
|
shutil.move(temp_file_path, file_path)
|
||||||
|
|
||||||
|
def rm_file(self) -> None:
|
||||||
|
"""
|
||||||
|
Delete the message file. Ignore empty file_path and not existing files.
|
||||||
|
"""
|
||||||
|
if self.file_path is not None:
|
||||||
|
self.file_path.unlink(missing_ok=True)
|
||||||
|
|
||||||
def filter_tags(self, prefix: Optional[str] = None, contain: Optional[str] = None) -> set[Tag]:
|
def filter_tags(self, prefix: Optional[str] = None, contain: Optional[str] = None) -> set[Tag]:
|
||||||
"""
|
"""
|
||||||
Filter tags based on their prefix (i. e. the tag starts with a given string)
|
Filter tags based on their prefix (i. e. the tag starts with a given string)
|
||||||
|
|||||||
@ -874,3 +874,22 @@ This is a question.
|
|||||||
{Answer.txt_header}
|
{Answer.txt_header}
|
||||||
This is an answer."""
|
This is an answer."""
|
||||||
self.assertEqual(self.message.to_str(with_tags=True, with_file=True), expected_output)
|
self.assertEqual(self.message.to_str(with_tags=True, with_file=True), expected_output)
|
||||||
|
|
||||||
|
|
||||||
|
class MessageRmFileTestCase(unittest.TestCase):
|
||||||
|
def setUp(self) -> None:
|
||||||
|
self.file = tempfile.NamedTemporaryFile(delete=False, suffix=msg_suffix)
|
||||||
|
self.file_path = pathlib.Path(self.file.name)
|
||||||
|
self.message = Message(Question('This is a question.'),
|
||||||
|
file_path=self.file_path)
|
||||||
|
self.message.to_file()
|
||||||
|
|
||||||
|
def tearDown(self) -> None:
|
||||||
|
self.file.close()
|
||||||
|
self.file_path.unlink(missing_ok=True)
|
||||||
|
|
||||||
|
def test_rm_file(self) -> None:
|
||||||
|
assert self.message.file_path
|
||||||
|
self.assertTrue(self.message.file_path.exists())
|
||||||
|
self.message.rm_file()
|
||||||
|
self.assertFalse(self.message.file_path.exists())
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user