Changeset 10:a5e3c32d23ca for wokkel/test/test_pubsub.py

Timestamp:

Jan 3, 2008, 10:58:54 AM (15 years ago)

Author:

Ralph Meijer <ralphm@…>

Branch:

default

Convert:

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

Message:

Add unsubscribe method to PubSubClient?.

Author: ralphm
Fixes #4.

File:

• wokkel/test/test_pubsub.py (modified) (6 diffs)

wokkel/test/test_pubsub.py

@@ -1 @@
-# Copyright (c) 2003-2007 Ralph Meijer
+# Copyright (c) 2003-2008 Ralph Meijer
 # See LICENSE for details.
 
@@ -12 +12 @@
 
 from wokkel import pubsub
+from wokkel.test.helpers import XmlStreamStub
+
+try:
+    from twisted.words.protocols.jabber.xmlstream import toResponse
+except ImportError:
+    from wokkel.compat import toResponse
+
+NS_PUBSUB = 'http://jabber.org/protocol/pubsub'
+NS_PUBSUB_ERRORS = 'http://jabber.org/protocol/pubsub#errors'
+
+class PubSubClientTest(unittest.TestCase):
+
+    def setUp(self):
+        self.stub = XmlStreamStub()
+        self.protocol = pubsub.PubSubClient()
+        self.protocol.xmlstream = self.stub.xmlstream
+
+    def test_unsubscribe(self):
+        """
+        Test sending unsubscription request.
+        """
+        d = self.protocol.unsubscribe(JID('pubsub.example.org'), 'test',
+                                      JID('user@example.org'))
+
+        iq = self.stub.output[-1]
+        self.assertEquals('pubsub.example.org', iq.getAttribute('to'))
+        self.assertEquals('set', iq.getAttribute('type'))
+        self.assertEquals('pubsub', iq.pubsub.name)
+        self.assertEquals(NS_PUBSUB, iq.pubsub.uri)
+        children = list(domish.generateElementsQNamed(iq.pubsub.children,
+                                                      'unsubscribe', NS_PUBSUB))
+        self.assertEquals(1, len(children))
+        child = children[0]
+        self.assertEquals('test', child['node'])
+        self.assertEquals('user@example.org', child['jid'])
+
+        self.stub.send(toResponse(iq, 'result'))
+        return d
+
 
 class PubSubServiceTest(unittest.TestCase):
@@ -28 +67 @@
         iq['to'] = 'pubsub.example.org'
         iq['type'] = 'set'
-        iq.addElement(('http://jabber.org/protocol/pubsub', 'pubsub'))
+        iq.addElement((NS_PUBSUB, 'pubsub'))
         iq.pubsub.addElement('publish')
         handler.handleRequest(iq)
@@ -47 +86 @@
         iq['from'] = 'user@example.org'
         iq['to'] = 'pubsub.example.org'
-        iq.addElement(('http://jabber.org/protocol/pubsub', 'pubsub'))
+        iq.addElement((NS_PUBSUB, 'pubsub'))
         iq.pubsub.addElement('publish')
         iq.pubsub.publish['node'] = 'test'
@@ -62 +101 @@
         iq['to'] = 'pubsub.example.org'
         iq['type'] = 'get'
-        iq.addElement(('http://jabber.org/protocol/pubsub', 'pubsub'))
+        iq.addElement((NS_PUBSUB, 'pubsub'))
         iq.pubsub.addElement('options')
         handler.handleRequest(iq)
@@ -69 +108 @@
         self.assertEquals('feature-not-implemented', e.condition)
         self.assertEquals('unsupported', e.appCondition.name)
-        self.assertEquals('http://jabber.org/protocol/pubsub#errors',
-                          e.appCondition.uri)
+        self.assertEquals(NS_PUBSUB_ERRORS, e.appCondition.uri)