Changeset 101:d88462a49f86

Timestamp:

Aug 3, 2011, 9:49:22 AM (11 years ago)

Author:

Ralph Meijer <ralphm@…>

Branch:

default

Message:

Add request method to XMPPHandler, passing up to its StreamManager?.

Location:

wokkel

Files:

• subprotocols.py (modified) (1 diff)
• test/test_subprotocols.py (modified) (3 diffs)

wokkel/subprotocols.py

@@ -93 +93 @@
         """
         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)
 
 

wokkel/test/test_subprotocols.py

@@ -72 +72 @@
 
 
+class IQGetStanza(generic.Stanza):
+    timeout = None
+
+    stanzaKind = 'iq'
+    stanzaType = 'get'
+    stanzaID = 'test'
+
+
+
 class XMPPHandlerTest(unittest.TestCase):
     """
@@ -127 +136 @@
 
 
+    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):
@@ -163 +193 @@
         self.assertNotIn(handler, self.collection)
         self.assertIdentical(None, handler.parent)
-
-
-
-class IQGetStanza(generic.Stanza):
-    stanzaKind = 'iq'
-    stanzaType = 'get'
-    stanzaID = 'test'