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()))