Mercurial > libervia-backend
annotate sat/test/test_plugin_xep_0334.py @ 3137:559a625a236b
fixed shebangs
| author | Goffi <goffi@goffi.org> |
|---|---|
| date | Wed, 29 Jan 2020 10:07:00 +0100 |
| parents | 9d0df638c8b4 |
| children | be6d91572633 |
| rev | line source |
|---|---|
| 3028 | 1 #!/usr/bin/env python3 |
| 3137 | 2 |
| 1279 | 3 |
| 4 # SAT: a jabber client | |
| 3136 | 5 # Copyright (C) 2009-2020 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 | |
| 3028 | 23 from .constants import Const as C |
| 1279 | 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 | |
|
2765
378188abe941
misc: replaced all "dummy" by the more conventional and readable "__" ("_" being used for gettext)
Goffi <goffi@goffi.org>
parents:
2624
diff
changeset
|
84 def cb(__): |
| 1279 | 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) |