0x1949 Team - FAZEMRX - MANAGER
Edit File: test_jabbersasl.cpython-310.pyc
o �b�# � @ s� d dl mZ d dlmZ d dlmZ d dlmZmZm Z m Z d dl m Z dZ ee j�G dd� d��ZG d d � d ej�ZG d d � d ej�ZG d d� dej�ZdS )� )� implementer)�defer)�unittest)�jid�sasl�sasl_mechanisms� xmlstream)�domish� urn:ietf:params:xml:ns:xmpp-saslc @ s4 e Zd ZdZdZdZdZdd� Zdd� Zd d � Z dS ) �DummySASLMechanisma� 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} N�DUMMY� c C s || _ d S �N��initialResponse)�selfr � r �D/usr/lib/python3/dist-packages/twisted/words/test/test_jabbersasl.py�__init__" s zDummySASLMechanism.__init__c C s | j S r r �r r r r �getInitialResponse% s z%DummySASLMechanism.getInitialResponsec C s || _ | jS r )� challenge�response)r r r r r � getResponse( s z DummySASLMechanism.getResponse) �__name__� __module__� __qualname__�__doc__r �namer r r r r r r r r s r c @ s e Zd ZdZdZdd� ZdS )� DummySASLInitiatingInitializerac 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}. Nc C s t | j�| _d S r )r r � mechanismr r r r � setMechanism; s z+DummySASLInitiatingInitializer.setMechanism)r r r r r r! r r r r r - s r c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� ZdS )� SASLInitiatingInitializerTestsz5 Tests for L{sasl.SASLInitiatingInitializer} c C sP g | _ t�� | _t�| j�| _| j j| j_| j�� | j�d� t | j�| _ d S )Ns� <stream:stream xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' from='example.com' id='12345' version='1.0'>) �outputr � Authenticator� authenticator� XmlStream�append�send�connectionMade� dataReceivedr �initr r r r �setUpD s �z$SASLInitiatingInitializerTests.setUpc s\ t �d�}|�d� t�� � j_� j�|� � �� jjt j � � jj� � fdd�� � jjS )zL Test that the SASL error condition is correctly extracted. )r �failure�not-authorizedc s � � d| j�S )Nr. )� assertEqual� condition)�er r r �<lambda>\ s z?SASLInitiatingInitializerTests.test_onFailure.<locals>.<lambda>) r �Element� addElementr �Deferredr+ � _deferred� onFailure� assertFailurer � SASLAuthError� addCallback)r r- r r r �test_onFailureR s �z-SASLInitiatingInitializerTests.test_onFailurec C s\ d| j _| j �� | jd }| �t|j� | �d|j� | �d|d � | �dt|�� dS )zH Test starting authentication with an initial response. s dummyr �authr r zZHVtbXk=N) r+ r �startr# r/ � NS_XMPP_SASL�urir �str�r r< r r r � test_sendAuthInitialResponse` s z;SASLInitiatingInitializerTests.test_sendAuthInitialResponsec C s0 d| j _| j �� | jd }| �dt|�� dS )zK Test starting authentication without an initial response. Nr � �r+ r r= r# r/ r@ rA r r r � test_sendAuthNoInitialResponsel � z=SASLInitiatingInitializerTests.test_sendAuthNoInitialResponsec C s0 d| j _| j �� | jd }| �dt|�� dS )zS Test starting authentication where the initial response is empty. r r �=NrD rA r r r �!test_sendAuthEmptyInitialResponseu rF z@SASLInitiatingInitializerTests.test_sendAuthEmptyInitialResponsec C sP | j �� }t�tdf�}|�d� | j �|� | �d| j jj � | j � d� |S )z5 Test receiving a challenge message. r �bXkgY2hhbGxlbmdls my challengeN) r+ r= r r3 r> � addContent� onChallenger/ r r � onSuccess�r �dr r r r �test_onChallenge~ s z/SASLInitiatingInitializerTests.test_onChallengec C sb | j �� }t�tdf�}|�d� d| j j_| j �|� | j d }| � dt |�� | j � d� |S )zS A non-empty response gets encoded and included as character data. r rI s response� z cmVzcG9uc2U=N) r+ r= r r3 r> rJ r r rK r# r/ r@ rL )r rN r r r r r �test_onChallengeResponse� s z7SASLInitiatingInitializerTests.test_onChallengeResponsec C sF | j �� }t�tdf�}| j �|� | �d| j jj� | j � d� |S )z< Test receiving an empty challenge message. r r N) r+ r= r r3 r> rK r/ r r rL rM r r r �test_onChallengeEmpty� s z4SASLInitiatingInitializerTests.test_onChallengeEmptyc C �@ | j �� }t�tdf�}|�d� | j �|� | �|tj � |S )zJ Test receiving a challenge message with illegal padding. r zbXkg=Y2hhbGxlbmdl� r+ r= r r3 r> rJ rK r8 r �SASLIncorrectEncodingErrorrM r r r � test_onChallengeIllegalPadding� � z=SASLInitiatingInitializerTests.test_onChallengeIllegalPaddingc C rS )zM Test receiving a challenge message with illegal characters. r zbXkg*Y2hhbGxlbmdlrT rM r r r �!test_onChallengeIllegalCharacters� rW z@SASLInitiatingInitializerTests.test_onChallengeIllegalCharactersc C rS )z? Test receiving a malformed challenge message. r �arT rM r r r �test_onChallengeMalformed� rW z8SASLInitiatingInitializerTests.test_onChallengeMalformedN)r r r r r, r; rB rE rH rO rQ rR rV rX rZ r r r r r" ? s r" c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )�*SASLInitiatingInitializerSetMechanismTestszB Test for L{sasl.SASLInitiatingInitializer.setMechanism}. c C sR g | _ t�� | _t�| j�| _| j j| j_| j�� | j�d� t � | j�| _ d S )Nz�<stream:stream xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' from='example.com' id='12345' version='1.0'>) r# r r$ r% r& r' r( r) r* r �SASLInitiatingInitializerr+ r r r r r, � s �z0SASLInitiatingInitializerSetMechanismTests.setUpc C sD t �tdf�}|jd|d� || jj|j|jf< | j� � | jj jS )zX Set up the XML Stream to have a SASL feature with the given mechanism. � mechanismsr )�content) r r3 r> r4 r �featuresr? r r+ r! r )r r �featurer r r � _setMechanism� s z8SASLInitiatingInitializerSetMechanismTests._setMechanismc C s0 t �d�| j_ d| j_d}| �|| �|�� dS )zI Test setting ANONYMOUS as the authentication mechanism. � example.comN� ANONYMOUS�r �JIDr% �passwordr/ ra �r r r r r �test_anonymous� � z9SASLInitiatingInitializerSetMechanismTests.test_anonymousc C �0 t �d�| j_ d| j_d}| �|| �|�� dS )zE Test setting PLAIN as the authentication mechanism. �test@example.com�secret�PLAINNrd rg r r r � test_plain� ri z5SASLInitiatingInitializerSetMechanismTests.test_plainc C rj )zJ Test setting DIGEST-MD5 as the authentication mechanism. rk rl z DIGEST-MD5Nrd rg r r r � test_digest� ri z6SASLInitiatingInitializerSetMechanismTests.test_digestc C �, t �d�| j_ d| j_| �tj| jd� dS )zK Test using an unacceptable SASL authentication mechanism. rk rl �SOMETHING_UNACCEPTABLEN�r re r% rf � assertRaisesr �SASLNoAcceptableMechanismra r r r r �test_notAcceptable s �z=SASLInitiatingInitializerSetMechanismTests.test_notAcceptablec C rp )zW Test using an unacceptable SASL authentication mechanism with no JID. rb rl rq Nrr r r r r � test_notAcceptableWithoutUser s �zHSASLInitiatingInitializerSetMechanismTests.test_notAcceptableWithoutUserN) r r r r r, ra rh rn ro ru rv r r r r r[ � s r[ N)�zope.interfacer �twisted.internetr � twisted.trialr � twisted.words.protocols.jabberr r r r �twisted.words.xishr r> �ISASLMechanismr r\ r �TestCaser" r[ r r r r �<module> s