Mercurial > libervia-backend
diff libervia/backend/test/test_plugin_xep_0085.py @ 4071:4b842c1fb686
refactoring: renamed `sat` package to `libervia.backend`
| author | Goffi <goffi@goffi.org> |
|---|---|
| date | Fri, 02 Jun 2023 11:49:51 +0200 |
| parents | sat/test/test_plugin_xep_0085.py@524856bd7b19 |
| children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/libervia/backend/test/test_plugin_xep_0085.py Fri Jun 02 11:49:51 2023 +0200 @@ -0,0 +1,104 @@ +#!/usr/bin/env python3 + + +# SAT: a jabber client +# Copyright (C) 2009-2021 Jérôme Poisson (goffi@goffi.org) +# Copyright (C) 2013-2016 Adrien Cossa (souliane@mailoo.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/>. + +""" Plugin chat states notification tests """ + +from .constants import Const +from libervia.backend.test import helpers +from libervia.backend.core.constants import Const as C +from libervia.backend.plugins import plugin_xep_0085 as plugin +from copy import deepcopy +from twisted.internet import defer +from wokkel.generic import parseXml + + +class XEP_0085Test(helpers.SatTestCase): + def setUp(self): + self.host = helpers.FakeSAT() + self.plugin = plugin.XEP_0085(self.host) + self.host.memory.param_set( + plugin.PARAM_NAME, + True, + plugin.PARAM_KEY, + C.NO_SECURITY_LIMIT, + Const.PROFILE[0], + ) + + def test_message_received(self): + for state in plugin.CHAT_STATES: + xml = """ + <message type="chat" from="%s" to="%s" id="test_1"> + %s + <%s xmlns='%s'/> + </message> + """ % ( + Const.JID_STR[1], + Const.JID_STR[0], + "<body>test</body>" if state == "active" else "", + state, + plugin.NS_CHAT_STATES, + ) + stanza = parseXml(xml.encode("utf-8")) + self.host.bridge.expect_call( + "chat_state_received", Const.JID_STR[1], state, Const.PROFILE[0] + ) + self.plugin.message_received_trigger( + self.host.get_client(Const.PROFILE[0]), stanza, None + ) + + def test_message_send_trigger(self): + def cb(data): + xml = data["xml"].toXml().encode("utf-8") + self.assert_equal_xml(xml, expected.toXml().encode("utf-8")) + + d_list = [] + + for state in plugin.CHAT_STATES: + mess_data = { + "to": Const.JID[0], + "type": "chat", + "message": "content", + "extra": {} if state == "active" else {"chat_state": state}, + } + stanza = """ + <message type="chat" from="%s" to="%s" id="test_1"> + %s + </message> + """ % ( + Const.JID_STR[1], + Const.JID_STR[0], + ("<body>%s</body>" % mess_data["message"]) if state == "active" else "", + ) + mess_data["xml"] = parseXml(stanza.encode("utf-8")) + expected = deepcopy(mess_data["xml"]) + expected.addElement(state, plugin.NS_CHAT_STATES) + post_treatments = defer.Deferred() + self.plugin.messageSendTrigger( + self.host.get_client(Const.PROFILE[0]), mess_data, None, post_treatments + ) + + post_treatments.addCallback(cb) + post_treatments.callback(mess_data) + d_list.append(post_treatments) + + def cb_list(__): # cancel the timer to not block the process + self.plugin.map[Const.PROFILE[0]][Const.JID[0]].timer.cancel() + + return defer.DeferredList(d_list).addCallback(cb_list)