File: //lib/python3/dist-packages/twisted/words/test/__pycache__/test_jabbercomponent.cpython-310.pyc
o
�bR6 � @ s d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
m Z m Z ddl
mZ ddlmZ dd lmZ G d
d � d �ZG d d
� d
ej�ZG dd� dej�ZG dd� dej�ZG dd� de
j�ZG dd� dej�ZG dd� dej�ZG dd� dej�Z G dd� dej�Z d S ) z7
Tests for L{twisted.words.protocols.jabber.component}
� )�sha1)� verifyObject)�failure)�unittest)� component�ijabber� xmlstream)�JID)�domish)�XmlPipec @ � 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__ �
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
� r
c @ r )�#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'>�12345)�outputr �
Authenticator�
authenticator�password� XmlStream� namespacer �send�connectionMade� dataReceived�sidr � 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+ �
initializer � assertEqual�name�urir � hexdigest�str�childrenr r( �toXml)r �dr/ r r r �
testHandshake- s
z1ComponentInitiatingInitializerTests.testHandshakeN)r r r r- r: r r r r r s r c @ r )�ComponentAuthTestsc C �
d| _ d S �NT)� authComplete)r �streamr r r �
authPassedD r 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 ) NF�cjidr sv <stream:stream xmlns='jabber:component:accept' xmlns:stream='http://etherx.jabber.org/streams' from='cjid' id='12345'>r0 �ascii� s <handshake>s </handshake>� <handshake/>T)r> r � ConnectComponentAuthenticatorr r$ r
� transport� addObserver�STREAM_AUTHD_EVENTr@ r' r( r r5 �encoder2 )r �outlist�ca�xs�hvr r r �testAuthG s
�
zComponentAuthTests.testAuthN)r r r r@ rN r r r r r; C r r; 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 �IService)r �servicer r r �test_interfacef s zServiceTests.test_interfaceN)r r r �__doc__rS r r r r rO a s rO 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 o s
z JabberServiceHarness.__init__c C r< r= )rV �r r r r r �componentConnectedt r z'JabberServiceHarness.componentConnectedc C r< r= )rW r, r r r �componentDisconnectedw r z*JabberServiceHarness.componentDisconnectedc C r< r= )rX rY r r r �transportConnectedz r z'JabberServiceHarness.transportConnectedN)r r r r rZ r[ r\ r r r r rU n s
rU 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 )N�foor# T)r �ServiceManagerrU �setServiceParent�
getFactory�
buildProtocolrF r r r' r2 rX �dispatchr rH rV �connectionLostrW )r �sm�svc�wlistrL r r r �testSM s
z JabberServiceManagerTests.testSMN)r r r rh r r r r r] ~ s r] 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 r �r ��element��routedr r �<lambda>� �
z+RouterTests.test_addRoute.<locals>.<lambda>z example.orgrC )�testns�testN)
r �Router�router �addRoute�sinkr2 �len�routesr
�Element�sourcer& )r �router�piperm r rn 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 rj r rk rl ��outgoingr r rp � rq z(RouterTests.test_route.<locals>.<lambda>�N�presence�from�toN� r r rt rv rw r{ rG r
rz r& r2 )r �
component1�
component2r| �stanzar r� 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.
r Nr� c rj r rk rl r� r r rp � rq z/RouterTests.test_routeDefault.<locals>.<lambda>r� r� z example.comr� r� )r r� �s2sr| r� r r� r �test_routeDefault� s
z RouterTests.test_routeDefaultN)r r r rT r~ r� r� r r r r ri � s
ri 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 � 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 rd 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 rk )�event�
observerfn�� 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 r� r. )r r� �makeConnection�assertIdenticalr) � assertFalse� _headerSentr( r2 �version�assertNotIdentical�
assertTruer" � onElement)r r� rL r r� r �test_streamStarted� s
� z4ListenComponentAuthenticatorTests.test_streamStartedc C �L 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'>rC zinvalid-namespacer. N�r r �sendStreamErrorr� r( r2 rx � condition�r � streamErrorsrL r r r � test_streamStartedWrongNamespace
s
�zBListenComponentAuthenticatorTests.test_streamStartedWrongNamespacec C r� )zJ
The received stream header should have a 'to' attribute.
z_<stream:stream xmlns='jabber:component:accept' xmlns:stream='http://etherx.jabber.org/streams'>rC zimproper-addressingr. Nr� 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
rz �
addContentr� r2 )r �
handshakesrL 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� r� r
rz r� r� r2 r� )r r� r� rL rm r r r �test_onElementNotHandshake9 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 � � � | � d S r rk �rL ��authdr r �
authenticatedO � zIListenComponentAuthenticatorTests.test_onHandshake.<locals>.authenticatedr� �(32532c0f7dbf1253c095b18b18e36d38d94c1256rD r. rC N) r r� rH r) r" r� r2 r rx )r r� rL �theHashr r� r �test_onHandshakeI 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 r� r rk r� r� r r r� a r� zRListenComponentAuthenticatorTests.test_onHandshakeWrongHash.<locals>.authenticatedr� r� r. r N) r r� rH r r� r) r" r� r2 r� rx )r r� r� rL r� r r� r �test_onHandshakeWrongHashZ s z;ListenComponentAuthenticatorTests.test_onHandshakeWrongHashN)
r r r rT r- r� r� r� r� r� r� r� r� r r r r r� � s r� 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 rt r| �XMPPComponentServerFactory�factoryrb r r �
thisEntityr, r r r r- t 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 rC N) r rc �STREAM_CONNECTED_EVENTr2 �serialr� r� � rawDataInFn� rawDataOutFnr, r r r �test_makeConnectionz 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� �
logTrafficr rc r� r� 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� rC N)
r rc r� � Exceptionr �Failure�STREAM_ERROR_EVENTr2 rx �flushLoggedErrors)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 rc rH �assertInr| ry r� 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 rc rH �STREAM_END_EVENT� assertNotInr| ry r, r r r �test_connectionLost� s z3XMPPComponentServerFactoryTests.test_connectionLostN)
r r r rT r- r� r� r� r� r� r r r r r� o s
r� N) rT �hashlibr �zope.interface.verifyr �twisted.pythonr �
twisted.trialr � twisted.words.protocols.jabberr r r �"twisted.words.protocols.jabber.jidr �twisted.words.xishr
�twisted.words.xish.utilityr r
�TestCaser r; rO rP rU r] ri r� r� r r r r �<module> s& (
"?