Changeset 61:5a5840e066fb for wokkel/test/test_client.py

Timestamp:

Jun 12, 2009, 5:32:06 PM (13 years ago)

Author:

Ralph Meijer <ralphm@…>

Branch:

default

Convert:

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

Message:

Fix regression that made DeferredClientFactory? have no stream manager.

Author: ralphm.
Fixes #51.

File:

• wokkel/test/test_client.py (modified) (2 diffs)

wokkel/test/test_client.py

@@ -13 +13 @@
 from twisted.words.protocols.jabber.xmlstream import STREAM_AUTHD_EVENT
 from twisted.words.protocols.jabber.xmlstream import INIT_FAILED_EVENT
+
+try:
+    from twisted.words.protocols.jabber.xmlstream import XMPPHandler
+except ImportError:
+    from wokkel.subprotocols import XMPPHandler
 
 from wokkel import client
@@ -78 +83 @@
 
 
+    def test_addHandler(self):
+        """
+        Test the addition of a protocol handler.
+        """
+        handler = XMPPHandler()
+        handler.setHandlerParent(self.factory.streamManager)
+        self.assertIn(handler, self.factory.streamManager)
+        self.assertIdentical(self.factory.streamManager, handler.parent)
+
+
+    def test_removeHandler(self):
+        """
+        Test removal of a protocol handler.
+        """
+        handler = XMPPHandler()
+        handler.setHandlerParent(self.factory.streamManager)
+        handler.disownHandlerParent(self.factory.streamManager)
+        self.assertNotIn(handler, self.factory.streamManager)
+        self.assertIdentical(None, handler.parent)
+
+
 
 class ClientCreatorTest(unittest.TestCase):