Mercurial > libervia-backend
view sat/test/test_plugin_xep_0334.py @ 2743:da59ff099b32
core (memory/encryption), plugin OTR: finished OTR integration in encryption:
- messageEncryptionStart and messageEncryptionStop are now async
- an encryption plugin can now have a startEncryption and endEncryption method, which are called automatically when suitable
- trust is now put in "trusted" key of message_data, and markAsTrusted has been added, it is used in OTR
- getTrustUI implemented for OTR, using legacy authenticate code
- catch potr.crypt.InvalidParameterError on decryption
- fixed some service OTR messages which were not correctly marked as not for storing, and thus were added to MAM archive/carbon copied
- other bug fixes
author | Goffi <goffi@goffi.org> |
---|---|
date | Thu, 03 Jan 2019 21:00:00 +0100 |
parents | 56f94936df1e |
children | 378188abe941 |
line wrap: on
line source
#!/usr/bin/env python2 # -*- coding: utf-8 -*- # SAT: a jabber client # Copyright (C) 2009-2018 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 sat.test import helpers from sat.plugins.plugin_xep_0334 import XEP_0334 from twisted.internet import defer from wokkel.generic import parseXml from sat.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_messageSendTrigger(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.assertEqualXML(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.getClient(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_messageReceivedTrigger(self): template_xml = """ <message from='romeo@montague.net/orchard' to='juliet@capulet.com' type='chat'> <body>text</body> %s </message> """ def cb(dummy): 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.messageReceivedTrigger( self.host.getClient(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)