100 lines
3.6 KiB
Python
100 lines
3.6 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# Libervia: an XMPP client
|
|
# Copyright (C) 2009-2025 Jérôme Poisson (goffi@goffi.org)
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
from twisted.words.protocols.jabber import jid
|
|
from twisted.words.xish import domish
|
|
|
|
from libervia.backend.core import exceptions
|
|
from libervia.backend.memory.sqla_mapping import History
|
|
from libervia.backend.models.core import MessageData
|
|
from libervia.backend.plugins.plugin_xep_0461 import NS_REPLY, ReplyTo, XEP_0461
|
|
from pytest_twisted import ensureDeferred as ed
|
|
|
|
|
|
class TestXEP0461:
|
|
@ed
|
|
async def test_reply_to_from_element(self):
|
|
"""<reply> element parsing works."""
|
|
# Test correct case
|
|
element = domish.Element(
|
|
(NS_REPLY, "reply"), attribs={"id": "123", "to": "user@example.com"}
|
|
)
|
|
reply = ReplyTo.from_element(element)
|
|
assert reply.to == jid.JID("user@example.com")
|
|
assert reply.id == "123"
|
|
assert not reply.internal_uid
|
|
|
|
# Test child element
|
|
parent = domish.Element((None, "some"))
|
|
child = domish.Element((NS_REPLY, "reply"), attribs={"id": "456"})
|
|
parent.addChild(child)
|
|
reply = ReplyTo.from_element(parent)
|
|
assert reply.id == "456"
|
|
assert reply.to is None
|
|
|
|
# Test missing id
|
|
element = domish.Element((NS_REPLY, "reply"))
|
|
with pytest.raises(exceptions.DataError):
|
|
ReplyTo.from_element(element)
|
|
|
|
# Test no reply
|
|
element = domish.Element((None, "some"))
|
|
with pytest.raises(exceptions.NotFound):
|
|
ReplyTo.from_element(element)
|
|
|
|
@ed
|
|
async def test_reply_to_to_element(self, host, client):
|
|
"<reply> element generation works."
|
|
reply = ReplyTo(id="123", internal_uid=True)
|
|
with pytest.raises(exceptions.DataError):
|
|
reply.to_element()
|
|
|
|
reply = ReplyTo(id="123", internal_uid=False)
|
|
elt = reply.to_element()
|
|
assert elt.uri == NS_REPLY
|
|
assert elt.name == "reply"
|
|
assert elt["id"] == "123"
|
|
assert not elt.hasAttribute("to")
|
|
|
|
reply = ReplyTo(id="123", to=jid.JID("user@example.com"), internal_uid=False)
|
|
elt = reply.to_element()
|
|
assert elt["to"] == "user@example.com"
|
|
|
|
@ed
|
|
async def test_xep_0461_message_received_trigger(self, host, client):
|
|
"""post_treat callback is added."""
|
|
xep = XEP_0461(host)
|
|
post_treat = MagicMock()
|
|
xep._message_received_trigger(client, None, post_treat)
|
|
assert post_treat.addCallback.called
|
|
|
|
@ed
|
|
async def test_xep_0461_send_message_trigger(self, host, client):
|
|
"post_xml_treatments callback is added."
|
|
xep = XEP_0461(host)
|
|
mess_data = MessageData(extra={"reply": {"id": "123", "internal_uid": False}})
|
|
pre_xml_treatments = MagicMock()
|
|
post_xml_treatments = MagicMock()
|
|
xep._sendMessage_trigger(
|
|
client, mess_data, pre_xml_treatments, post_xml_treatments
|
|
)
|
|
assert post_xml_treatments.addCallback.called
|