Mercurial > libervia-backend
annotate src/test/test_plugin_xep_0334.py @ 1941:befdcfc55569
core (logs): use tools.config to handle sat.conf
| author | Goffi <goffi@goffi.org> |
|---|---|
| date | Mon, 18 Apr 2016 18:33:59 +0200 |
| parents | 2daf7b4c6756 |
| children | 633b5c21aefd |
| 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 | |
| 1766 | 5 # Copyright (C) 2009-2016 Jérôme Poisson (goffi@goffi.org) |
| 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 | |
| 30 HINTS = ('no-permanent-storage', 'no-storage', 'no-copy') | |
| 31 | |
| 32 | |
| 33 class XEP_0334Test(helpers.SatTestCase): | |
| 34 | |
| 35 def setUp(self): | |
| 36 self.host = helpers.FakeSAT() | |
| 37 self.plugin = XEP_0334(self.host) | |
| 38 | |
| 39 def test_sendMessageTrigger(self): | |
| 40 template_xml = """ | |
| 41 <message | |
| 42 from='romeo@montague.net/orchard' | |
| 43 to='juliet@capulet.com' | |
| 44 type='chat'> | |
| 45 <body>text</body> | |
| 46 %s | |
| 47 </message> | |
| 48 """ | |
| 49 original_xml = template_xml % '' | |
| 50 | |
| 51 d_list = [] | |
| 52 | |
| 53 def cb(data, expected_xml): | |
| 54 result_xml = data['xml'].toXml().encode("utf-8") | |
| 55 self.assertEqualXML(result_xml, expected_xml, True) | |
| 56 | |
| 57 for key in (HINTS + ('', 'dummy_hint')): | |
| 58 mess_data = {'xml': parseXml(original_xml.encode("utf-8")), | |
| 59 'extra': {key: True} | |
| 60 } | |
| 61 treatments = defer.Deferred() | |
| 62 self.plugin.sendMessageTrigger(mess_data, defer.Deferred(), treatments, C.PROFILE[0]) | |
| 63 if treatments.callbacks: # the trigger added a callback | |
| 64 expected_xml = template_xml % ('<%s xmlns="urn:xmpp:hints"/>' % key) | |
| 65 treatments.addCallback(cb, expected_xml) | |
| 66 treatments.callback(mess_data) | |
| 67 d_list.append(treatments) | |
| 68 | |
| 69 return defer.DeferredList(d_list) | |
| 70 | |
| 71 def test_messageReceivedTrigger(self): | |
| 72 template_xml = """ | |
| 73 <message | |
| 74 from='romeo@montague.net/orchard' | |
| 75 to='juliet@capulet.com' | |
| 76 type='chat'> | |
| 77 <body>text</body> | |
| 78 %s | |
| 79 </message> | |
| 80 """ | |
| 81 | |
| 82 def cb(dummy): | |
| 83 raise Exception("Errback should not be ran instead of callback!") | |
| 84 | |
| 85 def eb(failure): | |
| 86 failure.trap(exceptions.SkipHistory) | |
| 87 | |
| 88 d_list = [] | |
| 89 | |
| 90 for key in (HINTS + ('dummy_hint',)): | |
| 91 message = parseXml(template_xml % ('<%s xmlns="urn:xmpp:hints"/>' % key)) | |
| 92 post_treat = defer.Deferred() | |
| 93 self.plugin.messageReceivedTrigger(message, post_treat, C.PROFILE[0]) | |
| 94 if post_treat.callbacks: | |
| 95 assert(key in ('no-permanent-storage', 'no-storage')) | |
| 96 post_treat.addCallbacks(cb, eb) | |
| 97 post_treat.callback(None) | |
| 98 d_list.append(post_treat) | |
| 99 else: | |
| 100 assert(key not in ('no-permanent-storage', 'no-storage')) | |
| 101 | |
| 102 return defer.DeferredList(d_list) |