mirror of
https://github.com/pdxlocations/contact.git
synced 2026-07-27 20:12:31 +02:00
pdxlocations
7efc1a640d
- Updated `send_message` to include reply ID and context for native replies. - Modified `onAckNak` to correctly update the database with the message context. - Implemented reply prefix formatting in `build_reply_prefix` for better message clarity. - Added functionality to preserve message selection when new messages are received. - Enhanced database handling to store packet IDs and reply IDs for messages. - Updated UI state to manage reply context and message highlighting effectively. - Improved tests to cover new reply features and message selection behavior.
154 lines
6.8 KiB
Python
154 lines
6.8 KiB
Python
import unittest
|
|
from unittest import mock
|
|
|
|
import contact.ui.default_config as config
|
|
from contact.message_handlers import rx_handler
|
|
from contact.utilities.singleton import interface_state, menu_state, ui_state
|
|
|
|
from tests.test_support import reset_singletons, restore_config, snapshot_config
|
|
|
|
|
|
class RxHandlerTests(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
reset_singletons()
|
|
self.saved_config = snapshot_config("notification_sound", "message_prefix")
|
|
config.notification_sound = "False"
|
|
|
|
def tearDown(self) -> None:
|
|
restore_config(self.saved_config)
|
|
reset_singletons()
|
|
|
|
def test_on_receive_text_message_refreshes_selected_channel(self) -> None:
|
|
interface_state.myNodeNum = 111
|
|
ui_state.channel_list = ["Primary"]
|
|
ui_state.all_messages = {"Primary": []}
|
|
ui_state.selected_channel = 0
|
|
|
|
packet = {
|
|
"from": 222,
|
|
"to": 999,
|
|
"channel": 0,
|
|
"hopStart": 3,
|
|
"hopLimit": 1,
|
|
"decoded": {"portnum": "TEXT_MESSAGE_APP", "payload": b"hello"},
|
|
}
|
|
|
|
with mock.patch.object(rx_handler, "refresh_node_list", return_value=True):
|
|
with mock.patch.object(rx_handler, "request_ui_redraw") as request_ui_redraw:
|
|
with mock.patch.object(rx_handler, "add_notification") as add_notification:
|
|
with mock.patch.object(rx_handler, "save_message_to_db") as save_message_to_db:
|
|
with mock.patch.object(rx_handler, "get_name_from_database", return_value="SAT2"):
|
|
rx_handler.on_receive(packet, interface=None)
|
|
|
|
self.assertEqual(
|
|
request_ui_redraw.call_args_list,
|
|
[
|
|
mock.call(nodes=True),
|
|
mock.call(
|
|
messages=True,
|
|
scroll_messages_to_bottom=True,
|
|
preserve_message_selection=False,
|
|
),
|
|
],
|
|
)
|
|
add_notification.assert_not_called()
|
|
save_message_to_db.assert_called_once_with("Primary", 222, "hello", packet_id=None, reply_id=None)
|
|
self.assertEqual(ui_state.all_messages["Primary"][-1][1], "hello")
|
|
self.assertIn("SAT2:", ui_state.all_messages["Primary"][-1][0])
|
|
self.assertIn("[2]", ui_state.all_messages["Primary"][-1][0])
|
|
|
|
def test_on_receive_preserves_message_selection_while_showing_new_message(self) -> None:
|
|
interface_state.myNodeNum = 111
|
|
ui_state.current_window = 1
|
|
ui_state.channel_list = ["Primary"]
|
|
ui_state.all_messages = {"Primary": []}
|
|
ui_state.selected_channel = 0
|
|
packet = {
|
|
"from": 222,
|
|
"to": 999,
|
|
"channel": 0,
|
|
"hopStart": 1,
|
|
"hopLimit": 1,
|
|
"decoded": {"portnum": "TEXT_MESSAGE_APP", "payload": b"hello"},
|
|
}
|
|
|
|
with mock.patch.object(rx_handler, "refresh_node_list", return_value=False):
|
|
with mock.patch.object(rx_handler, "request_ui_redraw") as request_ui_redraw:
|
|
with mock.patch.object(rx_handler, "save_message_to_db"):
|
|
with mock.patch.object(rx_handler, "get_name_from_database", return_value="SAT2"):
|
|
rx_handler.on_receive(packet, interface=None)
|
|
|
|
request_ui_redraw.assert_called_once_with(
|
|
messages=True,
|
|
scroll_messages_to_bottom=True,
|
|
preserve_message_selection=True,
|
|
)
|
|
|
|
def test_on_receive_direct_message_adds_channel_and_notification(self) -> None:
|
|
interface_state.myNodeNum = 111
|
|
ui_state.channel_list = ["Primary"]
|
|
ui_state.all_messages = {"Primary": []}
|
|
ui_state.selected_channel = 0
|
|
|
|
packet = {
|
|
"from": 222,
|
|
"to": 111,
|
|
"hopStart": 1,
|
|
"hopLimit": 1,
|
|
"decoded": {"portnum": "TEXT_MESSAGE_APP", "payload": b"dm"},
|
|
}
|
|
|
|
with mock.patch.object(rx_handler, "refresh_node_list", return_value=False):
|
|
with mock.patch.object(rx_handler, "request_ui_redraw") as request_ui_redraw:
|
|
with mock.patch.object(rx_handler, "add_notification") as add_notification:
|
|
with mock.patch.object(rx_handler, "update_node_info_in_db") as update_node_info_in_db:
|
|
with mock.patch.object(rx_handler, "save_message_to_db") as save_message_to_db:
|
|
with mock.patch.object(rx_handler, "get_name_from_database", return_value="SAT2"):
|
|
rx_handler.on_receive(packet, interface=None)
|
|
|
|
self.assertIn(222, ui_state.channel_list)
|
|
self.assertIn(222, ui_state.all_messages)
|
|
request_ui_redraw.assert_called_once_with(channels=True)
|
|
add_notification.assert_called_once_with(1)
|
|
update_node_info_in_db.assert_called_once_with(222, chat_archived=False)
|
|
save_message_to_db.assert_called_once_with(222, 222, "dm", packet_id=None, reply_id=None)
|
|
|
|
def test_on_receive_displays_context_for_native_reply_id(self) -> None:
|
|
interface_state.myNodeNum = 111
|
|
ui_state.channel_list = ["Primary"]
|
|
ui_state.all_messages = {"Primary": [("[06:00:00] >> SAT2: ", "hello world")]}
|
|
ui_state.message_packet_ids = {"Primary": [900]}
|
|
ui_state.selected_channel = 0
|
|
packet = {
|
|
"id": 901,
|
|
"from": 222,
|
|
"to": 999,
|
|
"channel": 0,
|
|
"hopStart": 1,
|
|
"hopLimit": 1,
|
|
"decoded": {"portnum": "TEXT_MESSAGE_APP", "payload": b"hi", "replyId": 900},
|
|
}
|
|
|
|
with mock.patch.object(rx_handler, "refresh_node_list", return_value=False):
|
|
with mock.patch.object(rx_handler, "request_ui_redraw"):
|
|
with mock.patch.object(rx_handler, "save_message_to_db") as save_message_to_db:
|
|
with mock.patch.object(rx_handler, "get_name_from_database", return_value="NODE"):
|
|
rx_handler.on_receive(packet, interface=None)
|
|
|
|
self.assertEqual(ui_state.all_messages["Primary"][-1][1], "<Re: SAT2: hello> hi")
|
|
self.assertEqual(ui_state.message_packet_ids["Primary"][-1], 901)
|
|
save_message_to_db.assert_called_once_with("Primary", 222, "hi", packet_id=901, reply_id=900)
|
|
|
|
def test_on_receive_trims_packet_buffer_even_when_packet_is_undecoded(self) -> None:
|
|
ui_state.packet_buffer = list(range(25))
|
|
ui_state.display_log = True
|
|
ui_state.current_window = 4
|
|
|
|
with mock.patch.object(rx_handler, "request_ui_redraw") as request_ui_redraw:
|
|
rx_handler.on_receive({"id": "new"}, interface=None)
|
|
|
|
request_ui_redraw.assert_called_once_with(packetlog=True)
|
|
self.assertEqual(len(ui_state.packet_buffer), 20)
|
|
self.assertEqual(ui_state.packet_buffer[-1], {"id": "new"})
|
|
self.assertTrue(menu_state.need_redraw)
|