Gecko [ aodai.devttt.com ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
__init__.py	14 B	-rw-r--r--
test_basechat.py	2.42 KB	-rw-r--r--
test_basesupport.py	2.93 KB	-rw-r--r--
test_domish.py	19.36 KB	-rw-r--r--
test_irc.py	98.94 KB	-rw-r--r--
test_irc_service.py	9.81 KB	-rw-r--r--
test_ircsupport.py	10.36 KB	-rw-r--r--
test_jabberclient.py	16.16 KB	-rw-r--r--
test_jabbercomponent.py	13.58 KB	-rw-r--r--
test_jabbererror.py	11.08 KB	-rw-r--r--
test_jabberjid.py	6.72 KB	-rw-r--r--
test_jabberjstrports.py	945 B	-rw-r--r--
test_jabbersasl.py	8.95 KB	-rw-r--r--
test_jabbersaslmechanisms.py	5.83 KB	-rw-r--r--
test_jabberxmlstream.py	44.04 KB	-rw-r--r--
test_jabberxmppstringprep.py	5.31 KB	-rw-r--r--
test_service.py	27.99 KB	-rw-r--r--
test_tap.py	2.12 KB	-rw-r--r--
test_xishutil.py	9.11 KB	-rw-r--r--
test_xmlstream.py	5.92 KB	-rw-r--r--
test_xmpproutertap.py	2.34 KB	-rw-r--r--
test_xpath.py	10.35 KB	-rw-r--r--

Code Editor : test_jabbersasl.py

# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.

from zope.interface import implementer

from twisted.internet import defer
from twisted.trial import unittest
from twisted.words.protocols.jabber import jid, sasl, sasl_mechanisms, xmlstream
from twisted.words.xish import domish

NS_XMPP_SASL = "urn:ietf:params:xml:ns:xmpp-sasl"

@implementer(sasl_mechanisms.ISASLMechanism)
class DummySASLMechanism:
    """
    Dummy SASL mechanism.

This just returns the initialResponse passed on creation, stores any
    challenges and replies with the value of C{response}.

@ivar challenge: Last received challenge.
    @type challenge: C{unicode}.
    @ivar initialResponse: Initial response to be returned when requested
                           via C{getInitialResponse} or L{None}.
    @type initialResponse: C{unicode}
    """

challenge = None
    name = "DUMMY"
    response = b""

def __init__(self, initialResponse):
        self.initialResponse = initialResponse

def getInitialResponse(self):
        return self.initialResponse

def getResponse(self, challenge):
        self.challenge = challenge
        return self.response

class DummySASLInitiatingInitializer(sasl.SASLInitiatingInitializer):
    """
    Dummy SASL Initializer for initiating entities.

This hardwires the SASL mechanism to L{DummySASLMechanism}, that is
    instantiated with the value of C{initialResponse}.

@ivar initialResponse: The initial response to be returned by the
                           dummy SASL mechanism or L{None}.
    @type initialResponse: C{unicode}.
    """

initialResponse = None

def setMechanism(self):
        self.mechanism = DummySASLMechanism(self.initialResponse)

class SASLInitiatingInitializerTests(unittest.TestCase):
    """
    Tests for L{sasl.SASLInitiatingInitializer}
    """

def setUp(self):
        self.output = []

self.authenticator = xmlstream.Authenticator()
        self.xmlstream = xmlstream.XmlStream(self.authenticator)
        self.xmlstream.send = self.output.append
        self.xmlstream.connectionMade()
        self.xmlstream.dataReceived(
            b"<stream:stream xmlns='jabber:client' "
            b"xmlns:stream='http://etherx.jabber.org/streams' "
            b"from='example.com' id='12345' version='1.0'>"
        )
        self.init = DummySASLInitiatingInitializer(self.xmlstream)

def test_onFailure(self):
        """
        Test that the SASL error condition is correctly extracted.
        """
        failure = domish.Element(("urn:ietf:params:xml:ns:xmpp-sasl", "failure"))
        failure.addElement("not-authorized")
        self.init._deferred = defer.Deferred()
        self.init.onFailure(failure)
        self.assertFailure(self.init._deferred, sasl.SASLAuthError)
        self.init._deferred.addCallback(
            lambda e: self.assertEqual("not-authorized", e.condition)
        )
        return self.init._deferred

def test_sendAuthInitialResponse(self):
        """
        Test starting authentication with an initial response.
        """
        self.init.initialResponse = b"dummy"
        self.init.start()
        auth = self.output[0]
        self.assertEqual(NS_XMPP_SASL, auth.uri)
        self.assertEqual("auth", auth.name)
        self.assertEqual("DUMMY", auth["mechanism"])
        self.assertEqual("ZHVtbXk=", str(auth))

def test_sendAuthNoInitialResponse(self):
        """
        Test starting authentication without an initial response.
        """
        self.init.initialResponse = None
        self.init.start()
        auth = self.output[0]
        self.assertEqual("", str(auth))

def test_sendAuthEmptyInitialResponse(self):
        """
        Test starting authentication where the initial response is empty.
        """
        self.init.initialResponse = b""
        self.init.start()
        auth = self.output[0]
        self.assertEqual("=", str(auth))

def test_onChallenge(self):
        """
        Test receiving a challenge message.
        """
        d = self.init.start()
        challenge = domish.Element((NS_XMPP_SASL, "challenge"))
        challenge.addContent("bXkgY2hhbGxlbmdl")
        self.init.onChallenge(challenge)
        self.assertEqual(b"my challenge", self.init.mechanism.challenge)
        self.init.onSuccess(None)
        return d

def test_onChallengeResponse(self):
        """
        A non-empty response gets encoded and included as character data.
        """
        d = self.init.start()
        challenge = domish.Element((NS_XMPP_SASL, "challenge"))
        challenge.addContent("bXkgY2hhbGxlbmdl")
        self.init.mechanism.response = b"response"
        self.init.onChallenge(challenge)
        response = self.output[1]
        self.assertEqual("cmVzcG9uc2U=", str(response))
        self.init.onSuccess(None)
        return d

def test_onChallengeEmpty(self):
        """
        Test receiving an empty challenge message.
        """
        d = self.init.start()
        challenge = domish.Element((NS_XMPP_SASL, "challenge"))
        self.init.onChallenge(challenge)
        self.assertEqual(b"", self.init.mechanism.challenge)
        self.init.onSuccess(None)
        return d

def test_onChallengeIllegalPadding(self):
        """
        Test receiving a challenge message with illegal padding.
        """
        d = self.init.start()
        challenge = domish.Element((NS_XMPP_SASL, "challenge"))
        challenge.addContent("bXkg=Y2hhbGxlbmdl")
        self.init.onChallenge(challenge)
        self.assertFailure(d, sasl.SASLIncorrectEncodingError)
        return d

def test_onChallengeIllegalCharacters(self):
        """
        Test receiving a challenge message with illegal characters.
        """
        d = self.init.start()
        challenge = domish.Element((NS_XMPP_SASL, "challenge"))
        challenge.addContent("bXkg*Y2hhbGxlbmdl")
        self.init.onChallenge(challenge)
        self.assertFailure(d, sasl.SASLIncorrectEncodingError)
        return d

def test_onChallengeMalformed(self):
        """
        Test receiving a malformed challenge message.
        """
        d = self.init.start()
        challenge = domish.Element((NS_XMPP_SASL, "challenge"))
        challenge.addContent("a")
        self.init.onChallenge(challenge)
        self.assertFailure(d, sasl.SASLIncorrectEncodingError)
        return d

class SASLInitiatingInitializerSetMechanismTests(unittest.TestCase):
    """
    Test for L{sasl.SASLInitiatingInitializer.setMechanism}.
    """

def setUp(self):
        self.output = []

self.authenticator = xmlstream.Authenticator()
        self.xmlstream = xmlstream.XmlStream(self.authenticator)
        self.xmlstream.send = self.output.append
        self.xmlstream.connectionMade()
        self.xmlstream.dataReceived(
            "<stream:stream xmlns='jabber:client' "
            "xmlns:stream='http://etherx.jabber.org/streams' "
            "from='example.com' id='12345' version='1.0'>"
        )

self.init = sasl.SASLInitiatingInitializer(self.xmlstream)

def _setMechanism(self, name):
        """
        Set up the XML Stream to have a SASL feature with the given mechanism.
        """
        feature = domish.Element((NS_XMPP_SASL, "mechanisms"))
        feature.addElement("mechanism", content=name)
        self.xmlstream.features[(feature.uri, feature.name)] = feature

self.init.setMechanism()
        return self.init.mechanism.name

def test_anonymous(self):
        """
        Test setting ANONYMOUS as the authentication mechanism.
        """
        self.authenticator.jid = jid.JID("example.com")
        self.authenticator.password = None
        name = "ANONYMOUS"

self.assertEqual(name, self._setMechanism(name))

def test_plain(self):
        """
        Test setting PLAIN as the authentication mechanism.
        """
        self.authenticator.jid = jid.JID("test@example.com")
        self.authenticator.password = "secret"
        name = "PLAIN"

self.assertEqual(name, self._setMechanism(name))

def test_digest(self):
        """
        Test setting DIGEST-MD5 as the authentication mechanism.
        """
        self.authenticator.jid = jid.JID("test@example.com")
        self.authenticator.password = "secret"
        name = "DIGEST-MD5"

self.assertEqual(name, self._setMechanism(name))

def test_notAcceptable(self):
        """
        Test using an unacceptable SASL authentication mechanism.
        """

self.authenticator.jid = jid.JID("test@example.com")
        self.authenticator.password = "secret"

self.assertRaises(
            sasl.SASLNoAcceptableMechanism, self._setMechanism, "SOMETHING_UNACCEPTABLE"
        )

def test_notAcceptableWithoutUser(self):
        """
        Test using an unacceptable SASL authentication mechanism with no JID.
        """
        self.authenticator.jid = jid.JID("example.com")
        self.authenticator.password = "secret"

self.assertRaises(
            sasl.SASLNoAcceptableMechanism, self._setMechanism, "SOMETHING_UNACCEPTABLE"
        )

${this.title}

Code Editor : test_jabbersasl.py