Mercurial > libervia-backend
diff libervia/backend/test/test_plugin_xep_0334.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_0334.py@524856bd7b19 |
| children | 0d7bb4df2343 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/libervia/backend/test/test_plugin_xep_0334.py Fri Jun 02 11:49:51 2023 +0200 @@ -0,0 +1,106 @@ +#!/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 XEP-0334 """ + +from .constants import Const as C +from libervia.backend.test import helpers +from libervia.backend.plugins.plugin_xep_0334 import XEP_0334 +from twisted.internet import defer +from wokkel.generic import parseXml +from libervia.backend.core import exceptions + +HINTS = ("no-permanent-storage", "no-storage", "no-copy") + + +class XEP_0334Test(helpers.SatTestCase): + def setUp(self): + self.host = helpers.FakeSAT() + self.plugin = XEP_0334(self.host) + + def test_message_send_trigger(self): + template_xml = """ + <message + from='romeo@montague.net/orchard' + to='juliet@capulet.com' + type='chat'> + <body>text</body> + %s + </message> + """ + original_xml = template_xml % "" + + d_list = [] + + def cb(data, expected_xml): + result_xml = data["xml"].toXml().encode("utf-8") + self.assert_equal_xml(result_xml, expected_xml, True) + + for key in HINTS + ("", "dummy_hint"): + mess_data = { + "xml": parseXml(original_xml.encode("utf-8")), + "extra": {key: True}, + } + treatments = defer.Deferred() + self.plugin.messageSendTrigger( + self.host.get_client(C.PROFILE[0]), mess_data, defer.Deferred(), treatments + ) + if treatments.callbacks: # the trigger added a callback + expected_xml = template_xml % ('<%s xmlns="urn:xmpp:hints"/>' % key) + treatments.addCallback(cb, expected_xml) + treatments.callback(mess_data) + d_list.append(treatments) + + return defer.DeferredList(d_list) + + def test_message_received_trigger(self): + template_xml = """ + <message + from='romeo@montague.net/orchard' + to='juliet@capulet.com' + type='chat'> + <body>text</body> + %s + </message> + """ + + def cb(__): + raise Exception("Errback should not be ran instead of callback!") + + def eb(failure): + failure.trap(exceptions.SkipHistory) + + d_list = [] + + for key in HINTS + ("dummy_hint",): + message = parseXml(template_xml % ('<%s xmlns="urn:xmpp:hints"/>' % key)) + post_treat = defer.Deferred() + self.plugin.message_received_trigger( + self.host.get_client(C.PROFILE[0]), message, post_treat + ) + if post_treat.callbacks: + assert key in ("no-permanent-storage", "no-storage") + post_treat.addCallbacks(cb, eb) + post_treat.callback(None) + d_list.append(post_treat) + else: + assert key not in ("no-permanent-storage", "no-storage") + + return defer.DeferredList(d_list)