File: //lib/python3/dist-packages/twisted/words/test/__pycache__/test_jabbercomponent.cpython-38.pyc
U
W[|7 � @ s d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z m
Z
mZ ddlmZ dd lmZ dd
lmZ G d d � d �ZG d
d� de
j�ZG dd� de
j�ZG dd� de
j�ZG dd� de j�ZG dd� de
j�Z G dd� de
j�Z G dd� de
j�Z G dd � d e
j�Z d S ) z7
Tests for L{twisted.words.protocols.jabber.component}
� )�sha1)� verifyObject)�failure)�unicode)�unittest)� component�ijabber� xmlstream)�JID)�domish)�XmlPipec @ s e Zd Zdd� Zdd� ZdS )�DummyTransportc C s
|| _ d S �N)�list)�selfr � r �I/usr/lib/python3/dist-packages/twisted/words/test/test_jabbercomponent.py�__init__ s zDummyTransport.__init__c C s | j �|� d S r )r �append)r �bytesr r r �write s zDummyTransport.writeN)�__name__�
__module__� __qualname__r r r r r r r
s r
c @ s e Zd Zdd� Zdd� ZdS )�#ComponentInitiatingInitializerTestsc C sj g | _ t�� | _d| j_t�| j�| _d| j_| j j| j_| j� � | j�
d� d| j_ t �
| j�| _d S )N�secret�test:componentz�<stream:stream xmlns='test:component' xmlns:stream='http://etherx.jabber.org/streams' from='example.com' id='12345' version='1.0'>Z12345)�outputr Z
Authenticator�
authenticator�password� XmlStream� namespacer �send�connectionMade� dataReceived�sidr Z ComponentInitiatingInitializer�init�r r r r �setUp s
�z)ComponentInitiatingInitializerTests.setUpc C sb | j �� }| jd }| �d|j� | �d|j� | �td��� t|�� g |_ | j
� |� � � |S )z?
Test basic operations of component handshake.
���� handshaker � 12345secret)
r& Z
initializer � assertEqual�nameZurir � hexdigestr Zchildrenr r$ ZtoXml)r �dr* r r r �
testHandshake+ s
�z1ComponentInitiatingInitializerTests.testHandshakeN)r r r r( r0 r r r r r s r c @ s e Zd Zdd� Zdd� ZdS )�ComponentAuthTestsc C s
d| _ d S �NT)� authComplete)r �streamr r r �
authPassedB s z ComponentAuthTests.authPassedc C s� d| _ g }t�dd�}t�|�}t|�|_|�tj| j � |�
� |� d� t d��
� �d�}| �|d d| d � |� d
� | �| j d � d S ) NFZcjidr sv <stream:stream xmlns='jabber:component:accept' xmlns:stream='http://etherx.jabber.org/streams' from='cjid' id='12345'>r+ �ascii� s <handshake>s </handshake>� <handshake/>T)r3 r Z ConnectComponentAuthenticatorr r r
� transport� addObserver�STREAM_AUTHD_EVENTr5 r# r$ r r. �encoder, )r ZoutlistZca�xsZhvr r r �testAuthE s
�
zComponentAuthTests.testAuthN)r r r r5 r> r r r r r1 A s r1 c @ s e Zd ZdZdd� ZdS )� ServiceTestsz)
Tests for L{component.Service}.
c C s t �� }ttj|� dS )zF
L{component.Service} implements L{ijabber.IService}.
N)r �Servicer r ZIService)r �servicer r r �test_interfaced s zServiceTests.test_interfaceN)r r r �__doc__rB r r r r r? _ s r? c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�JabberServiceHarnessc C s d| _ d| _d| _d S )NF)�componentConnectedFlag�componentDisconnectedFlag�transportConnectedFlagr' r r r r n s z JabberServiceHarness.__init__c C s
d| _ d S r2 )rE �r r r r r �componentConnecteds s z'JabberServiceHarness.componentConnectedc C s
d| _ d S r2 )rF r' r r r �componentDisconnectedv s z*JabberServiceHarness.componentDisconnectedc C s
d| _ d S r2 )rG rH r r r �transportConnectedy s z'JabberServiceHarness.transportConnectedN)r r r r rI rJ rK r r r r rD m s rD c @ s e Zd Zdd� ZdS )�JabberServiceManagerTestsc C s� t �dd�}t� }|�|� g }|�� �d �}| |_|j|j_|� � | �
d|j � |� |t
j� | �
d|j� |�d � | �
d|j� d S )NZfoor T)r ZServiceManagerrD ZsetServiceParentZ
getFactory�
buildProtocolr9 r r r# r, rG �dispatchr r; rE �connectionLostrF )r ZsmZsvcZwlistr= r r r �testSM~ s
z JabberServiceManagerTests.testSMN)r r r rP r r r r rL } s rL c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � RouterTestsz(
Tests for L{component.Router}.
c s| t �� }g � � fdd�|_t� }|�d|j� | �dt|j�� | �|j|jd � t �
d�}|j � |� | �|g� � dS )zJ
Test route registration and routing on incoming stanzas.
c s
� � | �S r �r ��element�Zroutedr r �<lambda>� � z+RouterTests.test_addRoute.<locals>.<lambda>z example.orgr7 )ZtestnsZtestN)
r �RouterZrouter �addRoute�sinkr, �len�routesr �Element�sourcer" )r �router�piperT r rU r �
test_addRoute� s
zRouterTests.test_addRoutec s� t � }t � }t�� }|�d|j� |�d|j� g � |j�d� fdd�� t�d�}d|d< d|d< |j� |� | �
|g� � d S )
z,
Test routing of a message.
�component1.example.orgzcomponent2.example.org�/*c s
� � | �S r rR rS �Zoutgoingr r rV � rW z(RouterTests.test_route.<locals>.<lambda>�NZpresence�from�toN� r r rX rY rZ r^ r: r r] r" r, )r �
component1Z
component2r_ �stanzar rd r �
test_route� s
�
zRouterTests.test_routec s� t � }t � }t�� }|�d|j� |�d|j� g � |j�d� fdd�� t�d�}d|d< d|d < |j� |� | �
|g� � dS )
a
Test routing of a message using the default route.
The default route is the one with L{None} as its key in the
routing table. It is taken when there is no more specific route
in the routing table that matches the stanza's destination.
rb Nrc c s
� � | �S r rR rS rd r r rV � rW z/RouterTests.test_routeDefault.<locals>.<lambda>re rf z example.comrg rh )r ri Zs2sr_ rj r rd r �test_routeDefault� s
z RouterTests.test_routeDefaultN)r r r rC ra rk rl r r r r rQ � s rQ c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )�!ListenComponentAuthenticatorTestsz>
Tests for L{component.ListenComponentAuthenticator}.
c C s, g | _ t�d�}t�|�| _| j j| j_d S )Nr )r r Z ListenComponentAuthenticatorr r r r" )r r r r r r( � s
z'ListenComponentAuthenticatorTests.setUpc C s | j �d� dS )zA
Stub loseConnection because we are a transport.
z no reasonN)r rO r' r r r �loseConnection� s z0ListenComponentAuthenticatorTests.loseConnectionc s� g � � fdd�}| j }||_|�| � | �d|j� | �|j� |�d� | �d|j � | �
d|j� | � |j� | �d|j j
f� d � dS )zK
The received stream header should set several attributes.
c s � � | |f� d S r rR )ZeventZ
observerfn�Z observersr r �addOnetimeObserver� s zPListenComponentAuthenticatorTests.test_streamStarted.<locals>.addOnetimeObserverNzz<stream:stream xmlns='jabber:component:accept' xmlns:stream='http://etherx.jabber.org/streams' to='component.example.org'>)r r rc r) )r rp �makeConnection�assertIdenticalr% � assertFalseZ _headerSentr$ r, �version�assertNotIdenticalZ
assertTruer � onElement)r rp r= r ro r �test_streamStarted� s
z4ListenComponentAuthenticatorTests.test_streamStartedc C sL g }| j }|j|_|�| � |�d� | �dt|�� | �d|d j� dS )zM
The received stream header should have a correct namespace.
zp<stream:stream xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' to='component.example.org'>r7 zinvalid-namespacer) N�r r �sendStreamErrorrq r$ r, r[ � condition�r � streamErrorsr= r r r � test_streamStartedWrongNamespace s
zBListenComponentAuthenticatorTests.test_streamStartedWrongNamespacec C sL g }| j }|j|_|�| � |�d� | �dt|�� | �d|d j� dS )zJ
The received stream header should have a 'to' attribute.
z_<stream:stream xmlns='jabber:component:accept' xmlns:stream='http://etherx.jabber.org/streams'>r7 zimproper-addressingr) Nrx r{ r r r �test_streamStartedNoTo s
z8ListenComponentAuthenticatorTests.test_streamStartedNoToc C sH g }| j }|j|j_t�d�}|�d� |j�|� | �d|d � dS )z<
We expect a handshake element with a hash.
)�jabber:component:acceptr* �1234r) N) r r r � onHandshaker r] Z
addContentrv r, )r �
handshakesr= r* r r r �test_onElement. s
z0ListenComponentAuthenticatorTests.test_onElementc C sV g }g }| j }|j|j_|j|_t�d�}|j�|� | �|� | � d|d j
� dS )z9
Reject elements that are not handshakes
)r �message�not-authorizedr) N) r r r r� ry r r] rv rs r, rz )r r� r| r= rT r r r �test_onElementNotHandshake< s
z<ListenComponentAuthenticatorTests.test_onElementNotHandshakec s` g � � fdd�}| j }|�t j|� d|_d}|j�|� | �d| jd � | �dt� �� dS ) zU
Receiving a handshake matching the secret authenticates the stream.
c s � � | � d S r rR �r= �Zauthdr r �
authenticatedS s zIListenComponentAuthenticatorTests.test_onHandshake.<locals>.authenticatedr� Z(32532c0f7dbf1253c095b18b18e36d38d94c1256r8 r) r7 N) r rp r; r% r r� r, r r[ )r r� r= �theHashr r� r �test_onHandshakeM s z2ListenComponentAuthenticatorTests.test_onHandshakec sl g }g � � fdd�}| j }|�t j|� |j|_d|_d}|j�|� | �d|d j � | �dt
� �� dS )zH
Receiving a bad handshake should yield a stream error.
c s � � | � d S r rR r� r� r r r� f s zRListenComponentAuthenticatorTests.test_onHandshakeWrongHash.<locals>.authenticatedr� r� r) r N) r rp r; r ry r% r r� r, rz r[ )r r| r� r= r� r r� r �test_onHandshakeWrongHash_ s z;ListenComponentAuthenticatorTests.test_onHandshakeWrongHashN)
r r r rC r( rn rw r} r~ r� r� r� r� r r r r rm � s rm c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� XMPPComponentServerFactoryTestsz<
Tests for L{component.XMPPComponentServerFactory}.
c C s8 t �� | _t �| jd�| _| j�d �| _td�| j_d S )Nr �component.example.org) r rX r_ ZXMPPComponentServerFactory�factoryrM r r
Z
thisEntityr' r r r r( z s
�z%XMPPComponentServerFactoryTests.setUpc C sV | j �| j t j� | �d| j j� | �d| jj� | �d| j j� | �d| j j� dS )zY
A new connection increases the stream serial count. No logs by default.
r r7 N) r rN �STREAM_CONNECTED_EVENTr, �serialr� rr � rawDataInFn� rawDataOutFnr' r r r �test_makeConnection� s
�z3XMPPComponentServerFactoryTests.test_makeConnectionc C s> d| j _| j�| jtj� | �d| jj� | �d| jj� dS )zD
Setting logTraffic should set up raw data loggers.
TN)r� Z
logTrafficr rN r� ru r� r� r' r r r � test_makeConnectionLogTraffic� s
�z=XMPPComponentServerFactoryTests.test_makeConnectionLogTrafficc C sX | j �| j t j� G dd� dt�}t�|� �}| j �|t j� | �dt| � |��� dS )zQ
An observer for stream errors should trigger onError to log it.
c @ s e Zd ZdS )z?XMPPComponentServerFactoryTests.test_onError.<locals>.TestErrorN)r r r r r r r � TestError� s r� r7 N)
r rN r� � Exceptionr ZFailureZSTREAM_ERROR_EVENTr, r[ ZflushLoggedErrors)r r� �reasonr r r � test_onError� s
� z,XMPPComponentServerFactoryTests.test_onErrorc C s< | j �| j t j� | �d| jj� | �| j | jjd � dS )zG
Make sure a new stream is added to the routing table.
r� N)r rN r; ZassertInr_ r\ rr r' r r r �test_connectionInitialized� s
�z:XMPPComponentServerFactoryTests.test_connectionInitializedc C s6 | j �| j t j� | j �dt j� | �d| jj� dS )zU
Make sure a stream is removed from the routing table on disconnect.
Nr� )r rN r; ZSTREAM_END_EVENTZ assertNotInr_ r\ r' r r r �test_connectionLost� s z3XMPPComponentServerFactoryTests.test_connectionLostN)
r r r rC r( r� r� r� r� r� r r r r r� u s
r� N) rC Zhashlibr Zzope.interface.verifyr Ztwisted.pythonr Ztwisted.python.compatr Z
twisted.trialr Z twisted.words.protocols.jabberr r r Z"twisted.words.protocols.jabber.jidr
Ztwisted.words.xishr Ztwisted.words.xish.utilityr r
ZTestCaser r1 r? r@ rD rL rQ rm r� r r r r �<module> s& ' #C