source: ralphm-patches/request-xmpphandler.patch @ 53:fd2b3f70b221

wokkel/subprotocols.py

@@ -94 +94 @@
         self.parent.send(obj)
 
 
+    def request(self, request):
+        """
+        Send an IQ request and track the response.
+
+        This passes the request to the parent for sending and response
+        tracking.
+
+        @see: L{StreamManager.request}.
+        """
+        return self.parent.request(request)
+
+
 
 class XMPPHandlerCollection(object):
     """

wokkel/test/test_subprotocols.py

@@ -71 +71 @@
 
 
 
+class IQGetStanza(generic.Stanza):
+    timeout = None
+
+    stanzaKind = 'iq'
+    stanzaType = 'get'
+    stanzaID = 'test'
+
+
+
 class XMPPHandlerTest(unittest.TestCase):
     """
     Tests for L{subprotocols.XMPPHandler}.
@@ -126 +135 @@
         self.assertIdentical(None, handler.xmlstream)
 
 
+    def test_request(self):
+        """
+        A request is passed up to the stream manager.
+        """
+        class DummyStreamManager(object):
+            def __init__(self):
+                self.requests = []
+
+            def request(self, request):
+                self.requests.append(request)
+                return defer.succeed(None)
+
+        handler = subprotocols.XMPPHandler()
+        handler.parent = DummyStreamManager()
+        request = IQGetStanza()
+        d = handler.request(request)
+        self.assertEquals(1, len(handler.parent.requests))
+        self.assertIdentical(request, handler.parent.requests[-1])
+        return d
+
+
 
 class XMPPHandlerCollectionTest(unittest.TestCase):
     """
@@ -165 +195 @@
 
 
 
-class IQGetStanza(generic.Stanza):
-    stanzaKind = 'iq'
-    stanzaType = 'get'
-    stanzaID = 'test'
-
-
-
 class StreamManagerTest(unittest.TestCase):
     """
     Tests for L{subprotocols.StreamManager}.