diff options
Diffstat (limited to 'tests/live_test.py')
| -rw-r--r-- | tests/live_test.py | 110 |
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())) |