Changeset 46:4ee1f9c08b22 for wokkel/test/helpers.py

Timestamp:

Dec 26, 2008, 2:40:11 PM (14 years ago)

Author:

Ralph Meijer <ralphm@…>

Branch:

default

Convert:

svn:b33ecbfc-034c-dc11-8662-000475d9059e/trunk@140

Message:

If provided, return NodeID in the response for Service Discovery requests.

This also renames the namespace constants to include DISCO in their name
and provides better tests and docstrings.

Author: ralphm.
Reviewer: tofu.
Fixes #7.

File:

• wokkel/test/helpers.py (modified) (2 diffs)

wokkel/test/helpers.py

@@ -6 +6 @@
 """
 
+from twisted.internet import defer
+from twisted.words.xish import xpath
 from twisted.words.xish.utility import EventDispatcher
+
+from wokkel.generic import parseXml
 
 class XmlStreamStub(object):
@@ -55 +59 @@
         """
         self.xmlstream.dispatch(obj)
+
+
+class TestableRequestHandlerMixin(object):
+    """
+    Mixin for testing XMPPHandlers that process iq requests.
+
+    Handlers that use L{wokkel.subprotocols.IQHandlerMixin} define a
+    C{iqHandlers} attribute that lists the handlers to be called for iq
+    requests. This mixin provides L{handleRequest} to mimic the handler
+    processing for easier testing.
+    """
+
+    def handleRequest(self, xml):
+        """
+        Find a handler and call it directly.
+
+        @param xml: XML stanza that may yield a handler being called.
+        @type xml: C{str}.
+        @return: Deferred that fires with the result of a handler for this
+                 stanza. If no handler was found, the deferred has its errback
+                 called with a C{NotImplementedError} exception.
+        """
+        handler = None
+        iq = parseXml(xml)
+        for queryString, method in self.service.iqHandlers.iteritems():
+            if xpath.internQuery(queryString).matches(iq):
+                handler = getattr(self.service, method)
+
+        if handler:
+            d = defer.maybeDeferred(handler, iq)
+        else:
+            d = defer.fail(NotImplementedError())
+
+        return d