Mercurial > libervia-backend

comparison src/test/test_plugin_xep_0334.py @ 1279:d84905c3e124

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
plugin XEP-0334: first draft
author souliane <souliane@mailoo.org>
date Thu, 25 Dec 2014 12:09:05 +0100
parents
children 069ad98b360d
comparison
equal deleted inserted replaced
1278:347aee3a3f5c 1279:d84905c3e124
1 #!/usr/bin/python
2 # -*- coding: utf-8 -*-
3
4 # SAT: a jabber client
5 # Copyright (C) 2009, 2010, 2011, 2012, 2013, 2014 Jérôme Poisson (goffi@goffi.org)
6 # Copyright (C) 2013, 2014 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 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)