File: //lib/python3/dist-packages/twisted/words/test/__pycache__/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