Changeset 17:d42fd6c92576 for wokkel

Timestamp:

Apr 15, 2008, 7:32:14 PM (14 years ago)

Author:

Ralph Meijer <ralphm@…>

Branch:

default

Convert:

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

Message:

Add client support for getting items from a node.

Location:

wokkel

Files:

• pubsub.py (modified) (2 diffs)
• test/test_pubsub.py (modified) (1 diff)

wokkel/pubsub.py

@@ -231 +231 @@
         return item
 
+class Items(PubSubRequest):
+    verb = 'items'
+    method = 'get'
+
+    def __init__(self, xs, node):
+        PubSubRequest.__init__(self, xs)
+        self.command["node"] = node
 
 class PubSubClient(XMPPHandler):
@@ -334 +341 @@
         return request.send(service)
 
+    def items(self, service, nodeIdentifier):
+        def cb(iq):
+            items = []
+            for element in iq.pubsub.items.elements():
+                if element.uri == NS_PUBSUB and element.name == 'item':
+                    items.append(element)
+            return items
+
+        request = Items(self.xmlstream, nodeIdentifier)
+        return request.send(service).addCallback(cb)
 
 class PubSubService(XMPPHandler, IQHandlerMixin):

wokkel/test/test_pubsub.py

@@ -140 +140 @@
         return d
 
+
+    def test_items(self):
+        """
+        Test sending items request.
+        """
+        def cb(items):
+            self.assertEquals([], items)
+
+        d = self.protocol.items(JID('pubsub.example.org'), 'test')
+        d.addCallback(cb)
+
+        iq = self.stub.output[-1]
+        self.assertEquals('pubsub.example.org', iq.getAttribute('to'))
+        self.assertEquals('get', iq.getAttribute('type'))
+        self.assertEquals('pubsub', iq.pubsub.name)
+        self.assertEquals(NS_PUBSUB, iq.pubsub.uri)
+        children = list(domish.generateElementsQNamed(iq.pubsub.children,
+                                                      'items', NS_PUBSUB))
+        self.assertEquals(1, len(children))
+        child = children[0]
+        self.assertEquals('test', child['node'])
+
+        response = toResponse(iq, 'result')
+        items = response.addElement((NS_PUBSUB, 'pubsub')).addElement('items')
+        items['node'] = 'test'
+
+        self.stub.send(response)
+
+        return d
+
 class PubSubServiceTest(unittest.TestCase):