Added example live stream test.

Run using:
python tests/live_test.py

1 files changed, 110 insertions, 0 deletions

diff --git a/tests/live_test.py b/tests/live_test.py
new file mode 100644
index 00000000..7e8945f1
--- /dev/null
+++ b/tests/live_test.py
@@ -0,0 +1,110 @@
+from sleekxmpp.test import *
+import sleekxmpp.plugins.xep_0033 as xep_0033
+
+
+class TestLiveStream(SleekTest):
+    """
+    Test that we can test a live stanza stream.
+    """
+
+    def tearDown(self):
+        self.stream_close()
+
+    def testClientConnection(self):
+        """Test that we can interact with a live ClientXMPP instance."""
+        self.stream_start(mode='client',
+                          socket='live',
+                          skip=False,
+                          jid='user@localhost/test',
+                          password='user')
+
+        # Use sid=None to ignore any id sent by the server since
+        # we can't know it in advance.
+        self.stream_recv_header(sfrom='localhost', sid=None)
+        self.stream_send_header(sto='localhost')
+        self.stream_recv_feature("""
+          <stream:features>
+            <starttls xmlns="urn:ietf:params:xml:ns:xmpp-tls" />
+            <mechanisms xmlns="urn:ietf:params:xml:ns:xmpp-sasl">
+              <mechanism>DIGEST-MD5</mechanism>
+              <mechanism>PLAIN</mechanism>
+            </mechanisms>
+            <c xmlns="http://jabber.org/protocol/caps"
+               node="http://www.process-one.net/en/ejabberd/"
+               ver="TQ2JFyRoSa70h2G1bpgjzuXb2sU=" hash="sha-1" />
+            <register xmlns="http://jabber.org/features/iq-register" />
+          </stream:features>
+        """)
+        self.stream_send_feature("""
+          <starttls xmlns="urn:ietf:params:xml:ns:xmpp-tls" />
+        """)
+        self.stream_recv_feature("""
+          <proceed xmlns="urn:ietf:params:xml:ns:xmpp-tls" />
+        """)
+        self.stream_send_header(sto='localhost')
+        self.stream_recv_header(sfrom='localhost', sid=None)
+        self.stream_recv_feature("""
+          <stream:features>
+            <mechanisms xmlns="urn:ietf:params:xml:ns:xmpp-sasl">
+              <mechanism>DIGEST-MD5</mechanism>
+              <mechanism>PLAIN</mechanism>
+            </mechanisms>
+            <c xmlns="http://jabber.org/protocol/caps"
+               node="http://www.process-one.net/en/ejabberd/"
+               ver="TQ2JFyRoSa70h2G1bpgjzuXb2sU="
+               hash="sha-1" />
+            <register xmlns="http://jabber.org/features/iq-register" />
+          </stream:features>
+        """)
+        self.stream_send_feature("""
+          <auth xmlns="urn:ietf:params:xml:ns:xmpp-sasl"
+                mechanism="PLAIN">AHVzZXIAdXNlcg==</auth>
+        """)
+        self.stream_recv_feature("""
+          <success xmlns="urn:ietf:params:xml:ns:xmpp-sasl" />
+        """)
+        self.stream_send_header(sto='localhost')
+        self.stream_recv_header(sfrom='localhost', sid=None)
+        self.stream_recv_feature("""
+          <stream:features>
+            <bind xmlns="urn:ietf:params:xml:ns:xmpp-bind" />
+            <session xmlns="urn:ietf:params:xml:ns:xmpp-session" />
+            <c xmlns="http://jabber.org/protocol/caps"
+               node="http://www.process-one.net/en/ejabberd/"
+               ver="TQ2JFyRoSa70h2G1bpgjzuXb2sU="
+               hash="sha-1" />
+            <register xmlns="http://jabber.org/features/iq-register" />
+          </stream:features>
+        """)
+
+        # Should really use stream_send_iq, but our Iq stanza objects
+        # can't handle bind element payloads yet.
+        self.stream_send_feature("""
+          <iq type="set" id="1">
+            <bind xmlns="urn:ietf:params:xml:ns:xmpp-bind">
+              <resource>test</resource>
+            </bind>
+          </iq>
+        """)
+        self.stream_recv_feature("""
+          <iq type="result" id="1">
+            <bind xmlns="urn:ietf:params:xml:ns:xmpp-bind">
+              <jid>user@localhost/test</jid>
+            </bind>
+          </iq>
+        """)
+        self.stream_close()
+
+
+suite = unittest.TestLoader().loadTestsFromTestCase(TestLiveStream)
+
+if __name__ == '__main__':
+    tests = unittest.TestSuite([suite])
+    result = unittest.TextTestRunner(verbosity=2).run(tests)
+    test_ns = 'http://andyet.net/protocol/tests'
+    print("<tests xmlns='%s' %s %s %s %s />" % (
+        test_ns,
+        'ran="%s"' % result.testsRun,
+        'errors="%s"' % len(result.errors),
+        'fails="%s"' % len(result.failures),
+        'success="%s"' % result.wasSuccessful()))