mirror of
https://git.quad4.io/RNS-Things/MeshChatX.git
synced 2026-03-31 06:05:49 +00:00
d9cbc59ab3
- Removed redundant comments from various test files to enhance readability and focus on the test logic. - Added a few tests for prop sync and markdown security.
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
from hypothesis import given, strategies as st
|
|
import LXMF
|
|
from meshchatx.meshchat import ReticulumMeshChat
|
|
from unittest.mock import MagicMock
|
|
|
|
|
|
def get_mock_mesh_chat():
|
|
app = ReticulumMeshChat.__new__(ReticulumMeshChat)
|
|
app.current_context = MagicMock()
|
|
app.reticulum = MagicMock()
|
|
return app
|
|
|
|
|
|
@given(content=st.text())
|
|
def test_fuzz_reply_detection_no_crash(content):
|
|
mesh_chat = get_mock_mesh_chat()
|
|
mock_msg = MagicMock(spec=LXMF.LXMessage)
|
|
mock_msg.hash = b"h" * 16
|
|
mock_msg.source_hash = b"s" * 16
|
|
mock_msg.destination_hash = b"d" * 16
|
|
mock_msg.content = content.encode("utf-8", errors="replace")
|
|
mock_msg.get_fields.return_value = {}
|
|
mock_msg.timestamp = 0
|
|
mock_msg.progress = 0
|
|
mock_msg.incoming = True
|
|
mock_msg.state = 0
|
|
mock_msg.method = 0
|
|
mock_msg.delivery_attempts = 0
|
|
mock_msg.title = b""
|
|
mock_msg.rssi = 0
|
|
mock_msg.snr = 0
|
|
mock_msg.q = 0
|
|
|
|
mesh_chat.db_upsert_lxmf_message(mock_msg)
|
|
|
|
|
|
def test_explicit_reply_detection():
|
|
mesh_chat = get_mock_mesh_chat()
|
|
test_hash = "a" * 32
|
|
content = f"> {test_hash}\nThis is a reply"
|
|
|
|
mock_msg = MagicMock(spec=LXMF.LXMessage)
|
|
mock_msg.hash = b"h" * 16
|
|
mock_msg.source_hash = b"s" * 16
|
|
mock_msg.destination_hash = b"d" * 16
|
|
mock_msg.content = content.encode("utf-8")
|
|
mock_msg.get_fields.return_value = {}
|
|
mock_msg.timestamp = 0
|
|
mock_msg.progress = 0
|
|
mock_msg.incoming = True
|
|
mock_msg.state = 0
|
|
mock_msg.method = 0
|
|
mock_msg.delivery_attempts = 0
|
|
mock_msg.title = b""
|
|
mock_msg.rssi = 0
|
|
mock_msg.snr = 0
|
|
mock_msg.q = 0
|
|
|
|
# Mock database upsert to capture what was sent
|
|
mesh_chat.current_context.database.messages.upsert_lxmf_message = MagicMock()
|
|
mesh_chat.current_context.local_lxmf_destination.hexhash = "local"
|
|
|
|
mesh_chat.db_upsert_lxmf_message(mock_msg)
|
|
|
|
args, _ = mesh_chat.current_context.database.messages.upsert_lxmf_message.call_args
|
|
upserted_dict = args[0]
|
|
|
|
assert upserted_dict["reply_to_hash"] == test_hash
|