File: //usr/lib/python3/dist-packages/twisted/words/test/__pycache__/test_jabberxmlstream.cpython-38.pyc
U
s�@g�� � @ s8 d Z ddlmZmZ ddlmZ ddlmZ ddlm Z m
Z
ddl m Z ddl
mZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZmZm Z m Z zddlm Z W n e k
r� dZ dZ Y n X dZ 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%e j0�Z1G d&d'� d'ej&�Z2G d(d)� d)ej&�Z3G d*d+� d+e4�Z5G d,d-� d-e j6�Z7G d.d/� d/e j6�Z8G d0d1� d1ej&�Z9G d2d3� d3ej&�Z:G d4d5� d5ej&�Z;G d6d7� d7e�Z<dS )8z8
Tests for L{twisted.words.protocols.jabber.xmlstream}.
� )�absolute_import�division)�unittest)� verifyObject)�defer�task)�ConnectionLost)�IProtocolFactory)�failure)�unicode)�
proto_helpers)�!GenericXmlStreamFactoryTestsMixin)�domish)�error�ijabber�jid� xmlstream)�sslNzSSL not available)�CertificateOptions)�ClientTLSOptions� urn:ietf:params:xml:ns:xmpp-tlsc @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �HashPasswordTestsz.
Tests for L{xmlstream.hashPassword}.
c C s t �dd�}| �d|� dS )zS
The sid and secret are concatenated to calculate sha1 hex digest.
�12345�secretZ(99567ee91b2c7cabf607f10cb9f4a3634fa820e0N�r � hashPassword� assertEqual��self�hash� r �I/usr/lib/python3/dist-packages/twisted/words/test/test_jabberxmlstream.py�
test_basic+ s z HashPasswordTests.test_basicc C s | � ttjdd� dS )zB
The session identifier must be a unicode object.
s ¹2345r N�� assertRaises� TypeErrorr r �r r r r! �test_sidNotUnicode3 s �z$HashPasswordTests.test_sidNotUnicodec C s | � ttjdd� dS )z8
The password must be a unicode object.
r s secrétNr# r& r r r! �test_passwordNotUnicode; s �z)HashPasswordTests.test_passwordNotUnicodec C s t �dd�}| �d|� dS )z\
The concatenated sid and password must be encoded to UTF-8 before hashing.
r u secrétZ(659bf88d8f8e179081f7f3b4a8e7d224652d2853Nr r r r r! �test_unicodeSecretC s 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 )NZ otherhost�testnszm<stream:stream xmlns:stream='http://etherx.jabber.org/streams' xmlns='testns' from='otherhost' version='1.0'>�get)r �ConnectAuthenticator� namespace� XmlStreamr ZClock�clock� callLaterZ
_callLater�makeConnectionr �StringTransport� dataReceived�IQ�iq)r �
authenticatorr r r! �setUpQ s
�z
IQTests.setUpc C s& | � | jd d� | �| jd � d S )N�typer0 �id)r r: �
assertTruer& r r r! � testBasic^ s zIQTests.testBasicc C sR | j j�� | j�� | jd �d�}| �| j j�� d| d d| d g� d S )Nr> zutf-8s <iq type='get' id='s '/>s <iq id='s ' type='get'/>)r � transport�clearr: �send�encode�assertIn�value)r ZidBytesr r r! �testSendc s
�zIQTests.testSendc s> � fdd�}� j �� }|�|� � j}|�d� j d � |S )Nc s � � | d d� d S )Nr= �result)r )rH r& r r! �cbn s z&IQTests.testResultResponse.<locals>.cb�<iq type='result' id='%s'/>r> )r: rC � addCallbackr r8 )r rI �d�xsr r& r! �testResultResponsem s
zIQTests.testResultResponsec C s6 | j �� }| �|tj� | j}|�d| j d � |S )Nz<iq type='error' id='%s'/>r> )r: rC �
assertFailurer Z StanzaErrorr r8 �r rL rM r r r! �testErrorResponsey 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 )NZhandledF)� assertFalse�getattr)r: r& r r! rI � s z*IQTests.testNonTrackedResponse.<locals>.cbz/iq���z <iq type='result' id='test'/>N)r Z upgradeWithIQResponseTrackerrR � iqDeferreds� addObserverr8 )r rM rI r r& r! �testNonTrackedResponse� 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.
rJ r> )r: rC r r8 � assertNotInrU rP 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)r: rC r �connectionLostrO r rP 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 r9 rC �
addErrback)r
rL ��ebr r r! r` � s z'IQTests.testNoModifyingDict.<locals>.ebrZ )r: rC r^ r r[ �r rL r r_ 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.
�< � ) r: �timeoutrC rO r � TimeoutErrorr4 �pumprR �callsrU ra 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.
rc rd rJ r> )
r: re rC r4 r5 r r8 rg rR rh ra 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.
rc rZ )
r: re rC r r[ rO r rR r4 rh rP r r r! �!testDisconnectTimeoutCancellation� s
z)IQTests.testDisconnectTimeoutCancellationN)r* r+ r, r- r< r@ rG rN rQ rW rY r\ rb ri rj rk r r r r! r. L s
r. 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 s
d| _ d S �NT)�gotStreamStart�r �objr r r! �
onStreamStart� s z XmlStreamTests.onStreamStartc C s
d| _ d S rm )� gotStreamEndro r r r! � onStreamEnd� s zXmlStreamTests.onStreamEndc C s
d| _ d S rm )�gotStreamErrorro r r r! �
onStreamError� s 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/errorr/ �rd r N)rn rr rt r r3 �
AuthenticatorrV rq rs ru r6 r Z StringTransportWithDisconnectionrA Zprotocolr2 �version�r rM r r r! r< 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 rT � � <stream:stream�/ xmlns:stream='http://etherx.jabber.org/streams'� xmlns='testns'�
version='1.0'N)r �
sendHeaderrA rF �splitrE r? � _headerSent�r rM Z 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.
Zdbzjabber:server:dialbackr rT r{ r| r} s! xmlns:db='jabber:server:dialback'r~ r N) r �prefixesr� rA rF r� rE r? 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 rT r{ � to='otherhost'� from='thishost'N) r r �JID�
thisEntity� otherEntity�
initiatingr� rA rF r� rE r� r r r! �test_sendHeaderInitiating3 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� FZ session01r rT r{ r� r� s id='session01'N) r r r� r� r� r� �sidr� rA rF r� rE r� r r r! �test_sendHeaderReceivingA 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 r8 r? rt rr rz r r r! �test_receiveStreamErrorQ s
z&XmlStreamTests.test_receiveStreamErrorc C sP | 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� rA rB �sendStreamErrorr � StreamError�assertNotEqualrF r? rr rz r r r! � test_sendStreamErrorInitiating^ s
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� rA rB r� r r� � assertNotr� r rF r? rr rz r r r! �&test_sendStreamErrorInitiatingNoHeaderm s
z5XmlStreamTests.test_sendStreamErrorInitiatingNoHeaderc C sP | j }d|_|�� |j�� |�t�d�� | �d|j� � � | �
| j � dS )z�
Test sendStreamError on a receiving xmlstream with a header sent.
An error should be sent out and the connection lost.
Fr� r� Nr� rz r r r! � test_sendStreamErrorReceiving~ s
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� rA rB r� r r� r? r� r� rF rr rz r r r! �%test_sendStreamErrorReceivingNoHeader� s
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 rM 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/>)r/ Zpresence)z http://etherx.jabber.org/streams�featuress <stream:features/>N)r rC r rA rF rB r �Element)r rM Zelr 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 r] ��appendr& )�connectionMadeCallsr r! �connectionMade� s zKXmlStreamTests.test_authenticator.<locals>.TestAuthenticator.connectionMadec s � � |� d S r] r� )r Z rootElement)�streamStartedCallsr r! �
streamStarted� s zJXmlStreamTests.test_authenticator.<locals>.TestAuthenticator.streamStartedc s � � |� d S r] r� rz )�associateWithStreamCallsr r! �associateWithStream� s zPXmlStreamTests.test_authenticator.<locals>.TestAuthenticator.associateWithStreamN)r* r+ r, 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'>rd )r r3 r r� r8 �lenr� )r r� �arM r r� r! �test_authenticator� s
z!XmlStreamTests.test_authenticatorN)r* r+ r, rq rs ru r< r� r� r� r� r� r� r� r� r� r� r� r� r r r r! rl � s
rl c @ s e Zd ZdS )� TestErrorN)r* r+ r, r r r r! r� � s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �AuthenticatorTestsc C s t �� | _t �| j�| _ d S r] )r rx r; r3 r& r r r! r< � s
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'>rw NZinvalid) r r6 r r7 r8 r ry �assertIdenticalr� r2 r� r� rz r r r! �test_streamStart� s
z#AuthenticatorTests.test_streamStartc C s0 | 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 r6 r r7 r8 r ry rz r r r! �test_streamStartLegacy� s
z)AuthenticatorTests.test_streamStartLegacyc C s0 | j }|�t�� � |�d� | �d|j� dS )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� rz r r r! � test_streamBadVersionOneDigit s
z0AuthenticatorTests.test_streamBadVersionOneDigitc C s0 | j }|�t�� � |�d� | �d|j� dS )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� rz r r r! � test_streamBadVersionNoNumber s
z0AuthenticatorTests.test_streamBadVersionNoNumberN)r* r+ r, r< r� r� r� r� r r r r! r� � s
r� c @ s< 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 r1 r; r3 rV �onAuthenticated� onInitFailedr& r r r! r< $ s z ConnectAuthenticatorTests.setUpc C s
d| _ d S rm )r� ro r r r! r� - s z)ConnectAuthenticatorTests.onAuthenticatedc C s
|| _ d S r] )r� )r r
r r r! r� 1 s 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 @ s e Zd Zdd� ZdS )z9ConnectAuthenticatorTests.testSucces.<locals>.Initializerc S s d S r] r r& r r r! �
initialize: s zDConnectAuthenticatorTests.testSucces.<locals>.Initializer.initializeN�r* r+ r, r� r r r r! � Initializer9 s r� N)r � initializersr; �initializeStreamr r? r� �r r� �initr r r! �
testSucces5 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 @ s e Zd Zdd� ZdS )z:ConnectAuthenticatorTests.testFailure.<locals>.Initializerc S s t �d S r] )r� r& r r r! r� J s zEConnectAuthenticatorTests.testFailure.<locals>.Initializer.initializeNr� r r r r! r� I s r� N) r r� r; r� r rR r� �assertNotIdenticalr� r? Zcheckr� r� r r r! � testFailureE 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� r/ z�<stream:stream xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' from='example.com' to='example.org' id='12345' version='1.0'>rw r � example.comNz9<stream:features><test xmlns='testns'/></stream:features>�r/ Ztest)r; r2 r r6 r r7 r8 r ry r� r� �hostr� r� r� r� rE r� r? rz r r r! r� W s
z*ConnectAuthenticatorTests.test_streamStartN) r* r+ r, r< 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 s t �� | _t �| j�| _ d S r] )r ZListenAuthenticatorr; r3 r& r r r! r< v s
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� rw r �
jabber:clientr� )r r6 r r7 r� r� r8 r ry r� r� r2 r� r� r� rz 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 r6 r r7 r8 �assertIsInstancer� r rz r r r! � test_streamStartUnicodeSessionID� s
z9ListenAuthenticatorTests.test_streamStartUnicodeSessionIDN)r* r+ r, r- r< r� r� r r r r! r� q s r� c @ sv e Zd Zdd� Zdd� Zdd� Zdd� Zee_d d
� Z ee _d d � Z
ee
_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 �savedSSLr1 r; r3 r� rC r� r8 �TLSInitiatingInitializerr� r& r r r! r< � s
z#TLSInitiatingInitializerTests.setUpc C s | j t_d S r] )r� r r r& r r r! �tearDown� s z&TLSInitiatingInitializerTests.tearDownc C s$ t j| j dd�| _| �| jj� dS )z>
Passing required sets the instance variable.
T)�requiredN)r r� r� r? 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 � � �
d d
d g� j� |S ) zN
When TLS is wanted and SSL available, StartTLS is initiated.
c s � j �d�S �N�TLS�r� r� )�ctxr& r r! �<lambda>� r� zDTLSInitiatingInitializerTests.test_wantedSupported.<locals>.<lambda>c s � j �d�S �Nr� r� r r& r r! r� � r� c s � j �d�S �N�headerr� r r& r r! r� � r� � rd �starttls�<proceed xmlns='%s'/>r� r� r� )r r7 r rA �startTLSr� r� r� �startrK r ZResetr� r� �name� NS_XMPP_TLS�urir8 r� )r rL r� r r& r! �test_wantedSupported� s
z2TLSInitiatingInitializerTests.test_wantedSupportedc st � fdd�}t �� � j_|� jj_� fdd�� j_� fdd�� j_� j�� }� j� dt
� � � ddd g� j � |S )
z:
The server certificate will be verified.
c s* � � | t� � �| jd� � j�d� d S )Nr� r� )r� r r Z _hostnamer� r� ��contextFactoryr& r r! � fakeStartTLS� s zJTLSInitiatingInitializerTests.test_certificateVerify.<locals>.fakeStartTLSc s � j �d�S r� r� r r& r r! r� � r� zFTLSInitiatingInitializerTests.test_certificateVerify.<locals>.<lambda>c s � j �d�S r� r� r r& r r! r� � r� r� r� r� r� )
r r7 r rA r� r� r� r� r� r8 r� r r� �r r� rL 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
� ��dd d
g�j� |S ) zH
A custom contextFactory is passed through to startTLS.
)ZconfigurationForTLSc s �� � | � �j�d� d S r� )ZassertIsr� r� r� �r� r r r! r� � s zQTLSInitiatingInitializerTests.test_certificateVerifyContext.<locals>.fakeStartTLSc s � j �d�S r� r� r r& r r! r� � r� zMTLSInitiatingInitializerTests.test_certificateVerifyContext.<locals>.<lambda>c s � j �d�S r� r� r r& r r! r� � r� r� r� r� r� )r r r� r� r� r r7 rA r� r� r� r� r8 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.
NFrd ) r r r� r� r� rK r r� r� ra r r r! �"test_wantedNotSupportedNotRequired s
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.
NTrd )
r r r� r� r� rO ZTLSNotSupportedr r� r� ra r r r! � test_wantedNotSupportedRequired s
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� Frd )r r� Z
addElementr� r� r r� r� �wantedr� r r� r� rO Z TLSRequired�r ZtlsrL 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� FNrd )r r� r� r� r r� r� r� r� r� rK r r� r� r� r r r! �test_notWantedNotRequired0 s
z7TLSInitiatingInitializerTests.test_notWantedNotRequiredc C s2 dt _| j�� }| �|t j� | j �dt � |S )zN
TLSFailed is raised when the server responds with a failure.
rd z<failure xmlns='%s'/>)r r r� r� rO Z TLSFailedr8 r� ra r r r! � test_failed? s
z)TLSInitiatingInitializerTests.test_failedN)r* r+ r, r< r� r� r� �
skipWhenNoSSL�skipr� r� r� r� r� r� r� r r r r! r� � s r� c @ s e Zd ZdZdd� ZdS )�TestFeatureInitializerr� c C s
t �d �S r] )r Zsucceedr& r r r! r� R s z TestFeatureInitializer.startN)r* r+ r, �featurer� r r r r! r O s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�%BaseFeatureInitiatingInitializerTestsc C s t �t �� �| _ t| j �| _d S r] )r r3 rx r r� r& r r r! r< Y 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! �testAdvertized^ 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 ZFeatureNotAdvertizedr� r& r r r! �testNotAdvertizedRequiredg s 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! � testNotAdvertizedNotRequiredp s zBBaseFeatureInitiatingInitializerTests.testNotAdvertizedNotRequiredN)r* r+ r, r< r r r r r r r! r W s r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�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� r: r0 r= �user1@example.com�to�user2@example.com/resource�fromZstanza1r> rH N)r r� r �
toResponser� r �r ZstanzaZresponser r r! �test_toResponse| 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.
r r0 r= r
r r
N)r r� r r r rR � hasAttributer r r r! �test_toResponseNoFrom� s
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.
r r0 r= r r
r N)r r� r r rR r r r r r r! �test_toResponseNoTo� s
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� �messageZchatr= r r
N�r r� r r rR r r r r r! �test_toResponseNoAddressing� s
z+ToResponseTests.test_toResponseNoAddressingc C s( t �d�}t�|�}| �|�d�� dS )zP
Test that a proper response is generated without id attribute.
r r> Nr r r r r! � test_noID� s
zToResponseTests.test_noIDc C s( t �d�}t�|�}| �|�d�� dS )zR
Test that a proper response is generated without type attribute.
r r= Nr r r r r! � test_noType� s
zToResponseTests.test_noTypeN) r* r+ r, r r r r r r r r r r! r z s r c @ s e Zd ZdZdd� Zdd� ZdS )� DummyFactoryzJ
Dummy XmlStream factory that only registers bootstrap observers.
c C s
i | _ d S r] �� callbacksr& r r r! �__init__� s zDummyFactory.__init__c C s || j |< d S r] r )r Zevent�callbackr r r! � addBootstrap� s zDummyFactory.addBootstrapN)r* r+ r, r- r r r r r r! r � s r c @ s8 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 r] )r� rz r r r! r6 � s z DummyXMPPHandler.makeConnectionc C s | j d7 _ d S �Nrd �r! r& r r r! r� � s z DummyXMPPHandler.connectionMadec C s | j d7 _ d S r$ )r" r& r r r! �connectionInitialized� s z&DummyXMPPHandler.connectionInitializedc C s | j d7 _ d S r$ )r# �r �reasonr r r! r[ � s z DummyXMPPHandler.connectionLostN) r* r+ r, r- r r6 r� r& r[ r r r r! r � s r c @ s e Zd ZdZdd� Zdd� ZdS )�FailureReasonXMPPHandlerz>
Dummy handler specifically for failure Reason tests.
c C s
d| _ d S )NF)�gotFailureReasonr& r r r! r � s z!FailureReasonXMPPHandler.__init__c C s t |tj�rd| _d S rm )�
isinstancer
�Failurer* r' r r r! r[ � s z'FailureReasonXMPPHandler.connectionLostN)r* r+ r, r- r r[ r r r r! r) � s r) c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �XMPPHandlerTestsz-
Tests for L{xmlstream.XMPPHandler}.
c C s t tjt�� � dS )zN
L{xmlstream.XMPPHandler} implements L{ijabber.IXMPPHandler}.
N)r r Z IXMPPHandlerr � XMPPHandlerr& r r r! �test_interface� s z XMPPHandlerTests.test_interfacec C s@ G dd� dt �}t�� }|� |_|�d� | �dg|jj� dS )zP
Test that data is passed on for sending by the stream manager.
c @ s e Zd Zdd� Zdd� ZdS )z6XMPPHandlerTests.test_send.<locals>.DummyStreamManagerc S s
g | _ d S r] )�outlistr& r r r! r s z?XMPPHandlerTests.test_send.<locals>.DummyStreamManager.__init__c S s | j �|� d S r] )r0 r� )r �datar r r! rC s z;XMPPHandlerTests.test_send.<locals>.DummyStreamManager.sendN)r* r+ r, r rC r r r r! �DummyStreamManager s r2 r� N)�objectr r. �parentrC r r0 )r r2 �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 @ s e Zd Zdd� ZdS )z=XMPPHandlerTests.test_makeConnection.<locals>.TestXMPPHandlerc S s
d| _ d S rm r% r& r r r! r� s zLXMPPHandlerTests.test_makeConnection.<locals>.TestXMPPHandler.connectionMadeN)r* r+ r, r� r r r r! �TestXMPPHandler s r6 N)r r. r3 rx r6 r? r! r� )r r6 r5 rM 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 r. r3 rx r6 r[ � Exceptionr� )r r5 rM r r r! �test_connectionLost$ s
z$XMPPHandlerTests.test_connectionLostN)r* r+ r, r- r/ r� r7 r9 r r r r! r- � s
r- c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �XMPPHandlerCollectionTestsz7
Tests for L{xmlstream.XMPPHandlerCollection}.
c C s t �� | _d S r] )r ZXMPPHandlerCollection�
collectionr& r r r! r< 5 s z XMPPHandlerCollectionTests.setUpc C s t tj| j� dS )zZ
L{xmlstream.StreamManager} implements L{ijabber.IXMPPHandlerCollection}.
N)r r ZIXMPPHandlerCollectionr; r& r r r! r/ 9 s z)XMPPHandlerCollectionTests.test_interfacec C s4 t � }|�| j� | �|| j� | �| j|j� dS )z:
Test the addition of a protocol handler.
N)r �setHandlerParentr; rE r� r4 �r r5 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)r r< r; ZdisownHandlerParentrX r� r4 r= r r r! �test_removeHandlerJ s
z-XMPPHandlerCollectionTests.test_removeHandlerN)r* r+ r, r- r< r/ r> r? r r r r! r: 0 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 r] )r r �
StreamManager�
streamManager)r �factoryr r r! r< [ 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� rv r� ) rB r� r r Zhandlers�
_connectedrC r �_authd�
_disconnectedZinitializationFailed)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.
rd r N) rB r r< r r3 rx rD r r! r" r# �r rG r5 rM r r r! �test_connectedq s
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)
rB r r< r r3 rx rD r� � rawDataInFn� rawDataOutFnrH 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) rB Z
logTrafficr r< r r3 rx rD r� rJ rK rH r r r! � test_connectedLogTrafficTrue� 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 rd N) rB r r< r r3 rx rE r r! r" r# rH r r r! �
test_authd� s
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 rd N) rB r r< r r3 rx rF r r! r" r# rH r r r! �test_disconnected� s
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) rB r) r< rF r
r, r8 r r* �r rG r5 r r r! �test_disconnectedReason� 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)rB r r< r r! r" r# rP 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.
rd r N) rB r r3 rx rD rE r r< r r! r" r# )r rG rM r5 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 �XmlStreamFactoryrx rA �
buildProtocolr r7 rA r� r8 �dispatchrC r rF �r rC rG rM 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 rS rx rA r �
addHandlerrT r r7 rA rC r rF � _packetQueuer� r8 rU rR �r rC rG r5 rM 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 rS rx rA rT r r7 rA r� r8 rC r rF rY rV 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 rS rx rA r rX rT r� r r7 rA r[ rC r rF rY rZ r r r! �test_sendDisconnected, s
z(StreamManagerTests.test_sendDisconnectedN)r* r+ r, r- r< r" rI rL rM rN rO rQ r> rR rW r[ r\ r] r r r r! r@ V s
r@ c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�XmlStreamServerFactoryTestsz8
Tests for L{xmlstream.XmlStreamServerFactory}.
c s, G dd� dt �� � fdd�}t�|�| _dS )zQ
Set up a server factory with an authenticator factory function.
c @ s e Zd Zdd� Zdd� ZdS )z<XmlStreamServerFactoryTests.setUp.<locals>.TestAuthenticatorc S s
g | _ d S r] )�
xmlstreamsr& r r r! r M s zEXmlStreamServerFactoryTests.setUp.<locals>.TestAuthenticator.__init__c S s | j �|� d S r] )r_ r� rz r r r! r� P s zPXmlStreamServerFactoryTests.setUp.<locals>.TestAuthenticator.associateWithStreamN)r* r+ r, r r� r r r r! r� L s r� c s � � S r] r r �r� r r! �authenticatorFactoryS s z?XmlStreamServerFactoryTests.setUp.<locals>.authenticatorFactoryN)r3 r ZXmlStreamServerFactoryrC )r ra r r` r! r< H s z!XmlStreamServerFactoryTests.setUpc C s t t| j� dS )z<
L{XmlStreamServerFactory} is a L{Factory}.
N)r r rC r& r r r! r/ Y 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)rC rT r r; r_ rz 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)rC rT r� r r3 rz r r r! �test_buildProtocolXmlStreamn 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)rC rT r� r; )r Zxs1Zxs2r r r! �test_buildProtocolTwicev s z3XmlStreamServerFactoryTests.test_buildProtocolTwiceN) r* r+ r, r- r< r/ rb rc rd r r r r! r^ C s r^ )=r- Z
__future__r r Z
twisted.trialr Zzope.interface.verifyr Ztwisted.internetr r Ztwisted.internet.errorr Ztwisted.internet.interfacesr Ztwisted.pythonr
Ztwisted.python.compatr Z twisted.testr Z!twisted.words.test.test_xmlstreamr
Ztwisted.words.xishr Z twisted.words.protocols.jabberr r r r r � ImportErrorr Ztwisted.internet.sslr Ztwisted.internet._sslverifyr r� ZTestCaser r. rl r8 r� r� r� r� r� Z BaseFeatureInitiatingInitializerr r r r3 r r. r r) r- r: r@ r^ r r r r! �<module> sV
& ( k@O, 3#H
8& n