Mercurial > libervia-backend
annotate sat/test/test_plugin_xep_0334.py @ 2764:92af49cde255
jp (base): MAM and RSM arguments can now be used for pubsub commands:
- RSM and MAM args arguments have been added to pubsub group, and activated when multiple items are possible
- /!\ --max-items (i.e. pubsub original max) short option has been change for -M, and -m is now used for RSM max
- pubsub/search -m option has been renamed -M for same reason as above
- pubsub/search -o option has been replaced by -k (for keep), to avoid conflict with order-by (and long version is now --node-max)
author | Goffi <goffi@goffi.org> |
---|---|
date | Fri, 11 Jan 2019 10:18:02 +0100 |
parents | 56f94936df1e |
children | 378188abe941 |
rev | line source |
---|---|
1934
2daf7b4c6756
use of /usr/bin/env instead of /usr/bin/python in shebang
Goffi <goffi@goffi.org>
parents:
1766
diff
changeset
|
1 #!/usr/bin/env python2 |
1279 | 2 # -*- coding: utf-8 -*- |
3 | |
4 # SAT: a jabber client | |
2483 | 5 # Copyright (C) 2009-2018 Jérôme Poisson (goffi@goffi.org) |
1766 | 6 # Copyright (C) 2013-2016 Adrien Cossa (souliane@mailoo.org) |
1279 | 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 sat.test import helpers | |
25 from sat.plugins.plugin_xep_0334 import XEP_0334 | |
26 from twisted.internet import defer | |
27 from wokkel.generic import parseXml | |
28 from sat.core import exceptions | |
29 | |
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
30 HINTS = ("no-permanent-storage", "no-storage", "no-copy") |
1279 | 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 | |
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
38 def test_messageSendTrigger(self): |
1279 | 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 """ | |
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
48 original_xml = template_xml % "" |
1279 | 49 |
50 d_list = [] | |
51 | |
52 def cb(data, expected_xml): | |
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
53 result_xml = data["xml"].toXml().encode("utf-8") |
1279 | 54 self.assertEqualXML(result_xml, expected_xml, True) |
55 | |
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
56 for key in HINTS + ("", "dummy_hint"): |
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
57 mess_data = { |
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
58 "xml": parseXml(original_xml.encode("utf-8")), |
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
59 "extra": {key: True}, |
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
60 } |
1279 | 61 treatments = defer.Deferred() |
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
62 self.plugin.messageSendTrigger( |
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
63 self.host.getClient(C.PROFILE[0]), mess_data, defer.Deferred(), treatments |
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
64 ) |
1279 | 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_messageReceivedTrigger(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(dummy): | |
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 | |
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
92 for key in HINTS + ("dummy_hint",): |
1279 | 93 message = parseXml(template_xml % ('<%s xmlns="urn:xmpp:hints"/>' % key)) |
94 post_treat = defer.Deferred() | |
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
95 self.plugin.messageReceivedTrigger( |
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
96 self.host.getClient(C.PROFILE[0]), message, post_treat |
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
97 ) |
1279 | 98 if post_treat.callbacks: |
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
99 assert key in ("no-permanent-storage", "no-storage") |
1279 | 100 post_treat.addCallbacks(cb, eb) |
101 post_treat.callback(None) | |
102 d_list.append(post_treat) | |
103 else: | |
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
104 assert key not in ("no-permanent-storage", "no-storage") |
1279 | 105 |
106 return defer.DeferredList(d_list) |