Changeset 174:c0f51d95bf0f for wokkel/test/test_xmppim.py

Timestamp:

May 9, 2012, 2:24:28 PM (10 years ago)

Author:

Ralph Meijer <ralphm@…>

Branch:

default

Message:

Add support for adding and updating roster items.

Roster items can be added or updated by sending a roster set request using the
new wokkel.xmppim.RosterClientProtocol.setItem. It takes a RosterItem as
the only argument. Note that changes in presence subscriptions for the contact
need to be done using presence as provided by
wokkel.xmppim.PresenceProtocol.

Author: ralphm.
Fixes: #56.

File:

• wokkel/test/test_xmppim.py (modified) (1 diff)

wokkel/test/test_xmppim.py

@@ -966 +966 @@
 
 
+    def test_setItem(self):
+        """
+        Setting a roster item renders the item and sends it out.
+        """
+        item = xmppim.RosterItem(JID('test@example.org'),
+                                 name='Joe User',
+                                 groups=set(['Friends', 'Jabber']))
+        d = self.service.setItem(item)
+
+        # Inspect outgoing iq request
+
+        iq = self.stub.output[-1]
+        self.assertEqual('set', iq.getAttribute('type'))
+        self.assertNotIdentical(None, iq.query)
+        self.assertEqual(NS_ROSTER, iq.query.uri)
+
+        children = list(domish.generateElementsQNamed(iq.query.children,
+                                                      'item', NS_ROSTER))
+        self.assertEqual(1, len(children))
+        child = children[0]
+        self.assertEqual('test@example.org', child['jid'])
+        self.assertIdentical(None, child.getAttribute('subscription'))
+
+        # Fake successful response
+
+        response = toResponse(iq, 'result')
+        d.callback(response)
+        return d
+
+
+    def test_setItemIgnoreAttributes(self):
+        """
+        Certain attributes should be rendered for roster set.
+        """
+        item = xmppim.RosterItem(JID('test@example.org'),
+                                 subscriptionTo=True,
+                                 subscriptionFrom=False,
+                                 name='Joe User',
+                                 groups=set(['Friends', 'Jabber']))
+        item.pendingOut = True
+        item.approved = True
+        d = self.service.setItem(item)
+
+        # Inspect outgoing iq request
+
+        iq = self.stub.output[-1]
+        self.assertEqual('set', iq.getAttribute('type'))
+        self.assertNotIdentical(None, iq.query)
+        self.assertEqual(NS_ROSTER, iq.query.uri)
+
+        children = list(domish.generateElementsQNamed(iq.query.children,
+                                                      'item', NS_ROSTER))
+        self.assertEqual(1, len(children))
+        child = children[0]
+        self.assertIdentical(None, child.getAttribute('ask'))
+        self.assertIdentical(None, child.getAttribute('approved'))
+        self.assertIdentical(None, child.getAttribute('subscription'))
+
+        # Fake successful response
+
+        response = toResponse(iq, 'result')
+        d.callback(response)
+        return d
+
+
     def test_removeItem(self):
         """