File: //usr/lib/python3/dist-packages/twisted/words/test/__pycache__/test_jabberclient.cpython-38.pyc
U
s�@g�A � @ sL d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
mZmZmZ ddlmZ dd lmZ zdd
lmZ W n ek
r� d Zd ZY nX d Zd
ZdZdZde ZdZ de Z G dd� de j �Z G dd� de �Z!G dd� de!e j �Z"G dd� de!e j �Z#G dd � d e!e j �Z$G d d � d e j �Z%G d d � d e j �Z&d S )!z4
Tests for L{twisted.words.protocols.jabber.client}
� )�absolute_import�division)�sha1)�defer)�unicode)�unittest)�client�error�jid� xmlstream)�SASLInitiatingInitializer)�utility)�sslNzSSL not availablez//iq[@type="get"]/query[@xmlns="jabber:iq:auth"]z//iq[@type="set"]/query[@xmlns="jabber:iq:auth"]z urn:ietf:params:xml:ns:xmpp-bindz"/iq[@type="set"]/bind[@xmlns="%s"]z#urn:ietf:params:xml:ns:xmpp-sessionz%/iq[@type="set"]/session[@xmlns="%s"]c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� CheckVersionInitializerTestsc C s" t �� }t �|�}t�|�| _d S �N)r Z
AuthenticatorZ XmlStreamr ZCheckVersionInitializer�init)�self�a�xs� r �F/usr/lib/python3/dist-packages/twisted/words/test/test_jabberclient.py�setUp# s
z"CheckVersionInitializerTests.setUpc C s d| j j_| j �� dS )z3
Test supported version number 1.0
)� r N)r r �version�
initialize�r r r r �
testSupported) s
z*CheckVersionInitializerTests.testSupportedc C s. d| j j_| �tj| j j�}| �d|j� dS )zK
Test unsupported version number 0.0, and check exception.
)r r zunsupported-versionN) r r r Z assertRaisesr Z StreamErrorr � assertEqualZ condition)r �excr r r �testNotSupported1 s
z-CheckVersionInitializerTests.testNotSupportedN)�__name__�
__module__� __qualname__r r r r r r r r " s r c @ s e Zd ZdZdd� Zdd� ZdS )� InitiatingInitializerHarnessa�
Testing harness for interacting with XML stream initializers.
This sets up an L{utility.XmlPipe} to create a communication channel between
the initializer and the stubbed receiving entity. It features a sink and
source side that both act similarly to a real L{xmlstream.XmlStream}. The
sink is augmented with an authenticator to which initializers can be added.
The harness also provides some utility methods to work with event observers
and deferreds.
c C s4 g | _ t�� | _| jj| _t�d�| _| j| j_d S )Nz example.org)�outputr
ZXmlPipe�pipeZsinkr ZConnectAuthenticator�
authenticatorr r r r r H s
z"InitiatingInitializerHarness.setUpc C s( t �� }|�|� | jj�||j� |S )a�
Observe an output event, returning a deferred.
The returned deferred will be fired when the given event has been
observed on the source end of the L{XmlPipe} tied to the protocol
under test. The handler is added as the first callback.
@param event: The event to be observed. See
L{utility.EventDispatcher.addOnetimeObserver}.
@param handler: The handler to be called with the observed event object.
@rtype: L{defer.Deferred}.
)r ZDeferred� addCallbackr% �sourceZaddOnetimeObserver�callback)r ZeventZhandler�dr r r �waitForP s
z$InitiatingInitializerHarness.waitForN)r r! r" �__doc__r r+ r r r r r# ; s r# c s@ e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Z� Z S ) �IQAuthInitializerTestsz0
Tests for L{client.IQAuthInitializer}.
c s6 t t| ��� t�| j�| _t�d�| j _d| j _
d S )N�user@example.com/resource�secret) �superr- r r �IQAuthInitializerr r r
�JIDr& �passwordr �� __class__r r r i s z IQAuthInitializerTests.setUpc s> � �fdd�}�fdd�� �� t|�}�j�� }t�||g�S )z�
Test plain-text authentication.
Act as a server supporting plain-text authentication and expect the
C{password} field to be filled with the password. Then act as if
authentication succeeds.
c sX t �| d�}|�d� |j�d� |j�d� |j�d� ��t� �}�jj�|� |S �z�
Called when the initializer sent a query for authentication methods.
The response informs the client that plain-text authentication
is supported.
�result�zjabber:iq:auth�query�usernamer3 �resource� r �
toResponse�
addElementr9 r+ � IQ_AUTH_SETr% r( �send��iq�responser* �� onAuthSetr r r � onAuthGety s
z7IQAuthInitializerTests.testPlainText.<locals>.onAuthGetc sZ � � dt| jj�� � � dt| jj�� � � dt| jj�� t�| d�}� jj �
|� dS )��
Called when the initializer sent the authentication request.
The server checks the credentials and responds with an empty result
signalling success.
�userr/ r; r7 N) r r r9 r: r3 r; r r= r% r( r@ �rB rC r r r rE � s
z7IQAuthInitializerTests.testPlainText.<locals>.onAuthSet)r+ � IQ_AUTH_GETr r r �
gatherResults�r rF �d1�d2r rD r �
testPlainTextp s
z$IQAuthInitializerTests.testPlainTextc sF � �fdd�}�fdd�� d�j _��t|�}�j�� }t�||g�S )a"
Test digest authentication.
Act as a server supporting digest authentication and expect the
C{digest} field to be filled with a sha1 digest of the concatenated
stream session identifier and password. Then act as if authentication
succeeds.
c sX t �| d�}|�d� |j�d� |j�d� |j�d� ��t� �}�jj�|� |S )z�
Called when the initializer sent a query for authentication methods.
The response informs the client that digest authentication is
supported.
r7 r8 r: �digestr; r< rA rD r r rF � s
z4IQAuthInitializerTests.testDigest.<locals>.onAuthGetc sb � � dt| jj�� � � td��� t| jj�� � � dt| jj�� t� | d�}� j
j � |� dS )rG rH s 12345secretr; r7 N)
r r r9 r: r Z hexdigestrP r; r r= r% r( r@ rI r r r rE � s
� z4IQAuthInitializerTests.testDigest.<locals>.onAuthSetZ12345)r Zsidr+ rJ r r r rK rL r rD r �
testDigest� s
z!IQAuthInitializerTests.testDigestc s> � fdd�}� � t|�}� j�� }� �|tj� t�||g�S )zT
Test initializer failure of request for fields for authentication.
c s" t �d��| �}� jj�|� dS )z�
Called when the initializer sent a query for authentication methods.
The server responds that the client is not authorized to authenticate.
�not-authorizedN�r � StanzaErrorr= r% r( r@ rI r r r rF � s z?IQAuthInitializerTests.testFailRequestFields.<locals>.onAuthGet� r+ rJ r r �
assertFailurer rT r rK rL r r r �testFailRequestFields� s
z,IQAuthInitializerTests.testFailRequestFieldsc sL � �fdd�}�fdd�� �� t|�}�j�� }��|tj� t�||g�S )z;
Test initializer failure to authenticate.
c sX t �| d�}|�d� |j�d� |j�d� |j�d� ��t� �}�jj�|� |S r6 r< rA rD r r rF s
z6IQAuthInitializerTests.testFailAuth.<locals>.onAuthGetc s" t �d��| �}� jj�|� dS )z�
Called when the initializer sent the authentication request.
The server checks the credentials and responds with a not-authorized
stanza error.
rR NrS rI r r r rE s z6IQAuthInitializerTests.testFailAuth.<locals>.onAuthSetrU rL r rD r � testFailAuth� s
z#IQAuthInitializerTests.testFailAuth)
r r! r" r, r rO rQ rW rX �
__classcell__r r r4 r r- d s 7=r- c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )�BindInitializerTestsz.
Tests for L{client.BindInitializer}.
c s. t t| ��� t�| j�| _t�d�| j _d S )Nr. )
r0 rZ r r �BindInitializerr r r
r2 r& r r4 r r r 5 s zBindInitializerTests.setUpc sF � fdd�}� fdd�}� � t|�}� j�� }|�|� t�||g�S )zK
Set up a stream, and act as if resource binding succeeds.
c s< t �| d�}|�tdf� |jjddd� � jj�|� d S )Nr7 �bindr
� user@example.com/other resource)Zcontent)r r= r> �NS_BINDr\ r% r( r@ rI r r r �onBind? s �z.BindInitializerTests.testBasic.<locals>.onBindc s � � t�d�� jj� d S )Nr] )r r
r2 r& )r7 r r r �cbF s �z*BindInitializerTests.testBasic.<locals>.cb)r+ � IQ_BIND_SETr �startr' r rK )r r_ r` rM rN r r r � testBasic; s
z BindInitializerTests.testBasicc s> � fdd�}� � t|�}� j�� }� �|tj� t�||g�S )zH
Set up a stream, and act as if resource binding fails.
c s" t �d��| �}� jj�|� d S )NZconflictrS rI r r r r_ T s z0BindInitializerTests.testFailure.<locals>.onBind) r+ ra r rb rV r rT r rK )r r_ rM rN r r r � testFailureP s
z BindInitializerTests.testFailure)r r! r" r, r rc rd rY r r r4 r rZ 0 s rZ c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )�SessionInitializerTestsz1
Tests for L{client.SessionInitializer}.
c s t t| ��� t�| j�| _d S r )r0 re r r �SessionInitializerr r r r4 r r r d s z SessionInitializerTests.setUpc s0 � fdd�}� � t|�}� j�� }t�||g�S )zP
Set up a stream, and act as if session establishment succeeds.
c s t �| d�}� jj�|� d S )Nr7 )r r= r% r( r@ rI r r r � onSessionn s z6SessionInitializerTests.testSuccess.<locals>.onSession)r+ �IQ_SESSION_SETr rb r rK �r rg rM rN r r r � testSuccessi s
z#SessionInitializerTests.testSuccessc s> � fdd�}� � t|�}� j�� }� �|tj� t�||g�S )zM
Set up a stream, and act as if session establishment fails.
c s" t �d��| �}� jj�|� d S )NZ forbiddenrS rI r r r rg { s z6SessionInitializerTests.testFailure.<locals>.onSession) r+ rh r rb rV r rT r rK ri r r r rd w s
z#SessionInitializerTests.testFailure)r r! r" r, r rj rd rY r r r4 r re _ s re c @ s e Zd ZdZdd� ZdS )�BasicAuthenticatorTestszB
Test for both BasicAuthenticator and basicClientFactory.
c C s� t �d�| _t�| jd��d�}| �d|jj� | �| j|jj � | �d|jj � |j
\}}| � |t j
� | � |tj� | �|j� dS )a
Authenticator and stream are properly constructed by the factory.
The L{xmlstream.XmlStream} protocol created by the factory has the new
L{client.BasicAuthenticator} instance in its C{authenticator}
attribute. It is set up with C{jid} and C{password} as passed to the
factory, C{otherHost} taken from the client JID. The stream futher has
two initializers, for TLS and authentication, of which the first has
its C{required} attribute set to C{True}.
r. r/ N� example.com)r
r2 �
client_jidr ZbasicClientFactory�
buildProtocolr r& � otherHostr3 � initializers�assertIsInstancer �TLSInitiatingInitializerr1 � assertFalse�required)r r �tlsZauthr r r �
test_basic� s ��
z"BasicAuthenticatorTests.test_basicN)r r! r" r, rv r r r r rk � s rk c @ s& e Zd ZdZdd� Zdd� Zee_dS )�XMPPAuthenticatorTestsz@
Test for both XMPPAuthenticator and XMPPClientFactory.
c C s� t �d�| _t�| jd��d�}| �d|jj� | �| j|jj � | �d|jj � |j
\}}}}}| � |t j
� | � |t� | � |tj� | � |tj� | �|j� | �|j� | �|j� | �|j� dS )z�
Test basic operations.
Setup an XMPPClientFactory, which sets up an XMPPAuthenticator, and let
it produce a protocol instance. Then inspect the instance variables of
the authenticator and XML stream objects.
r. r/ Nrl )r
r2 rm r �XMPPClientFactoryrn r r& ro r3 rp rq r rr r r[ rf Z
assertTruert rs )r r r ru �saslr\ �sessionr r r rv � s$ �� z!XMPPAuthenticatorTests.test_basicc
s� g � d
� fdd� }t �d�| _t�� }tj| jd|d�}| �tj d|� |�
d�}|j \}}}}} | � |tj � | �
|� d � dS ) zG
A TLS configuration is passed to the TLS initializer.
TNc s � � |� d S r )�append)r r rt �configurationForTLS�Zconfigsr r r � s z:XMPPAuthenticatorTests.test_tlsConfiguration.<locals>.initr. r/ )r| �__init__r )TN)r
r2 rm r ZCertificateOptionsr rx Zpatchr rr rn rp rq ZassertIs)
r r r| �factoryr r ru ry r\ rz r r} r �test_tlsConfiguration� s �
z,XMPPAuthenticatorTests.test_tlsConfigurationN)r r! r" r, rv r� �
skipWhenNoSSL�skipr r r r rw � s # rw )'r, Z
__future__r r Zhashlibr Ztwisted.internetr Ztwisted.python.compatr Z
twisted.trialr Z twisted.words.protocols.jabberr r r
r Z#twisted.words.protocols.jabber.saslr Ztwisted.words.xishr
r � ImportErrorr� rJ r? r^ ra Z
NS_SESSIONrh ZTestCaser �objectr# r- rZ re rk rw r r r r �<module> s8
) M/''