File: //lib/python3/dist-packages/twisted/words/test/__pycache__/test_jabberclient.cpython-310.pyc
o
�b�@ � @ s8 d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z m
Z
m Z m Z ddl
mZ ddlmZ zdd lmZ W n
eyG d
Zd ZY nw d Zd
ZdZdZde ZdZde ZG dd� dej�ZG dd� d�Z G dd� de ej�Z G dd� de ej�Z G dd � d e ej�Z G d d � d ej�Z G d d � d ej�Z!d
S )!z4
Tests for L{twisted.words.protocols.jabber.client}
� )�sha1)�skipIf)�defer)�unittest)�client�error�jid� xmlstream)�SASLInitiatingInitializer)�utility)�sslN)TzSSL not available)F� z//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 �
Authenticator� XmlStreamr �CheckVersionInitializer�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 � assertRaisesr � StreamErrorr � assertEqual� condition)r �excr r r �testNotSupported/ 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 �XmlPipe�pipe�sinkr �ConnectAuthenticator�
authenticatorr r r r r E 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 �Deferred� addCallbackr+ �source�addOnetimeObserver�callback)r �event�handler�dr r r �waitForL s
z$InitiatingInitializerHarness.waitForN)r% r& r' �__doc__r r7 r r r r r( 8 s r( c s@ e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Z� Z S ) �IQAuthInitializerTestsz0
Tests for L{client.IQAuthInitializer}.
c s2 t � �� t�| j�| _t�d�| j_d| j_ d S )N�user@example.com/resource�secret)
�superr r �IQAuthInitializerr r r �JIDr. �passwordr �� __class__r r r d 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 �X 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�usernamer? �resource� r �
toResponse�
addElementrF r7 � IQ_AUTH_SETr+ r1 �send��iq�responser6 �� onAuthSetr r r � onAuthGets �
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; rH rD N) r! �strrF rG r? rH r rJ r+ r1 rM �rO rP r r r rR � s
z7IQAuthInitializerTests.testPlainText.<locals>.onAuthSet)r7 � IQ_AUTH_GETr r r �
gatherResults�r rS �d1�d2r rQ r �
testPlainTextj 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 rB )z�
Called when the initializer sent a query for authentication methods.
The response informs the client that digest authentication is
supported.
rD rE rG �digestrH rI rN rQ r r rS � rT z4IQAuthInitializerTests.testDigest.<locals>.onAuthGetc sb � � dt| jj�� � � td��� t| jj�� � � dt| jj�� t� | d�}� j
j � |� dS )rU rV s 12345secretrH rD N)
r! rW rF rG r � hexdigestr_ rH r rJ r+ r1 rM rX r r r rR � s
z4IQAuthInitializerTests.testDigest.<locals>.onAuthSet�12345)r �sidr7 rY r r r rZ r[ r rQ r �
testDigest� s
z!IQAuthInitializerTests.testDigestc �> � fdd�}� � t|�}� j�� }� �|tj� t�||g�S )zT
Test initializer failure of request for fields for authentication.
c �" 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 � StanzaErrorrJ r+ r1 rM rX r r r rS � s z?IQAuthInitializerTests.testFailRequestFields.<locals>.onAuthGet� r7 rY r r �
assertFailurer rh r rZ r[ r r r �testFailRequestFields� s
z,IQAuthInitializerTests.testFailRequestFieldsc sL � �fdd�}�fdd�� �� t|�}�j�� }��|tj� t�||g�S )z;
Test initializer failure to authenticate.
c rB rC rI rN rQ r r rS � rT z6IQAuthInitializerTests.testFailAuth.<locals>.onAuthGetc re )z�
Called when the initializer sent the authentication request.
The server checks the credentials and responds with a not-authorized
stanza error.
rf Nrg rX r r r rR s z6IQAuthInitializerTests.testFailAuth.<locals>.onAuthSetri r[ r rQ r � testFailAuth� s
z#IQAuthInitializerTests.testFailAuth)
r% r& r' r8 r r^ rc rk rl �
__classcell__r r r@ r r9 _ s 6;r9 c �0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )�BindInitializerTestsz.
Tests for L{client.BindInitializer}.
c s* t � �� t�| j�| _t�d�| j_d S )Nr: ) r< r r �BindInitializerr r r r> r. r r@ r r r + 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 )NrD �bindr � user@example.com/other resource)�content)r rJ rK �NS_BINDrq r+ r1 rM rX r r r �onBind5 s z.BindInitializerTests.testBasic.<locals>.onBindc s � � t�d�� jj� d S )Nrr )r! r r> r. )rD r r r �cb; s �z*BindInitializerTests.testBasic.<locals>.cb)r7 � IQ_BIND_SETr �startr0 r rZ )r ru rv r\ r] r r r � testBasic0 s
z BindInitializerTests.testBasicc rd )zH
Set up a stream, and act as if resource binding fails.
c �" t �d��| �}� jj�|� d S )N�conflictrg rX r r r ru J � z0BindInitializerTests.testFailure.<locals>.onBind) r7 rw r rx rj r rh r rZ )r ru r\ r] r r r � testFailureE �
z BindInitializerTests.testFailure)r% r& r' r8 r ry r} rm r r r@ r ro & s
ro c rn )�SessionInitializerTestsz1
Tests for L{client.SessionInitializer}.
c s t � �� t�| j�| _d S r )r< r r �SessionInitializerr r r r@ r r r Y 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 )NrD )r rJ r+ r1 rM rX r r r � onSessionb s z6SessionInitializerTests.testSuccess.<locals>.onSession)r7 �IQ_SESSION_SETr rx r rZ �r r� r\ r] r r r � testSuccess] s
z#SessionInitializerTests.testSuccessc rd )zM
Set up a stream, and act as if session establishment fails.
c rz )N� forbiddenrg rX r r r r� o r| z6SessionInitializerTests.testFailure.<locals>.onSession) r7 r� r rx rj r rh r rZ r� r r r r} j r~ z#SessionInitializerTests.testFailure)r% r& r' r8 r r� r} rm r r r@ r r T s
r 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 r> �
client_jidr �basicClientFactory�
buildProtocolr! r. � otherHostr? � initializers�assertIsInstancer �TLSInitiatingInitializerr= � assertFalse�required)r r �tls�authr r r �
test_basic~ s
z"BasicAuthenticatorTests.test_basicN)r% r& r' r8 r� r r r r r� y s r� c @ s( e Zd ZdZdd� Zee� dd� �ZdS )�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; Nr� )r r> r� r �XMPPClientFactoryr� r! r. r� r? r� r� r r� r
rp r� �
assertTruer� r� )r r r r� �saslrq �sessionr r r r� � 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 r� �configurationForTLS��configsr r r � s z:XMPPAuthenticatorTests.test_tlsConfiguration.<locals>.initr: r; )r� �__init__r )TN)r r> r� r �CertificateOptionsr r� �patchr r� r� r� r� �assertIs)
r r r� �factoryr r r� r� rq r� r r� r �test_tlsConfiguration� s �
z,XMPPAuthenticatorTests.test_tlsConfigurationN)r% r& r' r8 r� r �
skipWhenNoSSLr� r r r r r� � s
!r� )"r8 �hashlibr r r �twisted.internetr �
twisted.trial� twisted.words.protocols.jabberr r r r �#twisted.words.protocols.jabber.saslr
�twisted.words.xishr r � ImportErrorr� rY rL rt rw �
NS_SESSIONr� �TestCaser r( r9 ro r r� r� r r r r �<module> s: �' H.%%