37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
import unittest
|
|
from unittest.mock import MagicMock
|
|
|
|
from meshchatx.src.backend.message_handler import MessageHandler
|
|
|
|
|
|
class TestMessageHandler(unittest.TestCase):
|
|
def setUp(self):
|
|
self.db = MagicMock()
|
|
self.handler = MessageHandler(self.db)
|
|
|
|
def test_get_conversation_messages(self):
|
|
self.db.provider.fetchall.return_value = [{"id": 1, "content": "test"}]
|
|
|
|
messages = self.handler.get_conversation_messages("local", "dest", limit=50)
|
|
|
|
self.assertEqual(len(messages), 1)
|
|
self.db.provider.fetchall.assert_called()
|
|
args, kwargs = self.db.provider.fetchall.call_args
|
|
self.assertIn("peer_hash = ?", args[0])
|
|
self.assertIn("dest", args[1])
|
|
|
|
def test_delete_conversation(self):
|
|
self.handler.delete_conversation("local", "dest")
|
|
self.assertEqual(self.db.provider.execute.call_count, 2)
|
|
call_args_list = self.db.provider.execute.call_args_list
|
|
first_call_args, _ = call_args_list[0]
|
|
second_call_args, _ = call_args_list[1]
|
|
self.assertIn("DELETE FROM lxmf_messages", first_call_args[0])
|
|
self.assertIn("dest", first_call_args[1])
|
|
self.assertIn("DELETE FROM lxmf_conversation_folders", second_call_args[0])
|
|
self.assertIn("dest", second_call_args[1])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|