From 6b7fde10d3a63353f303abd969bb2c47d3ae3d5f Mon Sep 17 00:00:00 2001
From: Lance Stout <lancestout@gmail.com>
Date: Tue, 30 Aug 2011 22:27:02 -0700
Subject: [PATCH] Test pubsub subscribe.

---
 tests/test_stream_xep_0060.py | 81 +++++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)

diff --git a/tests/test_stream_xep_0060.py b/tests/test_stream_xep_0060.py
index 1d912ca5..a740884e 100644
--- a/tests/test_stream_xep_0060.py
+++ b/tests/test_stream_xep_0060.py
@@ -124,5 +124,86 @@ class TestStreamPubsub(SleekTest):
 
         t.join()
 
+    def testSubscribe(self):
+        """Test subscribing to a node"""
+
+        def run_test(jid, bare, ifrom, send, recv):
+            t = threading.Thread(name='subscribe',
+                                 target=self.xmpp['xep_0060'].subscribe,
+                                 args=('pubsub.example.com', 'some_node'),
+                                 kwargs={'subscribee': jid,
+                                         'bare': bare,
+                                         'ifrom': ifrom})
+            t.start()
+            self.send(send)
+            self.recv(recv)
+            t.join()
+
+        # Case 1: No subscribee, default 'from' JID, bare JID
+        run_test(None, True, None,
+            """
+              <iq type="set" id="1" to="pubsub.example.com">
+                <pubsub xmlns="http://jabber.org/protocol/pubsub">
+                  <subscribe node="some_node" jid="tester@localhost" />
+                </pubsub>
+              </iq>
+            """,
+            """
+              <iq type="result" id="1"
+                  to="tester@localhost" from="pubsub.example.com" />
+            """)
+
+        # Case 2: No subscribee, given 'from' JID, bare JID
+        run_test(None, True, 'foo@comp.example.com/bar',
+            """
+              <iq type="set" id="2"
+                  to="pubsub.example.com" from="foo@comp.example.com/bar">
+                <pubsub xmlns="http://jabber.org/protocol/pubsub">
+                  <subscribe node="some_node" jid="foo@comp.example.com" />
+                </pubsub>
+              </iq>
+            """,
+            """
+              <iq type="result" id="2"
+                  to="foo@comp.example.com/bar" from="pubsub.example.com" />
+            """)
+
+        # Case 3: No subscribee, given 'from' JID, full JID
+        run_test(None, False, 'foo@comp.example.com/bar',
+            """
+              <iq type="set" id="3"
+                  to="pubsub.example.com" from="foo@comp.example.com/bar">
+                <pubsub xmlns="http://jabber.org/protocol/pubsub">
+                  <subscribe node="some_node" jid="foo@comp.example.com/bar" />
+                </pubsub>
+              </iq>
+            """,
+            """
+              <iq type="result" id="3"
+                  to="foo@comp.example.com/bar" from="pubsub.example.com" />
+            """)
+
+        # Case 4: Subscribee
+        run_test('user@example.com/foo', True, 'foo@comp.example.com/bar',
+            """
+              <iq type="set" id="4"
+                  to="pubsub.example.com" from="foo@comp.example.com/bar">
+                <pubsub xmlns="http://jabber.org/protocol/pubsub">
+                  <subscribe node="some_node" jid="user@example.com/foo" />
+                </pubsub>
+              </iq>
+            """,
+            """
+              <iq type="result" id="4"
+                  to="foo@comp.example.com/bar" from="pubsub.example.com" />
+            """)
+
+    def testSubscribeWithOptions(self):
+        pass
+
+    def testUnsubscribe(self):
+        pass
+
+
 
 suite = unittest.TestLoader().loadTestsFromTestCase(TestStreamPubsub)