plugin XEP-0334: first draft

src/test/test_plugin_xep_0334.py

@@ -0,0 +1,102 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+# SAT: a jabber client
+# Copyright (C) 2009, 2010, 2011, 2012, 2013, 2014 Jérôme Poisson (goffi@goffi.org)
+# Copyright (C) 2013, 2014 Adrien Cossa (souliane@mailoo.org)
+
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+""" Plugin XEP-0334 """
+
+from constants import Const as C
+from sat.test import helpers
+from sat.plugins.plugin_xep_0334 import XEP_0334
+from twisted.internet import defer
+from wokkel.generic import parseXml
+from sat.core import exceptions
+
+HINTS = ('no-permanent-storage', 'no-storage', 'no-copy')
+
+
+class XEP_0334Test(helpers.SatTestCase):
+
+    def setUp(self):
+        self.host = helpers.FakeSAT()
+        self.plugin = XEP_0334(self.host)
+
+    def test_sendMessageTrigger(self):
+        template_xml = """
+        <message
+            from='romeo@montague.net/orchard'
+            to='juliet@capulet.com'
+            type='chat'>
+          <body>text</body>
+          %s
+        </message>
+        """
+        original_xml = template_xml % ''
+
+        d_list = []
+
+        def cb(data, expected_xml):
+            result_xml = data['xml'].toXml().encode("utf-8")
+            self.assertEqualXML(result_xml, expected_xml, True)
+
+        for key in (HINTS + ('', 'dummy_hint')):
+            mess_data = {'xml': parseXml(original_xml.encode("utf-8")),
+                         'extra': {key: True}
+                         }
+            treatments = defer.Deferred()
+            self.plugin.sendMessageTrigger(mess_data, defer.Deferred(), treatments, C.PROFILE[0])
+            if treatments.callbacks:  # the trigger added a callback
+                expected_xml = template_xml % ('<%s xmlns="urn:xmpp:hints"/>' % key)
+                treatments.addCallback(cb, expected_xml)
+                treatments.callback(mess_data)
+                d_list.append(treatments)
+
+        return defer.DeferredList(d_list)
+
+    def test_messageReceivedTrigger(self):
+        template_xml = """
+        <message
+            from='romeo@montague.net/orchard'
+            to='juliet@capulet.com'
+            type='chat'>
+          <body>text</body>
+          %s
+        </message>
+        """
+
+        def cb(dummy):
+            raise Exception("Errback should not be ran instead of callback!")
+
+        def eb(failure):
+            failure.trap(exceptions.SkipHistory)
+
+        d_list = []
+
+        for key in (HINTS + ('dummy_hint',)):
+            message = parseXml(template_xml % ('<%s xmlns="urn:xmpp:hints"/>' % key))
+            post_treat = defer.Deferred()
+            self.plugin.messageReceivedTrigger(message, post_treat, C.PROFILE[0])
+            if post_treat.callbacks:
+                assert(key in ('no-permanent-storage', 'no-storage'))
+                post_treat.addCallbacks(cb, eb)
+                post_treat.callback(None)
+                d_list.append(post_treat)
+            else:
+                assert(key not in ('no-permanent-storage', 'no-storage'))
+
+        return defer.DeferredList(d_list)