libervia-backend: tests/unit/test_plugin_xep

annotate tests/unit/test_plugin_xep_0033.py @ 4351:6a0a081485b8

plugin autocrypt: Autocrypt protocol implementation: Implementation of autocrypt: `autocrypt` header is checked, and if present and no public key is known for the peer, the key is imported. `autocrypt` header is also added to outgoing message (only if an email gateway is detected). For the moment, the JID is use as identifier, but the real email used by gateway should be used in the future. rel 456

author	Goffi <goffi@goffi.org>
date	Fri, 28 Feb 2025 09:23:35 +0100
parents	6a0155f410bd
children

rev	line source
4307 6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	1 #!/usr/bin/env python3
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	2
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	3 # Libervia: an XMPP client
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	4 # Copyright (C) 2009-2024 Jérôme Poisson (goffi@goffi.org)
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	5
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	6 # This program is free software: you can redistribute it and/or modify
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	7 # it under the terms of the GNU Affero General Public License as published by
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	8 # the Free Software Foundation, either version 3 of the License, or
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	9 # (at your option) any later version.
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	10
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	11 # This program is distributed in the hope that it will be useful,
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	14 # GNU Affero General Public License for more details.
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	15
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	16 # You should have received a copy of the GNU Affero General Public License
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	18
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	19 from typing import cast
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	20 from unittest.mock import AsyncMock, MagicMock, patch
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	21
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	22 from pytest_twisted import ensureDeferred as ed
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	23 from twisted.words.protocols.jabber import jid
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	24 from twisted.words.xish import domish
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	25
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	26 from libervia.backend.models.core import MessageData
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	27 from libervia.backend.plugins.plugin_xep_0033 import (
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	28 AddressType,
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	29 AddressesData,
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	30 NS_ADDRESS,
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	31 XEP_0033,
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	32 )
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	33
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	34
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	35 class TestXEP0033:
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	36 def test_address_type_from_element(self):
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	37 """
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	38 AddressType instance is correctly created from an XML <address> element.
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	39 """
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	40 address_elt = domish.Element((NS_ADDRESS, "address"))
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	41 address_elt["jid"] = "test@example.com"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	42 address_elt["desc"] = "Test Description"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	43 address_elt["delivered"] = "true"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	44 address = AddressType.from_element(address_elt)
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	45 assert address.jid == jid.JID("test@example.com")
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	46 assert address.desc == "Test Description"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	47 assert address.delivered is True
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	48
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	49 def test_address_type_to_element(self):
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	50 """
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	51 XML <address> element is correctly built from an AddressType instance.
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	52 """
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	53 address = AddressType(
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	54 jid=jid.JID("test@example.com"),
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	55 desc="Test Description",
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	56 delivered=True,
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	57 )
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	58 address_elt = address.to_element()
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	59 assert address_elt.uri == NS_ADDRESS
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	60 assert address_elt.name == "address"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	61 assert address_elt["jid"] == "test@example.com"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	62 assert address_elt["desc"] == "Test Description"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	63 assert address_elt["delivered"] == "true"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	64
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	65 def test_addresses_data_from_element(self):
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	66 """
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	67 AddressesData instance is correctly created from an XML <addresses> element.
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	68 """
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	69 addresses_elt = domish.Element((NS_ADDRESS, "addresses"))
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	70 address_elt1 = addresses_elt.addElement("address")
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	71 address_elt1["type"] = "to"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	72 address_elt1["jid"] = "test1@example.com"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	73 address_elt2 = addresses_elt.addElement("address")
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	74 address_elt2["type"] = "cc"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	75 address_elt2["jid"] = "test2@example.com"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	76 address_elt3 = addresses_elt.addElement("address")
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	77 address_elt3["type"] = "bcc"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	78 address_elt3["jid"] = "test3@example.com"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	79 address_elt4 = addresses_elt.addElement("address")
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	80 address_elt4["type"] = "noreply"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	81 addresses = AddressesData.from_element(addresses_elt)
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	82 assert addresses.to is not None and len(addresses.to) == 1
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	83 assert addresses.to[0].jid == jid.JID("test1@example.com")
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	84 assert addresses.cc is not None and len(addresses.cc) == 1
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	85 assert addresses.cc[0].jid == jid.JID("test2@example.com")
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	86 assert addresses.bcc is not None and len(addresses.bcc) == 1
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	87 assert addresses.bcc[0].jid == jid.JID("test3@example.com")
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	88 assert addresses.noreply
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	89
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	90 def test_addresses_data_to_element(self):
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	91 """
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	92 XML <addresses> element is correctly built from an AddressesData instance.
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	93 """
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	94 addresses = AddressesData(
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	95 to=[AddressType(jid=jid.JID("test1@example.com"))],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	96 cc=[AddressType(jid=jid.JID("test2@example.com"))],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	97 bcc=[AddressType(jid=jid.JID("test3@example.com"))],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	98 noreply=True,
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	99 )
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	100 addresses_elt = addresses.to_element()
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	101 assert addresses_elt.uri == NS_ADDRESS
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	102 assert addresses_elt.name == "addresses"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	103 assert len(addresses_elt.children) == 4
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	104 for elt in addresses_elt.children:
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	105 assert elt.uri == NS_ADDRESS
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	106 assert elt.name == "address"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	107 assert addresses_elt.children[0]["type"] == "to"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	108 assert addresses_elt.children[0]["jid"] == "test1@example.com"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	109 assert addresses_elt.children[1]["type"] == "cc"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	110 assert addresses_elt.children[1]["jid"] == "test2@example.com"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	111 assert addresses_elt.children[2]["type"] == "bcc"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	112 assert addresses_elt.children[2]["jid"] == "test3@example.com"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	113 assert addresses_elt.children[3]["type"] == "noreply"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	114
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	115 @ed
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	116 async def test_handle_addresses(self):
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	117 """
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	118 Server JID is used, <addresses> element is added and messages are delivered.
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	119 """
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	120 xep_0033 = XEP_0033(MagicMock())
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	121 client = MagicMock()
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	122 client.server_jid = jid.JID("server.example.com")
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	123 client.profile = "test_profile"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	124
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	125 mess_data = MessageData(
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	126 {
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	127 "to": jid.JID("recipient@example.com"),
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	128 "extra": {
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	129 "addresses": {
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	130 "to": [{"jid": "to@example.com"}],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	131 "cc": [{"jid": "cc@example.com"}],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	132 }
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	133 },
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	134 "xml": domish.Element(("jabber:client", "message")),
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	135 }
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	136 )
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	137
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	138 xep_0033.host.memory.disco.has_feature = AsyncMock(return_value=True)
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	139 xep_0033.deliver_messages = AsyncMock()
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	140
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	141 with patch.object(xep_0033, "_stop_if_all_delivered", AsyncMock()):
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	142 result = await xep_0033._handle_addresses(client, mess_data)
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	143 cast(AsyncMock, xep_0033._stop_if_all_delivered).assert_called_once()
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	144
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	145 assert result["to"] == client.server_jid
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	146 assert "addresses" in result["xml"].children[0].name
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	147 xep_0033.deliver_messages.assert_called_once()
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	148
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	149 @ed
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	150 async def test_deliver_messages(self):
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	151 """Delivery is done for all recipients."""
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	152 xep_0033 = XEP_0033(MagicMock())
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	153 client = MagicMock()
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	154 client.server_jid = jid.JID("server.example.com")
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	155 client.a_send = AsyncMock()
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	156
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	157 mess_data = MessageData(
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	158 {
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	159 "xml": domish.Element(("jabber:client", "message")),
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	160 }
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	161 )
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	162 to_example_addr = AddressType(jid=jid.JID("to@example.com"))
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	163 cc_example_addr = AddressType(jid=jid.JID("cc@example.com"))
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	164 bcc_example_addr = AddressType(jid=jid.JID("bcc@other.com"))
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	165
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	166 addr_data = AddressesData(
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	167 to=[to_example_addr],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	168 cc=[cc_example_addr],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	169 bcc=[bcc_example_addr],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	170 )
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	171
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	172 domains = {
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	173 "example.com": [
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	174 to_example_addr,
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	175 cc_example_addr,
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	176 ],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	177 "other.com": [bcc_example_addr],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	178 }
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	179
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	180 xep_0033.host.memory.disco.has_feature = AsyncMock(return_value=True)
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	181
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	182 await xep_0033.deliver_messages(client, mess_data, addr_data, domains)
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	183
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	184 # Check that messages were sent to the multicast domain and individual recipients
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	185 assert client.a_send.call_count == 2
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	186 calls = client.a_send.call_args_list
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	187 # First call is to the multicast service.
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	188 assert calls[0][0][0]["to"] == "example.com"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	189 # Second call is the individual BCC.
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	190 assert calls[1][0][0]["to"] == "bcc@other.com"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	191
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	192 # Everything must have been delivered.
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	193 assert all(address.delivered for address in addr_data.addresses)
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	194
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	195 # And BCC must have been removed.
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	196 assert addr_data.bcc is None
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	197
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	198 @ed
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	199 async def test_deliver_messages_multicast_only(self):
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	200 """Delivery is done only to multicast services."""
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	201 xep_0033 = XEP_0033(MagicMock())
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	202 client = MagicMock()
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	203 client.server_jid = jid.JID("server.example.com")
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	204 client.a_send = AsyncMock()
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	205
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	206 mess_data = MessageData(
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	207 {
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	208 "xml": domish.Element(("jabber:client", "message")),
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	209 }
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	210 )
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	211 to_example_addr = AddressType(jid=jid.JID("to@example.com"))
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	212 cc_example_addr = AddressType(jid=jid.JID("cc@example.com"))
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	213 bcc_example_addr = AddressType(jid=jid.JID("bcc@other.com"))
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	214
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	215 addr_data = AddressesData(
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	216 to=[to_example_addr],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	217 cc=[cc_example_addr],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	218 bcc=[bcc_example_addr],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	219 )
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	220
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	221 domains = {
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	222 "example.com": [
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	223 to_example_addr,
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	224 cc_example_addr,
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	225 ],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	226 "other.com": [bcc_example_addr],
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	227 }
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	228
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	229 xep_0033.host.memory.disco.has_feature = AsyncMock(return_value=True)
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	230
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	231 await xep_0033.deliver_messages(
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	232 client, mess_data, addr_data, domains, multicast_only=True
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	233 )
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	234
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	235 # Check that only the multicast message was sent
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	236 assert client.a_send.call_count == 1
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	237 assert client.a_send.call_args[0][0]["to"] == "example.com"
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	238
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	239 # Check that only addresses from the multicast domain are marked as delivered
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	240 assert addr_data.to and addr_data.to[0].delivered is True
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	241 assert addr_data.cc and addr_data.cc[0].delivered is True
6a0155f410bd test (unit): add test for plugin XEP-0033: Goffi <goffi@goffi.org> parents: diff changeset	242 assert addr_data.bcc and addr_data.bcc[0].delivered is None

Mercurial > libervia-backend

annotate tests/unit/test_plugin_xep_0033.py @ 4351:6a0a081485b8