File: //usr/lib/python3/dist-packages/twisted/words/test/__pycache__/test_xishutil.cpython-38.pyc
U
W[�$ � @ s� d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
G dd � d �ZG d
d � d �ZG d d
� d
ej�ZG dd� dej�ZdS )z+
Test cases for twisted.words.xish.utility
� )�absolute_import�division)� OrderedDict)�unittest)�utility)�Element)�EventDispatcherc @ s e Zd ZdZdd� Zdd� ZdS )�CallbackTrackerz�
Test helper for tracking callbacks.
Increases a counter on each call to L{call} and stores the object
passed in the call.
c C s d| _ d | _d S )Nr ��called�obj��self� r �B/usr/lib/python3/dist-packages/twisted/words/test/test_xishutil.py�__init__ s zCallbackTracker.__init__c C s | j d | _ || _d S )N� r
)r r r r r �call s zCallbackTracker.callN)�__name__�
__module__� __qualname__�__doc__r r r r r r r s r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �OrderedCallbackTrackerz=
Test helper for tracking callbacks and their order.
c C s
g | _ d S �N)�callListr
r r r r * s z OrderedCallbackTracker.__init__c C s | j �| j� d S r )r �append�call1�r �objectr r r r . s z OrderedCallbackTracker.call1c C s | j �| j� d S r )r r �call2r r r r r 2 s z OrderedCallbackTracker.call2c C s | j �| j� d S r )r r �call3r r r r r 6 s z OrderedCallbackTracker.call3N)r r r r 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 )�EventDispatcherTestsz'
Tests for L{EventDispatcher}.
c C s< t � }t� }t� }t� }|�d|j� |�d|j� |�d|j� |�d|j� td�}|�d� td�}|�d� |�|� | �|jd � | �|j |� | �|jd
� |�|� | �|jd � | �|jd � | �|j |� | �|jd
� |�|d� | �|jd � | �|j |� |�
d|j� |�|� | �|jd � d S ) N�
/message/body�/message� /presence�//event/testevent��ns�message�body�r' �presencer+ � r r ) r r � addObserverr r �
addElement�dispatch� assertEqualr r �removeObserver)r �d�cb1�cb2Zcb3�msg�presr r r � testStuff@ s6
z EventDispatcherTests.testStuffc C sv t � }t� }t� }|�d|j� |�d|j� |�|d� | �|jd� | �|j|� | �|jd� | �|j|� dS )z�
Test adding two observers for the same query.
When the event is dispatched both of the observers need to be called.
r% r N)r r r- r r/ r0 r r )r r2 r3 r4 r r r �test_addObserverTwicee s z*EventDispatcherTests.test_addObserverTwicec sz t � �td�}t� � � �fdd�}��d|� ��|� | �� jd� ��|� | �� jd� ��|� | �� jd� dS ) zG
Test for registration of an observer during dispatch.
r& c s �� d� j� d S �Nr# )r- r ��_��cbr2 r r � onMessage� s zBEventDispatcherTests.test_addObserverInDispatch.<locals>.onMessager# r r r, N�r r r �addOnetimeObserverr/ r0 r �r r5 r> r r< r �test_addObserverInDispatchy s
z/EventDispatcherTests.test_addObserverInDispatchc sz t � �td�}t� � � �fdd�}��d|� ��|� | �� jd� ��|� | �� jd� ��|� | �� jd� dS )zN
Test for registration of a onetime observer during dispatch.
r& c s �� d� j� d S r9 )r@ r )r5 r< r r r> � s zIEventDispatcherTests.test_addOnetimeObserverInDispatch.<locals>.onMessager# r r Nr? rA r r< r �!test_addOnetimeObserverInDispatch� s
z6EventDispatcherTests.test_addOnetimeObserverInDispatchc C sV t � }td�}t� }|�d|j� |�|� | �|jd� |�|� | �|jd� d S )Nr& r# r )r r r r@ r r/ r0 r )r r2 r5 r= r r r �testOnetimeDispatch� s
z(EventDispatcherTests.testOnetimeDispatchc C sZ t � }td�}td�}t� }|�d|j� |�|�}| �d|� |�|�}| �d|� d S )Nr& r* r$ FT)r r r r- r r/ r0 )r r2 r5 r6 r= �resultr r r �testDispatcherResult� s
z)EventDispatcherTests.testDispatcherResultc C s� t � }t� }|�d|j� |�d|jd� |�d|jd� td�}|�d� |�|� | � |j
|j|j|jgdt dd � |j
D �� � d S )
Nr" r# ���r r& r) zCalls out of order: %sc S s g | ]
}|j �qS r )r )�.0�cr r r �
<listcomp>� s zAEventDispatcherTests.testOrderedXPathDispatch.<locals>.<listcomp>) r r r- r r r r r. r/ r0 r �repr�r r2 r= r5 r r r �testOrderedXPathDispatch� s
��z-EventDispatcherTests.testOrderedXPathDispatchc C s^ t � }t� }|�d|j� |�dd� | �d|j� |�d|j� | �dt|j �
d��� dS )zO
Test observer clean-up after removeObserver for named events.
� //event/testNr r ) r r r- r r/ r0 r r1 �len�_eventObservers�pop�r r2 r= r r r � test_cleanUpRemoveEventObserver� s z4EventDispatcherTests.test_cleanUpRemoveEventObserverc C sd t � }t� }td�}|�d|j� |�|� | �d|j� |�d|j� | �dt |j
� d��� dS )zO
Test observer clean-up after removeObserver for XPath events.
�Nr( r# r r N) r r r r- r r/ r0 r r1 rO �_xpathObserversrQ rL r r r � test_cleanUpRemoveXPathObserver� s
z4EventDispatcherTests.test_cleanUpRemoveXPathObserverc C sP t � }t� }|�d|j� |�dd� | �d|j� | �dt|j� d��� dS )zD
Test observer clean-up after onetime named events.
rN Nr r )
r r r@ r r/ r0 r rO rP rQ rR r r r � test_cleanUpOnetimeEventObserver� s z5EventDispatcherTests.test_cleanUpOnetimeEventObserverc C sV t � }t� }td�}|�d|j� |�|� | �d|j� | �dt|j �
d��� dS )zD
Test observer clean-up after onetime XPath events.
rT r# r r N) r r r r@ r r/ r0 r rO rU rQ rL r r r � test_cleanUpOnetimeXPathObserver s
z5EventDispatcherTests.test_cleanUpOnetimeXPathObserverc s� G dd� dt j�}G dd� dt�� � fdd�}t� }t� }t j}zx|t _|�d|� |�d|j� z|�dd� W n � k
r� | �d � Y nX | � d
t
| � � ��� | � d
|j � W 5 |t _X dS ) z�
Test that exceptions in observers do not bubble up to dispatch.
The exceptions raised in observers should be logged and other
observers should be called as if nothing happened.
c @ s e Zd Zdd� ZdS )zOEventDispatcherTests.test_observerRaisingException.<locals>.OrderedCallbackListc S s t � | _d S r )r Z callbacksr
r r r r s zXEventDispatcherTests.test_observerRaisingException.<locals>.OrderedCallbackList.__init__N)r r r r r r r r �OrderedCallbackList s rY c @ s e Zd ZdS )zEEventDispatcherTests.test_observerRaisingException.<locals>.TestErrorN)r r r r r r r � TestError s rZ c s
� � �d S r r r: �rZ r r �
raiseError! s zFEventDispatcherTests.test_observerRaisingException.<locals>.raiseErrorrN Nz2TestError raised. Should have been logged instead.r )
r Z CallbackList� Exceptionr r r- r r/ Zfailr0 rO ZflushLoggedErrorsr )r rY r\ r2 r= ZoriginalCallbackListr r[ r � test_observerRaisingException s" z2EventDispatcherTests.test_observerRaisingExceptionN)r r r r r7 r8 rB rC rD rF rM rS rV rW rX r^ r r r r r! ; s % r! c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � XmlPipeTestsz:
Tests for L{twisted.words.xish.utility.XmlPipe}.
c C s t �� | _d S r )r ZXmlPipe�piper
r r r �setUp? s zXmlPipeTests.setUpc sH � fdd�}g � | j j�d|� td�}| j j�|� | �|g� � dS )zO
Send an element from the source and observe it from the sink.
c s � � | � d S r �r �r �r r r r= G s z,XmlPipeTests.test_sendFromSource.<locals>.cb�/test[@xmlns="testns"]�ZtestnsZtestN)r` �sinkr- r �source�sendr0 �r r= �elementr rd r �test_sendFromSourceC s z XmlPipeTests.test_sendFromSourcec sH � fdd�}g � | j j�d|� td�}| j j�|� | �|g� � dS )zO
Send an element from the sink and observe it from the source.
c s � � | � d S r rb rc rd r r r= U s z*XmlPipeTests.test_sendFromSink.<locals>.cbre rf N)r` rh r- r rg ri r0 rj r rd r �test_sendFromSinkQ s z XmlPipeTests.test_sendFromSinkN)r r r r ra rl rm r r r r r_ : s r_ N)r Z
__future__r r � collectionsr Z
twisted.trialr Ztwisted.words.xishr Ztwisted.words.xish.domishr Ztwisted.words.xish.utilityr r r ZTestCaser! r_ r r r r �<module> s