File: //lib/python3/dist-packages/twisted/words/test/__pycache__/test_jabberxmlstream.cpython-310.pyc
o
�b'� � @ s( d Z ddlmZ ddlmZ ddlmZmZ ddlm Z ddl
m Z ddl m
Z
ddlmZ dd lmZ dd
lmZmZmZmZ dd lmZ dd lmZ zdd
lmZ W n
e ya dZdZ Y nw eZdZ ddl m Z ddl m!Z! dZ"G dd� dej#�Z$G dd� dej#�Z%G dd� dej#�Z&G dd� de'�Z(G d d � d ej#�Z)G d d � d ej#�Z*G d d!� d!ej#�Z+G d"d#� d#ej#�Z,G d$d%� d%ej-�Z.G d&d'� d'ej#�Z/G d(d)� d)ej#�Z0G d*d+� d+�Z1G d,d-� d-ej2�Z3G d.d/� d/ej2�Z4G d0d1� d1ej#�Z5G d2d3� d3ej#�Z6G d4d5� d5ej#�Z7G d6d7� d7e�Z8dS )8z8
Tests for L{twisted.words.protocols.jabber.xmlstream}.
� )�skipIf)� verifyObject)�defer�task)�ConnectionLost)�IProtocolFactory)�failure)�
proto_helpers)�unittest)�error�ijabber�jid� xmlstream)�!GenericXmlStreamFactoryTestsMixin)�domish)�sslN)TzSSL not available)F� )�ClientTLSOptions)�CertificateOptions� urn:ietf:params:xml:ns:xmpp-tlsc @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �HashPasswordTestsz.
Tests for L{xmlstream.hashPassword}.
c C � t �dd�}| �d|� dS )zS
The sid and secret are concatenated to calculate sha1 hex digest.
�12345�secret�(99567ee91b2c7cabf607f10cb9f4a3634fa820e0N�r � hashPassword� assertEqual��self�hash� r" �I/usr/lib/python3/dist-packages/twisted/words/test/test_jabberxmlstream.py�
test_basic) � z HashPasswordTests.test_basicc C � | � ttjdd� dS )zB
The session identifier must be a unicode object.
s ¹2345r N�� assertRaises� TypeErrorr r �r r" r" r# �test_sidNotUnicode0 � z$HashPasswordTests.test_sidNotUnicodec C r& )z8
The password must be a unicode object.
r s secrétNr' r* r" r" r# �test_passwordNotUnicode6 r, z)HashPasswordTests.test_passwordNotUnicodec C r )z\
The concatenated sid and password must be encoded to UTF-8 before hashing.
r u secrét�(659bf88d8f8e179081f7f3b4a8e7d224652d2853Nr r r" r" r# �test_unicodeSecret< r% z$HashPasswordTests.test_unicodeSecretN)�__name__�
__module__� __qualname__�__doc__r$ r+ r- r/ r" r" r" r# r $ s r c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�IQTestszG
Tests both IQ and the associated IIQResponseTracker callback.
c C sb t �d�}d|_t �|�| _ t�� | _| jj| j _| j � t
� � � | j � d� t �
| j d�| _d S )N� otherhost�testnszm<stream:stream xmlns:stream='http://etherx.jabber.org/streams' xmlns='testns' from='otherhost' version='1.0'>�get)r �ConnectAuthenticator� namespace� XmlStreamr �Clock�clock� callLater�
_callLater�makeConnectionr �StringTransport� dataReceived�IQ�iq)r �
authenticatorr" r" r# �setUpI s
�z
IQTests.setUpc C s& | � | jd d� | �| jd � d S )N�typer7 �id)r rC �
assertTruer* r" r" r# � testBasicV s zIQTests.testBasicc C sR | j j�� | j�� | jd �d�}| �| j j�� d| d d| d g� d S )NrG zutf-8s <iq type='get' id='s '/>s <iq id='s ' type='get'/>)r � transport�clearrC �send�encode�assertIn�value)r �idBytesr" r" r# �testSendZ s
��zIQTests.testSendc s> � fdd�}� j �� }|�|� � j}|�d� j d � |S )Nc s � � | d d� d S )NrF �result)r )rR r* r" r# �cbg s z&IQTests.testResultResponse.<locals>.cb�<iq type='result' id='%s'/>rG )rC rL � addCallbackr rA )r rS �d�xsr" r* r# �testResultResponsef s
zIQTests.testResultResponsec C s6 | j �� }| �|tj� | j}|�d| j d � |S )Nz<iq type='error' id='%s'/>rG )rC rL �
assertFailurer � StanzaErrorr rA �r rV rW r" r" r# �testErrorResponseq s
zIQTests.testErrorResponsec sD � j }t �|� � �|j� � fdd�}|�d|d� |�d� dS )z�
Test that untracked iq responses don't trigger any action.
Untracked means that the id of the incoming response iq is not
in the stream's C{iqDeferreds} dictionary.
c s � � t| dd�� d S )N�handledF)� assertFalse�getattr)rC r* r" r# rS � s z*IQTests.testNonTrackedResponse.<locals>.cbz/iq���z <iq type='result' id='test'/>N)r � upgradeWithIQResponseTrackerr^ � iqDeferreds� addObserverrA )r rW rS r" r* r# �testNonTrackedResponsey s
z IQTests.testNonTrackedResponsec C s< | j �� }| j}|�d| j d � | �| j d |j� |S )z�
Test if the deferred associated with an iq request is removed
from the list kept in the L{XmlStream} object after it has
been fired.
rT rG )rC rL r rA � assertNotInrb r[ r" r" r# � testCleanup� s
zIQTests.testCleanupc C s* | j �� }| j}|�d� | �|t� |S )z�
Test if deferreds for iq's that haven't yet received a response
have their errback called on stream disconnect.
�Closed by peer)rC rL r �connectionLostrY r r[ r" r" r# �testDisconnectCleanup� s
z IQTests.testDisconnectCleanupc s2 � �fdd�� �j �� }|�� � �j�d� |S )z~
Test to make sure the errbacks cannot cause the iteration of the
iqDeferreds to blow up in our face.
c s t ��j ��� }|�� � d S �N)r rB rL �
addErrback)r rV ��ebr r" r# rm � s z'IQTests.testNoModifyingDict.<locals>.ebrg )rC rL rk r rh �r rV r" rl r# �testNoModifyingDict� s
zIQTests.testNoModifyingDictc C sP d| j _| j �� }| �|tj� | j�ddg� | �| jj � | �| jj
� |S )zK
Test that an iq request with a defined timeout times out.
�< � ) rC �timeoutrL rY r � TimeoutErrorr< �pumpr^ �callsrb rn r" r" r# �testRequestTimingOut� s
z IQTests.testRequestTimingOutc C sR d| j _| j �� }| j�d| jjd| j d � | j�ddg� | �| jj � |S )z�
Test that an iq request with a defined timeout does not time out
when a response was received before the timeout period elapsed.
rp rq rT rG )
rC rr rL r< r= r rA rt r^ ru rn r" r" r# �testRequestNotTimingOut� s
�z IQTests.testRequestNotTimingOutc C s@ d| j _| j �� }| j}|�d� | �|t� | �| jj � |S )z|
Test if timeouts for iq's that haven't yet received a response
are cancelled on stream disconnect.
rp rg )
rC rr rL r rh rY r r^ r< ru r[ r" r" r# �!testDisconnectTimeoutCancellation� s
z)IQTests.testDisconnectTimeoutCancellationN)r0 r1 r2 r3 rE rI rQ rX r\ rd rf ri ro rv rw rx r" r" r" r# r4 D s
r4 c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!S )"�XmlStreamTestsc C �
d| _ d S �NT)�gotStreamStart�r �objr" r" r# �
onStreamStart� �
z XmlStreamTests.onStreamStartc C rz r{ )� gotStreamEndr} r" r" r# � onStreamEnd� r� zXmlStreamTests.onStreamEndc C rz r{ )�gotStreamErrorr} r" r" r# �
onStreamError� r� z XmlStreamTests.onStreamErrorc C sv d| _ d| _d| _t�t�� �}|�d| j� |�d| j� |�d| j � |�
t � � � ||j
_d|_d|_|| _dS )z9
Set up XmlStream and several observers.
Fz//event/stream/start�//event/stream/endz//event/stream/errorr6 �rq r N)r| r� r� r r: �
Authenticatorrc r r� r� r? r � StringTransportWithDisconnectionrJ �protocolr9 �version�r rW r" r" r# rE � s
zXmlStreamTests.setUpc C sh | j }|�� | j j�� dd� �d�}| �d|� | �d|� | �d|� | �d|� | �|j� dS ) z>
Basic test on the header sent by sendHeader.
r r` � � <stream:stream�/ xmlns:stream='http://etherx.jabber.org/streams'� xmlns='testns'�
version='1.0'N)r �
sendHeaderrJ rO �splitrN rH � _headerSent�r rW � splitHeaderr" r" r# �test_sendHeaderBasic s z#XmlStreamTests.test_sendHeaderBasicc C s~ | j }d|jd< |�� | j j�� dd� �d�}| �d|� | �d|� | �d|� | �d |� | �d
|� | �|j� d S ) z=
Test for additional namespace declarations.
�dbzjabber:server:dialbackr r` r� r� r� s! xmlns:db='jabber:server:dialback'r� r� N) r �prefixesr� rJ rO r� rN rH r� r� r" r" r# �#test_sendHeaderAdditionalNamespaces s
z2XmlStreamTests.test_sendHeaderAdditionalNamespacesc C s` | j }t�d�|_t�d�|_d|_|�� |j�� dd� � d�}| �
d|� | �
d|� d S )
z;
Test addressing when initiating a stream.
�thisHost� otherHostTr r` r� � to='otherhost'� from='thishost'N) r r
�JID�
thisEntity� otherEntity�
initiatingr� rJ rO r� rN r� r" r" r# �test_sendHeaderInitiating s z(XmlStreamTests.test_sendHeaderInitiatingc C sr | j }t�d�|_t�d�|_d|_d|_|�� |j� � dd� �
d�}| � d|� | � d |� | � d
|� d S ) z:
Test addressing when receiving a stream.
r� r� F� session01r r` r� r� r� s id='session01'N) r r
r� r� r� r� �sidr� rJ rO r� rN r� r" r" r# �test_sendHeaderReceiving* s z'XmlStreamTests.test_sendHeaderReceivingc C s6 | j }|�d� |�d� | �| j� | �| j� dS )z>
Test events when a stream error is received.
��<stream:stream xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' from='example.com' id='12345' version='1.0'>z<stream:error/>N)r rA rH r� r� r� r" r" r# �test_receiveStreamError9 s �
z&XmlStreamTests.test_receiveStreamErrorc C �P | j }d|_|�� |j�� |�t�d�� | �d|j� � � | �
| j � dS )z�
Test sendStreamError on an initiating xmlstream with a header sent.
An error should be sent out and the connection lost.
T�version-unsupported� N� r r� r� rJ rK �sendStreamErrorr � StreamError�assertNotEqualrO rH r� r� r" r" r# � test_sendStreamErrorInitiatingG �
z-XmlStreamTests.test_sendStreamErrorInitiatingc C sT | j }d|_|j�� |�t�d�� | �|j� | � d|j�
� � | � | j � dS )z�
Test sendStreamError on an initiating xmlstream without having sent a
header.
In this case, no header should be generated. Also, the error should
not be sent out on the stream. Just closing the connection.
Tr� r� N)
r r� rJ rK r� r r� � assertNotr� r rO rH r� r� r" r" r# �&test_sendStreamErrorInitiatingNoHeaderU �
z5XmlStreamTests.test_sendStreamErrorInitiatingNoHeaderc C r� )z�
Test sendStreamError on a receiving xmlstream with a header sent.
An error should be sent out and the connection lost.
Fr� r� Nr� r� r" r" r# � test_sendStreamErrorReceivinge r� z,XmlStreamTests.test_sendStreamErrorReceivingc C sT | j }d|_|j�� |�t�d�� | �|j� | � d|j�
� � | �| j � dS )z�
Test sendStreamError on a receiving xmlstream without having sent a
header.
In this case, a header should be generated. Then, the error should
be sent out on the stream followed by closing the connection.
Fr� r� N) r r� rJ rK r� r r� rH r� r� rO r� r� r" r" r# �%test_sendStreamErrorReceivingNoHeaders r� z4XmlStreamTests.test_sendStreamErrorReceivingNoHeaderc C s: | j }|�� |j}|�� | �||j� | �|j� dS )zE
Test resetting the XML stream to start a new layer.
N)r r� �stream�resetr� r� r� )r rW r� r" r" r# �
test_reset� s zXmlStreamTests.test_resetc C s� | j }|�d� | �|j�� d� |j�� t�d�}|�|� | �|j�� d� |j�� t�d�}|�|� | �|j�� d� dS )z:
Test send with various types of objects.
� <presence/>� <presence/>)r6 �presence)z http://etherx.jabber.org/streams�featuress <stream:features/>N)r rL r rJ rO rK r �Element)r rW �elr" r" r# � test_send� s
zXmlStreamTests.test_sendc s� g �g �g � G � ��fdd�d�}|� }t �|�}| �|g� � |�� | �dg�� |�d� | �dt��� |�� | �dg�� dS )zM
Test that the associated authenticator is correctly called.
c s0 e Zd Z�fdd�Z�fdd�Z� fdd�ZdS )z<XmlStreamTests.test_authenticator.<locals>.TestAuthenticatorc s � � d � d S rj ��appendr* )�connectionMadeCallsr" r# �connectionMade� � zKXmlStreamTests.test_authenticator.<locals>.TestAuthenticator.connectionMadec � � � |� d S rj r� )r � rootElement)�streamStartedCallsr" r# �
streamStarted� r� zJXmlStreamTests.test_authenticator.<locals>.TestAuthenticator.streamStartedc r� rj r� r� )�associateWithStreamCallsr" r# �associateWithStream� r� zPXmlStreamTests.test_authenticator.<locals>.TestAuthenticator.associateWithStreamN)r0 r1 r2 r� r� r� r" �r� r� r� r" r# �TestAuthenticator� s r� N�s<stream:stream xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' from='example.com' id='12345'>rq )r r: r r� rA �lenr� )r r� �arW r" r� r# �test_authenticator� s
�z!XmlStreamTests.test_authenticatorN)r0 r1 r2 r r� r� rE r� r� r� r� r� r� r� r� r� r� r� r� r" r" r" r# ry � s"
ry c @ s e Zd ZdS )� TestErrorN)r0 r1 r2 r" r" r" r# r� � s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �AuthenticatorTestsc C � t �� | _t �| j�| _ d S rj )r r� rD r: r* r" r" r# rE � �
zAuthenticatorTests.setUpc C sh | j }|�t�� � |�d� | �d|j� | �d|j� | �d|j � | �d|j
� | �d|j � dS )�e
Test streamStart to fill the appropriate attributes from the
stream header.
��<stream:stream xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' from='example.org' to='example.com' id='12345' version='1.0'>r� N�invalid) r r? r r@ rA r r� �assertIdenticalr� r9 r� r� r� r" r" r# �test_streamStart� s �z#AuthenticatorTests.test_streamStartc C �0 | j }|�t�� � |�d� | �d|j� dS )z
Test streamStart to fill the appropriate attributes from the
stream header for a pre-XMPP-1.0 header.
r� �r r N�r r? r r@ rA r r� r� r" r" r# �test_streamStartLegacy� � �z)AuthenticatorTests.test_streamStartLegacyc C r� )z�
Test streamStart to fill the appropriate attributes from the
stream header for a version with only one digit.
z<stream:stream xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' from='example.com' id='12345' version='1'>r� Nr� r� r" r" r# � test_streamBadVersionOneDigit� r� z0AuthenticatorTests.test_streamBadVersionOneDigitc C r� )z}
Test streamStart to fill the appropriate attributes from the
stream header for a malformed version.
z�<stream:stream xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' from='example.com' id='12345' version='blah'>r� Nr� r� r" r" r# � test_streamBadVersionNoNumber� r� z0AuthenticatorTests.test_streamBadVersionNoNumberN)r0 r1 r2 rE r� r� r� r� r" r" r" r# r� � s r� c @ �< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�ConnectAuthenticatorTestsc C sJ d| _ d | _t�d�| _t�| j�| _| j�d| j� | j�d| j� d S )NFr� �//event/stream/authd�//event/xmpp/initfailed) �gotAuthenticated� initFailurer r8 rD r: rc �onAuthenticated� onInitFailedr* r" r" r# rE s z ConnectAuthenticatorTests.setUpc C rz r{ )r� r} r" r" r# r� r� z)ConnectAuthenticatorTests.onAuthenticatedc C s
|| _ d S rj )r� )r r r" r" r# r� r� z&ConnectAuthenticatorTests.onInitFailedc C sH G dd� d�}|� }|g| j _| j�� | �g | j j� | �| j� dS )zG
Test successful completion of an initialization step.
c @ � e Zd Zdd� ZdS )z9ConnectAuthenticatorTests.testSucces.<locals>.Initializerc S s d S rj r" r* r" r" r# �
initialize � zDConnectAuthenticatorTests.testSucces.<locals>.Initializer.initializeN�r0 r1 r2 r� r" r" r" r# � Initializer � r� N)r � initializersrD �initializeStreamr rH r� �r r� �initr" r" r# �
testSucces s
z$ConnectAuthenticatorTests.testSuccesc C sj G dd� d�}|� }|g| j _| j�� | �|g| j j� | �| j� | �d| j� | � | j�
t �� dS )z9
Test failure of an initialization step.
c @ r� )z:ConnectAuthenticatorTests.testFailure.<locals>.Initializerc S s t �rj )r� r* r" r" r# r� - r� zEConnectAuthenticatorTests.testFailure.<locals>.Initializer.initializeNr� r" r" r" r# r� , r� r� N) r r� rD r� r r^ r� �assertNotIdenticalr� rH �checkr� r� r" r" r# � testFailure' s
z%ConnectAuthenticatorTests.testFailurec C s� d| j _| j}|�t�� � |�d� | �d|j� | �d|j � | �d|j� | �d|j
j � | � d|j
� | �| j� |�d� | �d|j� | �| j� dS ) r� r6 z�<stream:stream xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' from='example.com' to='example.org' id='12345' version='1.0'>r� r � example.comNz9<stream:features><test xmlns='testns'/></stream:features>�r6 �test)rD r9 r r? r r@ rA r r� r� r� �hostr� r� r� r� rN r� rH r� r" r" r# r� 9 s" � �z*ConnectAuthenticatorTests.test_streamStartN) r0 r1 r2 rE r� r� r� r� r� r" r" r" r# r� s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �ListenAuthenticatorTestsz4
Tests for L{xmlstream.ListenAuthenticator}
c C r� rj )r �ListenAuthenticatorrD r: r* r" r" r# rE Y r� z ListenAuthenticatorTests.setUpc C s� | j }|�t�� � | �d|j� |�d� | �d|j� | � d|j� | �
d|j� | �d|j � | �d|j � | �d|j
j� dS )r� Nr� r� r �
jabber:clientr� )r r? r r@ r� r� rA r r� r� r� r9 r� r� r� r� r" r" r# r� ] s �z)ListenAuthenticatorTests.test_streamStartc C s0 | j }|�t�� � |�d� | �|jt� dS )zD
The generated session id must be a unicode object.
r� N)r r? r r@ rA �assertIsInstancer� �strr� r" r" r# � test_streamStartUnicodeSessionIDr s �z9ListenAuthenticatorTests.test_streamStartUnicodeSessionIDN)r0 r1 r2 r3 rE r� r r" r" r" r# r� T s
r� c @ s| e Zd Zdd� Zdd� Zdd� Zee� dd� �Zee� d d
� �Z ee� d d � �Z
d
d� Z dd� Z dd� Z
dd� Zdd� ZdS )� TLSInitiatingInitializerTestsc C sb g | _ g | _tj| _t�d�| _t�| j�| _| j j| j_ | j�
� | j� d� t� | j�| _
d S )Nr� r� )�output�doner r �savedSSLr8 rD r: r� rL r� rA �TLSInitiatingInitializerr� r* r" r" r# rE � s
�z#TLSInitiatingInitializerTests.setUpc C s | j t_d S rj )r r r r* r" r" r# �tearDown� � z&TLSInitiatingInitializerTests.tearDownc C s$ t j| j dd�| _| �| jj� dS )z>
Passing required sets the instance variable.
T)�requiredN)r r r� rH r r* r" r" r# �test_initRequired� s z/TLSInitiatingInitializerTests.test_initRequiredc s� t �� � j_� fdd�� jj_� fdd�� j_� fdd�� j_� j�� }|� � j
tj � � �
dt � j
�� � j
d }� �
d|j� � �
t|j� � j�dt � � �
g d �� j� |S )
zN
When TLS is wanted and SSL available, StartTLS is initiated.
c � � j �d�S �N�TLS�r r� )�ctxr* r" r# �<lambda>� � zDTLSInitiatingInitializerTests.test_wantedSupported.<locals>.<lambda>c r
�Nr� r
r" r* r" r# r � r c r
�N�headerr
r" r* r" r# r � r � rq �starttls�<proceed xmlns='%s'/>�r r� r )r r@ r rJ �startTLSr� r� r� �startrU r �Resetr� r �name� NS_XMPP_TLS�urirA r )r rV r r" r* r# �test_wantedSupported� s
z2TLSInitiatingInitializerTests.test_wantedSupportedc sr � fdd�}t �� � j_|� jj_� fdd�� j_� fdd�� j_� j�� }� j� dt
� � � g d�� j � |S )z:
The server certificate will be verified.
c s* � � | t� � �| jd� � j�d� d S )Nr� r )r� r r � _hostnamer r� ��contextFactoryr* r" r# � fakeStartTLS� s zJTLSInitiatingInitializerTests.test_certificateVerify.<locals>.fakeStartTLSc r
r r
r" r* r" r# r � r zFTLSInitiatingInitializerTests.test_certificateVerify.<locals>.<lambda>c r
r r
r" r* r" r# r � r r r )
r r@ r rJ r r� r� r� r rA r r r �r r" rV r" r* r# �test_certificateVerify� s
z4TLSInitiatingInitializerTests.test_certificateVerifyc s� t � � tj�j� d��_� �j_� �fdd�}t�� �j_|�jj_�fdd��j_ �fdd��j_
�j� � }�j� dt
� ��g d��j� |S ) zH
A custom contextFactory is passed through to startTLS.
)�configurationForTLSc s �� � | � �j�d� d S r )�assertIsr r� r �r r r" r# r" � s zQTLSInitiatingInitializerTests.test_certificateVerifyContext.<locals>.fakeStartTLSc r
r r
r" r* r" r# r � r zMTLSInitiatingInitializerTests.test_certificateVerifyContext.<locals>.<lambda>c r
r r
r" r* r" r# r � r r r )r r r r� r! r r@ rJ r r� r� r rA r r r r# r" r' r# � test_certificateVerifyContext� s �
z;TLSInitiatingInitializerTests.test_certificateVerifyContextc C s< dt _d| j_| j�� }|�| jd� | �dt| j�� |S )zX
No StartTLS is initiated when wanted, not required, SSL not available.
NFrq ) r r r� r r rU r r� r rn r" r" r# �"test_wantedNotSupportedNotRequired� �
z@TLSInitiatingInitializerTests.test_wantedNotSupportedNotRequiredc C s< dt _d| j_| j�� }| �|t j� | �dt| j �� |S )zS
TLSNotSupported is raised when TLS is required but not available.
NTrq )
r r r� r r rY �TLSNotSupportedr r� r rn r" r" r# � test_wantedNotSupportedRequired� r* z=TLSInitiatingInitializerTests.test_wantedNotSupportedRequiredc C s^ t �d�}|�d� |j|jf|i| j_d| j_| j� � }| �
dt | j �� | �
|tj� |S )zW
TLSRequired is raised when TLS is not wanted, but required by server.
�r r r Frq )r r� �
addElementr r r r� r� �wantedr r r� r rY � TLSRequired�r �tlsrV r" r" r# �test_notWantedRequired� s
z4TLSInitiatingInitializerTests.test_notWantedRequiredc C s\ t �d�}|j|jf|i| j_d| j_d| j_| j� � }|�
| j d� | � dt | j
�� |S )zL
No StartTLS is initiated when not wanted and not required.
r- FNrq )r r� r r r r� r� r/ r r rU r r� r r1 r" r" r# �test_notWantedNotRequired
s
z7TLSInitiatingInitializerTests.test_notWantedNotRequiredc C s2 dt _| j�� }| �|t j� | j �dt � |S )zN
TLSFailed is raised when the server responds with a failure.
rq z<failure xmlns='%s'/>)r r r� r rY � TLSFailedrA r rn r" r" r# � test_failed s
z)TLSInitiatingInitializerTests.test_failedN)r0 r1 r2 rE r r r �
skipWhenNoSSLr r$ r( r) r, r3 r4 r6 r" r" r" r# r � s
r c @ s e Zd ZdZdd� ZdS )�TestFeatureInitializerr� c C s
t �d �S rj )r �succeedr* r" r" r# r * r� z TestFeatureInitializer.startN)r0 r1 r2 �featurer r" r" r" r# r8 ' s r8 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�%BaseFeatureInitiatingInitializerTestsc C s t �t �� �| _ t| j �| _d S rj )r r: r� r8 r� r* r" r" r# rE / s z+BaseFeatureInitiatingInitializerTests.setUpc C s$ | j jt�| j j�i| j_| j �� S )zW
Test that an advertized feature results in successful initialization.
)r� r: r r� r r� r� r* r" r" r# �testAdvertized3 s
z4BaseFeatureInitiatingInitializerTests.testAdvertizedc C s d| j _| �tj| j j� dS )z�
Test that when the feature is not advertized, but required by the
initializer, an exception is raised.
TN)r� r r( r �FeatureNotAdvertizedr� r* r" r" r# �testNotAdvertizedRequired: � z?BaseFeatureInitiatingInitializerTests.testNotAdvertizedRequiredc C s d| j _| �d| j �� � dS )z�
Test that when the feature is not advertized, and not required by the
initializer, the initializer silently succeeds.
FN)r� r r� r� r* r" r" r# � testNotAdvertizedNotRequiredB r? zBBaseFeatureInitiatingInitializerTests.testNotAdvertizedNotRequiredN)r0 r1 r2 rE r< r> r@ r" r" r" r# r; . s
r; c @ r� )�ToResponseTestsc C s� t �d�}d|d< d|d< d|d< d|d < t�|d
�}| �||� | �|d d� | �|d d� | �|d d
� | �|d d� d S ) zS
Test that a response stanza is generated with addressing swapped.
�r� rC r7 rF �user1@example.com�to�user2@example.com/resource�from�stanza1rG rR N)r r� r �
toResponser� r �r �stanza�responser" r" r# �test_toResponseL s
z ToResponseTests.test_toResponsec C sH t �d�}d|d< d|d< t�|�}| �|d d� | �|�d�� dS )zY
Test that a response is generated from a stanza without a from address.
rB r7 rF rC rD rF N)r r� r rH r r^ � hasAttributerI r" r" r# �test_toResponseNoFrom\ �
z%ToResponseTests.test_toResponseNoFromc C sH t �d�}d|d< d|d< t�|�}| �|�d�� | �|d d� dS )zW
Test that a response is generated from a stanza without a to address.
rB r7 rF rE rF rD N)r r� r rH r^ rM r rI r" r" r# �test_toResponseNoTog rO z#ToResponseTests.test_toResponseNoToc C s@ t �d�}d|d< t�|�}| �|�d�� | �|�d�� dS )zY
Test that a response is generated from a stanza without any addressing.
�r� �message�chatrF rD rF N�r r� r rH r^ rM rI r" r" r# �test_toResponseNoAddressingr s
z+ToResponseTests.test_toResponseNoAddressingc C �( t �d�}t�|�}| �|�d�� dS )zP
Test that a proper response is generated without id attribute.
rQ rG NrT rI r" r" r# � test_noID| �
zToResponseTests.test_noIDc C rV )zR
Test that a proper response is generated without type attribute.
rQ rF NrT rI r" r" r# � test_noType� rX zToResponseTests.test_noTypeN) r0 r1 r2 rL rN rP rU rW rY r" r" r" r# rA K s
rA c @ � e Zd ZdZdd� Zdd� ZdS )� DummyFactoryzJ
Dummy XmlStream factory that only registers bootstrap observers.
c C s
i | _ d S rj �� callbacksr* r" r" r# �__init__� r� zDummyFactory.__init__c C s || j |< d S rj r\ )r �event�callbackr" r" r# � addBootstrap� r� zDummyFactory.addBootstrapN)r0 r1 r2 r3 r^ ra r" r" r" r# r[ � � r[ c @ �8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�DummyXMPPHandlerzO
Dummy XMPP subprotocol handler to count the methods are called on it.
c C s d| _ d| _d| _d S )Nr )�doneMade�doneInitialized�doneLostr* r" r" r# r^ � s
zDummyXMPPHandler.__init__c C s | � � d S rj )r� r� r" r" r# r? � r z DummyXMPPHandler.makeConnectionc C � | j d7 _ d S �Nrq �re r* r" r" r# r� � � z DummyXMPPHandler.connectionMadec C rh ri )rf r* r" r" r# �connectionInitialized� rk z&DummyXMPPHandler.connectionInitializedc C rh ri )rg �r �reasonr" r" r# rh � rk z DummyXMPPHandler.connectionLostN) r0 r1 r2 r3 r^ r? r� rl rh r" r" r" r# rd � s rd c @ rZ )�FailureReasonXMPPHandlerz>
Dummy handler specifically for failure Reason tests.
c C rz )NF)�gotFailureReasonr* r" r" r# r^ � r� z!FailureReasonXMPPHandler.__init__c C s t |tj�r d| _d S d S r{ )�
isinstancer �Failurerp rm r" r" r# rh � s
�z'FailureReasonXMPPHandler.connectionLostN)r0 r1 r2 r3 r^ rh r" r" r" r# ro � rb ro c @ r ) �XMPPHandlerTestsz-
Tests for L{xmlstream.XMPPHandler}.
c C s t tjt�� � dS )zN
L{xmlstream.XMPPHandler} implements L{ijabber.IXMPPHandler}.
N)r r � IXMPPHandlerr � XMPPHandlerr* r" r" r# �test_interface� s z XMPPHandlerTests.test_interfacec C s> G dd� d�}t �� }|� |_|�d� | �dg|jj� dS )zP
Test that data is passed on for sending by the stream manager.
c @ � e Zd Zdd� Zdd� ZdS )z6XMPPHandlerTests.test_send.<locals>.DummyStreamManagerc S �
g | _ d S rj )�outlistr* r" r" r# r^ � r� z?XMPPHandlerTests.test_send.<locals>.DummyStreamManager.__init__c S � | j �|� d S rj )ry r� )r �datar" r" r# rL � � z;XMPPHandlerTests.test_send.<locals>.DummyStreamManager.sendN)r0 r1 r2 r^ rL r" r" r" r# �DummyStreamManager� � r} r� N)r ru �parentrL r ry )r r} �handlerr" r" r# r� � s
zXMPPHandlerTests.test_sendc C sN G dd� dt j�}|� }t �t �� �}|�|� | �|j� | �||j � dS )zY
Test that makeConnection saves the XML stream and calls connectionMade.
c @ r� )z=XMPPHandlerTests.test_makeConnection.<locals>.TestXMPPHandlerc S rz r{ rj r* r" r" r# r� � r� zLXMPPHandlerTests.test_makeConnection.<locals>.TestXMPPHandler.connectionMadeN)r0 r1 r2 r� r" r" r" r# �TestXMPPHandler� r� r� N)r ru r: r� r? rH re r� )r r� r� rW r" r" r# �test_makeConnection� s
z$XMPPHandlerTests.test_makeConnectionc C s> t �� }t �t �� �}|�|� |�t� � | �d|j � dS )zB
Test that connectionLost forgets the XML stream.
N)r ru r: r� r? rh � Exceptionr� )r r� rW r" r" r# �test_connectionLost� s
z$XMPPHandlerTests.test_connectionLostN)r0 r1 r2 r3 rv r� r� r� r" r" r" r# rs � s rs c @ r ) �XMPPHandlerCollectionTestsz7
Tests for L{xmlstream.XMPPHandlerCollection}.
c C s t �� | _d S rj )r �XMPPHandlerCollection�
collectionr* r" r" r# rE � r� z XMPPHandlerCollectionTests.setUpc C s t tj| j� dS )zZ
L{xmlstream.StreamManager} implements L{ijabber.IXMPPHandlerCollection}.
N)r r �IXMPPHandlerCollectionr� r* r" r" r# rv � s z)XMPPHandlerCollectionTests.test_interfacec C s4 t � }|�| j� | �|| j� | �| j|j� dS )z:
Test the addition of a protocol handler.
N)rd �setHandlerParentr� rN r� r �r r� r" r" r# �test_addHandler s z*XMPPHandlerCollectionTests.test_addHandlerc C s> t � }|�| j� |�| j� | �|| j� | �d|j� dS )z5
Test removal of a protocol handler.
N)rd r� r� �disownHandlerParentre r� r r� r" r" r# �test_removeHandler
s
z-XMPPHandlerCollectionTests.test_removeHandlerN)r0 r1 r2 r3 rE rv r� r� r" r" r" r# r� � s r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd S ) �StreamManagerTestsz/
Tests for L{xmlstream.StreamManager}.
c C s t � }t�|�| _d S rj )r[ r �
StreamManager�
streamManager)r �factoryr" r" r# rE s zStreamManagerTests.setUpc C s~ | j }| �d|j� | �g |j� | �|j|jjd � | �|j|jjd � | �|j |jjd � | �|j
|jjd � dS )zM
Test correct initialization and setup of factory observers.
Nz//event/stream/connectedr� r� r� ) r� r� r r �handlers�
_connectedr� r] �_authd�
_disconnected�initializationFailed)r �smr" r" r# r$ s ��z StreamManagerTests.test_basicc C s\ | j }t� }|�|� t�t�� �}|�|� | �d|j� | �d|j � | �d|j
� dS )z
Test that protocol handlers have their connectionMade method called
when the XML stream is connected.
rq r N) r� rd r� r r: r� r� r re rf rg �r r� r� rW r" r" r# �test_connected. �
z!StreamManagerTests.test_connectedc C sN | j }t� }|�|� t�t�� �}|�|� | �d|j� | �d|j � dS )zP
Test raw data functions unset when logTraffic is set to False.
N)
r� rd r� r r: r� r� r� � rawDataInFn� rawDataOutFnr� r" r" r# � test_connectedLogTrafficFalse< s
z0StreamManagerTests.test_connectedLogTrafficFalsec C sT | j }d|_t� }|�|� t�t�� �}|�|� | �d|j � | �d|j
� dS )zM
Test raw data functions set when logTraffic is set to True.
TN) r� �
logTrafficrd r� r r: r� r� r� r� r� r� r" r" r# � test_connectedLogTrafficTrueH s
z/StreamManagerTests.test_connectedLogTrafficTruec C s\ | j }t� }|�|� t�t�� �}|�|� | �d|j� | �d|j � | �d|j
� dS )z�
Test that protocol handlers have their connectionInitialized method
called when the XML stream is initialized.
r rq N) r� rd r� r r: r� r� r re rf rg r� r" r" r# �
test_authdU r� z StreamManagerTests.test_authdc C s\ | j }t� }|�|� t�t�� �}|�|� | �d|j� | �d|j � | �d|j
� dS )z�
Test that protocol handlers have their connectionLost method
called when the XML stream is disconnected.
r rq N) r� rd r� r r: r� r� r re rf rg r� r" r" r# �test_disconnectedc r� z$StreamManagerTests.test_disconnectedc C s< | j }t� }|�|� |�t�td��� | �d|j� dS )z�
A L{STREAM_END_EVENT} results in L{StreamManager} firing the handlers
L{connectionLost} methods, passing a L{failure.Failure} reason.
z no reasonTN) r� ro r� r� r rr r� r rp �r r� r� r" r" r# �test_disconnectedReasonq s
z*StreamManagerTests.test_disconnectedReasonc C sD | j }t� }|�|� | �d|j� | �d|j� | �d|j� dS )zN
Test the addition of a protocol handler while not connected.
r N)r� rd r� r re rf rg r� r" r" r# r� | s
z"StreamManagerTests.test_addHandlerc C sf | j }t�t�� �}|�|� |�|� t� }|�|� | �d|j � | �d|j
� | �d|j � dS )a
Test the addition of a protocol handler after the stream
have been initialized.
Make sure that the handler will have the connected stream
passed via C{makeConnection} and have C{connectionInitialized}
called.
rq r N) r� r r: r� r� r� rd r� r re rf rg )r r� rW r� r" r" r# �test_addHandlerInitialized� s
z-StreamManagerTests.test_addHandlerInitializedc C sj t �t �� �}t �|�}|�d�}t�� |_|�� |� d� |�
|d� |� d� | � d|j�
� � dS )z�
Test send when the stream has been initialized.
The data should be sent directly over the XML stream.
Nr� r� r� r� )r �XmlStreamFactoryr� r� �
buildProtocolr r@ rJ r� rA �dispatchrL r rO �r r� r� rW r" r" r# �test_sendInitialized� s
�
z'StreamManagerTests.test_sendInitializedc C s� t �t �� �}t �|�}t� }|�|� |�d�}t�� |_ |�
d� | � d|j � � � | � d|j
d � |�� | � d|j � � � | � d|j
d � |�d� |�|d� | � d|j � � � | �|j
� dS )z�
Test send when there is no established XML stream.
The data should be cached until an XML stream has been established and
initialized.
Nr� r� r r� r� r� )r r� r� r� rd �
addHandlerr� r r@ rJ rL r rO � _packetQueuer� rA r� r^ �r r� r� r� rW r" r" r# �test_sendNotConnected� s$
� z(StreamManagerTests.test_sendNotConnectedc C sp t �t �� �}t �|�}|�d�}t�� |_|�� |� d� |�
d� | � d|j� � � | � d|j
d � dS )z�
Test send when the stream is connected but not yet initialized.
The data should be cached until the XML stream has been initialized.
Nr� r� r� r )r r� r� r� r� r r@ rJ r� rA rL r rO r� r� r" r" r# �test_sendNotInitialized� s
�
z*StreamManagerTests.test_sendNotInitializedc C s� t �t �� �}t �|�}t� }|�|� |�d�}|�� t� � |_
|� d� |� d� | �
d|j
�� � | �
d|jd � dS )z�
Test send after XML stream disconnection.
The data should be cached until a new XML stream has been established
and initialized.
Nr� r� r )r r� r� r� rd r� r� r� r r@ rJ rh rL r rO r� r� r" r" r# �test_sendDisconnected� s
z(StreamManagerTests.test_sendDisconnectedN)r0 r1 r2 r3 rE r$ r� r� r� r� r� r� r� r� r� r� r� r� r" r" r" r# r� s
r� c @ rc )
�XmlStreamServerFactoryTestsz8
Tests for L{xmlstream.XmlStreamServerFactory}.
c s* G dd� d�� � fdd�}t �|�| _dS )zQ
Set up a server factory with an authenticator factory function.
c @ rw )z<XmlStreamServerFactoryTests.setUp.<locals>.TestAuthenticatorc S rx rj )�
xmlstreamsr* r" r" r# r^ r� zEXmlStreamServerFactoryTests.setUp.<locals>.TestAuthenticator.__init__c S rz rj )r� r� r� r" r" r# r� r| zPXmlStreamServerFactoryTests.setUp.<locals>.TestAuthenticator.associateWithStreamN)r0 r1 r2 r^ r� r" r" r" r# r� r~ r� c s � � S rj r" r" �r� r" r# �authenticatorFactory s z?XmlStreamServerFactoryTests.setUp.<locals>.authenticatorFactoryN)r �XmlStreamServerFactoryr� )r r� r" r� r# rE � s z!XmlStreamServerFactoryTests.setUpc C s t t| j� dS )z<
L{XmlStreamServerFactory} is a L{Factory}.
N)r r r� r* r" r" r# rv s z*XmlStreamServerFactoryTests.test_interfacec C s" | j �d�}| �|g|jj� dS )a�
The authenticator factory should be used to instantiate the
authenticator and pass it to the protocol.
The default protocol, L{XmlStream} stores the authenticator it is
passed, and calls its C{associateWithStream} method. so we use that to
check whether our authenticator factory is used and the protocol
instance gets an authenticator.
N)r� r� r rD r� r� r" r" r# �,test_buildProtocolAuthenticatorInstantiation s
zHXmlStreamServerFactoryTests.test_buildProtocolAuthenticatorInstantiationc C s | j �d�}| �|tj� dS )zV
The protocol factory creates Jabber XML Stream protocols by default.
N)r� r� r� r r: r� r" r" r# �test_buildProtocolXmlStream# s z7XmlStreamServerFactoryTests.test_buildProtocolXmlStreamc C s8 | j �d�}| j �d�}| �||� | �|j|j� dS )z�
Subsequent calls to buildProtocol should result in different instances
of the protocol, as well as their authenticators.
N)r� r� r� rD )r �xs1�xs2r" r" r# �test_buildProtocolTwice* s z3XmlStreamServerFactoryTests.test_buildProtocolTwiceN) r0 r1 r2 r3 rE rv r� r� r� r" r" r" r# r� � s
r� )9r3 r
r �zope.interface.verifyr �twisted.internetr r �twisted.internet.errorr �twisted.internet.interfacesr �twisted.pythonr � twisted.testr �
twisted.trial� twisted.words.protocols.jabberr r r
r �!twisted.words.test.test_xmlstreamr �twisted.words.xishr r �_ssl� ImportErrorr7 �twisted.internet._sslverifyr �twisted.internet.sslr r �TestCaser r4 ry r� r� r� r� r� r � BaseFeatureInitiatingInitializerr8 r; rA r[ ru rd ro rs r� r� r� r" r" r" r# �<module> sZ � # \CL- ' B
6" f