source: ralphm-patches/roster_server.patch @ 51:6edeb69e910c

wokkel/im.py

@@ -23 +23 @@
 NS_ROSTER = 'jabber:iq:roster'
 
 XPATH_ROSTER_SET = "/iq[@type='set']/query[@xmlns='%s']" % NS_ROSTER
+XPATH_ROSTER_GET = "/iq[@type='get']/query[@xmlns='%s']" % NS_ROSTER
 
 
 
@@ -549 +550 @@
         @param entity: The entity for which the roster item has been removed.
         @type entity: L{jid.JID}
         """
+
+
+
+class RosterServerProtocol(XMPPHandler, IQHandlerMixin):
+    """
+    XMPP subprotocol handler for the roster, server side.
+    """
+
+    iqHandlers = {
+            XPATH_ROSTER_GET: '_onRosterGet',
+            # XPATH_ROSTER_SET: '_onRosterSet',
+            }
+
+    def connectionInitialized(self):
+        self.xmlstream.addObserver(XPATH_ROSTER_GET, self.handleRequest)
+        self.xmlstream.addObserver(XPATH_ROSTER_SET, self.handleRequest)
+
+
+    def _toRosterReply(self, roster, request):
+        response = domish.Element((NS_ROSTER, 'query'))
+
+        for item in roster:
+            response.addChild(item.toElement())
+
+        return response
+
+
+    def _onRosterGet(self, iq):
+        request = Stanza.fromElement(iq)
+
+        d = self.getRoster(request)
+        d.addCallback(self._toRosterReply, request)
+        return d
+
+
+    def getRoster(self, entity):
+        raise NotImplementedError()

wokkel/test/test_im.py

@@ -844 +844 @@
         d.addCallback(cb)
         return d
 
+
+
+class RosterServerProtocolTest(unittest.TestCase, TestableRequestHandlerMixin):
+    """
+    Tests for L{im.RosterServerProtocol}.
+    """
+
+    def setUp(self):
+        self.stub = XmlStreamStub()
+        self.service = im.RosterServerProtocol()
+        self.service.makeConnection(self.stub.xmlstream)
+        self.service.connectionInitialized()
+
+
+    def test_onRosterGet(self):
+        """
+        A roster get request should trigger getRoster with the request.
+        """
+        xml = """
+          <iq type='get' from='user@example.org' to='user@example.org'>
+            <query xmlns='jabber:iq:roster'/>
+          </iq>
+        """
+
+        def getRoster(request):
+            self.assertEqual(JID('user@example.org'), request.sender)
+            item = im.RosterItem(JID('other@example.org'), True, False,
+                                 'The User')
+            return defer.succeed([item])
+
+        def cb(element):
+            self.assertEquals('query', element.name)
+            self.assertEquals(NS_ROSTER, element.uri)
+            itemElements = list(domish.generateElementsQNamed(
+                element.children, 'item', NS_ROSTER))
+            self.assertEqual(1, len(itemElements))
+            item = itemElements[0]
+            self.assertEquals(u'other@example.org', item.getAttribute('jid'))
+            self.assertEquals(u'to', item.getAttribute('subscription'))
+            self.assertEquals(u'The User', item.getAttribute('name'))
+
+        self.service.getRoster = getRoster
+        d = self.handleRequest(xml)
+        d.addCallback(cb)
+        return d
+
+
+    def test_onRosterGetNotOverridden(self):
+        """
+        If getRoster is not overridden, return feature-not-implemented.
+        """
+        xml = """
+          <iq type='get' from='user@example.org' to='user@example.org'>
+            <query xmlns='jabber:iq:roster'/>
+          </iq>
+        """
+
+        d = self.handleRequest(xml)
+        self.assertFailure(d, NotImplementedError)
+        return d
+
+
+    def test_onRosterSet(self):
+        """
+        Roster set is not yet supported.
+        """
+        xml = """
+          <iq type='set' from='user@example.org' to='user@example.org'>
+            <query xmlns='jabber:iq:roster'>
+              <item jid='other@example.org' name='Other User'/>
+            </query>
+          </iq>
+        """
+
+        d = self.handleRequest(xml)
+        self.assertFailure(d, NotImplementedError)
+        return d