File: //usr/lib/python3/dist-packages/twisted/words/test/__pycache__/test_jabbersasl.cpython-38.pyc
U
W[�$ � @ s� d dl mZmZ d dlmZ d dlmZ d dlmZ d dl m
Z
d dl m Z m
Z
mZmZ d dlmZ dZee
j�G d d
� d
e��ZG d d � d e j�ZG d
d� de
j�ZG dd� de
j�ZdS )� )�absolute_import�division)� implementer)�defer)�unicode)�unittest)�sasl�sasl_mechanisms� xmlstream�jid)�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 �setUpF 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 )Nr1 )� assertEqualZ condition)�er r r �<lambda>] s �z?SASLInitiatingInitializerTests.test_onFailure.<locals>.<lambda>) r �Element�
addElementr ZDeferredr. Z _deferredZ onFailure�
assertFailurer Z
SASLAuthErrorZ addCallback)r r0 r r r �test_onFailureS 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& r2 � NS_XMPP_SASL�urir! r �r r9 r r r � test_sendAuthInitialResponsec 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& r2 �strr= r r r � test_sendAuthNoInitialResponsep s
z=SASLInitiatingInitializerTests.test_sendAuthNoInitialResponsec C s0 d| j _| j �� | jd }| �dt|�� dS )zS
Test starting authentication where the initial response is empty.
r r �=N)r. r r: r& r2 r r= r r r �!test_sendAuthEmptyInitialResponsez s
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 r5 r; �
addContent� onChallenger2 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 rD s response� z cmVzcG9uc2U=N)
r. r: r r5 r; rE r# r rF r& r2 r rG )r rI 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 r5 r; rF r2 r# r rG rH r r r �test_onChallengeEmpty� s
z4SASLInitiatingInitializerTests.test_onChallengeEmptyc C s@ | j �� }t�tdf�}|�d� | j �|� | �|tj � |S )zJ
Test receiving a challenge message with illegal padding.
r zbXkg=Y2hhbGxlbmdl�
r. r: r r5 r; rE rF r7 r ZSASLIncorrectEncodingErrorrH r r r � test_onChallengeIllegalPadding� s
z=SASLInitiatingInitializerTests.test_onChallengeIllegalPaddingc C s@ | j �� }t�tdf�}|�d� | j �|� | �|tj � |S )zM
Test receiving a challenge message with illegal characters.
r zbXkg*Y2hhbGxlbmdlrN rH r r r �!test_onChallengeIllegalCharacters� s
z@SASLInitiatingInitializerTests.test_onChallengeIllegalCharactersc C s@ | j �� }t�tdf�}|�d� | j �|� | �|tj � |S )z?
Test receiving a malformed challenge message.
r �arN rH r r r �test_onChallengeMalformed� s
z8SASLInitiatingInitializerTests.test_onChallengeMalformedN)r r r r r/ r8 r> rA rC rJ rL rM rO rP rR r r r r r% A 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.
Z
mechanismsr# )Zcontent) r r5 r; r6 r
Zfeaturesr< r! r. r$ r# )r r! Zfeaturer r r �
_setMechanism� s
z8SASLInitiatingInitializerSetMechanismTests._setMechanismc C s0 t �d�| j_ d| j_d}| �|| �|�� dS )zI
Test setting ANONYMOUS as the authentication mechanism.
� example.comNZ ANONYMOUS�r �JIDr( �passwordr2 rU �r r! r r r �test_anonymous� s z9SASLInitiatingInitializerSetMechanismTests.test_anonymousc C s0 t �d�| j_ d| j_d}| �|| �|�� dS )zE
Test setting PLAIN as the authentication mechanism.
�test@example.com�secretZPLAINNrW rZ r r r �
test_plain� s z5SASLInitiatingInitializerSetMechanismTests.test_plainc C s0 t �d�| j_ d| j_d}| �|| �|�� dS )zJ
Test setting DIGEST-MD5 as the authentication mechanism.
r\ r] z
DIGEST-MD5NrW rZ r r r � test_digest s z6SASLInitiatingInitializerSetMechanismTests.test_digestc C s, t �d�| j_ d| j_| �tj| jd� dS )zK
Test using an unacceptable SASL authentication mechanism.
r\ r] �SOMETHING_UNACCEPTABLEN�r rX r( rY Z assertRaisesr ZSASLNoAcceptableMechanismrU r r r r �test_notAcceptable s �z=SASLInitiatingInitializerSetMechanismTests.test_notAcceptablec C s, t �d�| j_ d| j_| �tj| jd� dS )zW
Test using an unacceptable SASL authentication mechanism with no JID.
rV r] r` Nra r r r r � test_notAcceptableWithoutUser s �zHSASLInitiatingInitializerSetMechanismTests.test_notAcceptableWithoutUserN) r r r r r/ rU r[ r^ r_ rb rc r r r r rS � s rS N)Z
__future__r r Zzope.interfacer Ztwisted.internetr Ztwisted.python.compatr Z
twisted.trialr Z twisted.words.protocols.jabberr r r
r Ztwisted.words.xishr r; ZISASLMechanism�objectr rT r" ZTestCaser% rS r r r r �<module> s