comparison 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
comparison
equal deleted inserted replaced
4070:d10748475025 4071:4b842c1fb686
1 #!/usr/bin/env python3
2
3
4 # SAT: a jabber client
5 # Copyright (C) 2009-2021 Jérôme Poisson (goffi@goffi.org)
6 # Copyright (C) 2013-2016 Adrien Cossa (souliane@mailoo.org)
7
8 # This program is free software: you can redistribute it and/or modify
9 # it under the terms of the GNU Affero General Public License as published by
10 # the Free Software Foundation, either version 3 of the License, or
11 # (at your option) any later version.
12
13 # This program is distributed in the hope that it will be useful,
14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 # GNU Affero General Public License for more details.
17
18 # You should have received a copy of the GNU Affero General Public License
19 # along with this program. If not, see <http://www.gnu.org/licenses/>.
20
21 """ Plugin XEP-0334 """
22
23 from .constants import Const as C
24 from libervia.backend.test import helpers
25 from libervia.backend.plugins.plugin_xep_0334 import XEP_0334
26 from twisted.internet import defer
27 from wokkel.generic import parseXml
28 from libervia.backend.core import exceptions
29
30 HINTS = ("no-permanent-storage", "no-storage", "no-copy")
31
32
33 class XEP_0334Test(helpers.SatTestCase):
34 def setUp(self):
35 self.host = helpers.FakeSAT()
36 self.plugin = XEP_0334(self.host)
37
38 def test_message_send_trigger(self):
39 template_xml = """
40 <message
41 from='romeo@montague.net/orchard'
42 to='juliet@capulet.com'
43 type='chat'>
44 <body>text</body>
45 %s
46 </message>
47 """
48 original_xml = template_xml % ""
49
50 d_list = []
51
52 def cb(data, expected_xml):
53 result_xml = data["xml"].toXml().encode("utf-8")
54 self.assert_equal_xml(result_xml, expected_xml, True)
55
56 for key in HINTS + ("", "dummy_hint"):
57 mess_data = {
58 "xml": parseXml(original_xml.encode("utf-8")),
59 "extra": {key: True},
60 }
61 treatments = defer.Deferred()
62 self.plugin.messageSendTrigger(
63 self.host.get_client(C.PROFILE[0]), mess_data, defer.Deferred(), treatments
64 )
65 if treatments.callbacks: # the trigger added a callback
66 expected_xml = template_xml % ('<%s xmlns="urn:xmpp:hints"/>' % key)
67 treatments.addCallback(cb, expected_xml)
68 treatments.callback(mess_data)
69 d_list.append(treatments)
70
71 return defer.DeferredList(d_list)
72
73 def test_message_received_trigger(self):
74 template_xml = """
75 <message
76 from='romeo@montague.net/orchard'
77 to='juliet@capulet.com'
78 type='chat'>
79 <body>text</body>
80 %s
81 </message>
82 """
83
84 def cb(__):
85 raise Exception("Errback should not be ran instead of callback!")
86
87 def eb(failure):
88 failure.trap(exceptions.SkipHistory)
89
90 d_list = []
91
92 for key in HINTS + ("dummy_hint",):
93 message = parseXml(template_xml % ('<%s xmlns="urn:xmpp:hints"/>' % key))
94 post_treat = defer.Deferred()
95 self.plugin.message_received_trigger(
96 self.host.get_client(C.PROFILE[0]), message, post_treat
97 )
98 if post_treat.callbacks:
99 assert key in ("no-permanent-storage", "no-storage")
100 post_treat.addCallbacks(cb, eb)
101 post_treat.callback(None)
102 d_list.append(post_treat)
103 else:
104 assert key not in ("no-permanent-storage", "no-storage")
105
106 return defer.DeferredList(d_list)