File: //usr/lib/python3/dist-packages/twisted/words/test/__pycache__/test_xmlstream.cpython-38.pyc
U
W[� � @ s� d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z m Z m
Z
G dd� de j�ZG d d
� d
eje j�ZG d d � d e j�ZG d
d� de�ZG dd� de�ZdS )z,
Tests for L{twisted.words.xish.xmlstream}.
� )�absolute_import�division)�protocol)�failure)�unittest)�domish�utility� xmlstreamc @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�XmlStreamTestsc C s0 d| _ g | _t�� | _| | j_| jj| jj_d S )Nz no reason)�connectionLostMsg�outlistr Z XmlStreamZ transport�append�write��self� r �C/usr/lib/python3/dist-packages/twisted/words/test/test_xmlstream.py�setUp s
zXmlStreamTests.setUpc C s | j �t�t| j��� dS )zA
Stub loseConnection because we are a transport.
N)r ZconnectionLostr �Failure� Exceptionr r r r r �loseConnection s
�z XmlStreamTests.loseConnectionc C s, | j �� | j �d� | �| jd d� dS )zq
Calling L{xmlstream.XmlStream.send} results in the data being written
to the transport.
s <root>r N)r �connectionMade�send� assertEqualr r r r r � test_send s
zXmlStreamTests.test_sendc sJ g � � fdd�}| j �t j|� | j �� | j �d� | �dt� �� dS )zU
Receiving the starttag of the root element results in stream start.
c s � � d � d S �N�r
)Zrootelem�Z
streamStartedr r �streamStartEvent0 s z9XmlStreamTests.test_receiveRoot.<locals>.streamStartEvent�<root>� N)r � addObserverZSTREAM_START_EVENTr � dataReceivedr �len)r r r r r �test_receiveRoot* s
�
z XmlStreamTests.test_receiveRootc s� g �g � �fdd�}� fdd�}| j �t j|� | j �t j|� | j �� | j �d� | �dt��� | �dt� �� | j �d� | �dt��� | ��d � t
j �� | �dt� �� d S )
zN
Receiving malformed XML results in an L{STREAM_ERROR_EVENT}.
c s � � | � d S r r ��reason)� streamErrorr r �streamErrorEventA s z;XmlStreamTests.test_receiveBadXML.<locals>.streamErrorEventc s � � d � d S r r )�_�� streamEndr r �streamEndEventD s z9XmlStreamTests.test_receiveBadXML.<locals>.streamEndEventr r z<child><unclosed></child>r N) r r! ZSTREAM_ERROR_EVENT�STREAM_END_EVENTr r" r r# Z
assertTrueZcheckr Z ParserError)r r( r, r )r+ r' r �test_receiveBadXML: s$
�
�
z!XmlStreamTests.test_receiveBadXMLc sn g � � fdd�}| j �t j|� | j �� | �� | �dt� �� | �� d tj � | �� d �
� | j � dS )z@
Ending the stream fires a L{STREAM_END_EVENT}.
c s � � | � d S r r r% r* r r r, ] s z5XmlStreamTests.test_streamEnd.<locals>.streamEndEventr r N) r r! r- r r r r# ZassertIsInstancer r ZgetErrorMessager )r r, r r* r �test_streamEndW s
�
�z XmlStreamTests.test_streamEndN) �__name__�
__module__� __qualname__r r r r$ r. r/ r r r r r
s
r
c @ s e Zd ZdZdd� ZdS )�
DummyProtocolz�
I am a protocol with an event dispatcher without further processing.
This protocol is only used for testing XmlStreamFactoryMixin to make
sure the bootstrap observers are added to the protocol instance.
c O s" || _ || _g | _tj�| � d S r )�args�kwargsZ observersr �EventDispatcher�__init__)r r4 r5 r r r r7 s s zDummyProtocol.__init__N)r0 r1 r2 �__doc__r7 r r r r r3 k s r3 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �BootstrapMixinTestszq
Tests for L{xmlstream.BootstrapMixin}.
@ivar factory: Instance of the factory or mixin under test.
c C s t �� | _d S r )r ZBootstrapMixin�factoryr r r r r � s zBootstrapMixinTests.setUpc sP g � � fdd�}t � }| j�d|� | j�|� |�dd� | �dt� �� dS )zL
Dispatching an event fires registered bootstrap observers.
c s � � | � d S r r ��data�Zcalledr r �cb� s z6BootstrapMixinTests.test_installBootstraps.<locals>.cb�//event/myeventNr )r3 r: � addBootstrap�installBootstraps�dispatchr r# �r r> �
dispatcherr r= r �test_installBootstraps� s z*BootstrapMixinTests.test_installBootstrapsc sX g � � fdd�}| j �d|� | j �d|� t� }| j �|� |�dd� | �� � dS )zI
Test addition and removal of a bootstrap event handler.
c s � � | � d S r r r; r= r r r> � s z:BootstrapMixinTests.test_addAndRemoveBootstrap.<locals>.cbr? N)r: r@ ZremoveBootstrapr3 rA rB Z assertFalserC r r= r �test_addAndRemoveBootstrap� s z.BootstrapMixinTests.test_addAndRemoveBootstrapN)r0 r1 r2 r8 r rE rF r r r r r9 | s r9 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �!GenericXmlStreamFactoryTestsMixinz3
Generic tests for L{XmlStream} factories.
c C s t �� | _d S r )r ZXmlStreamFactoryr: r r r r r � s z'GenericXmlStreamFactoryTestsMixin.setUpc sJ g � � fdd�}| j �d|� | j �d�}|�dd� | �dt� �� dS )zY
The protocol factory installs bootstrap event handlers on the protocol.
c s � � | � d S r r r; r= r r r> � s zRGenericXmlStreamFactoryTestsMixin.test_buildProtocolInstallsBootstraps.<locals>.cbr? Nr )r: r@ �
buildProtocolrB r r# )r r> �xsr r= r �$test_buildProtocolInstallsBootstraps� s zFGenericXmlStreamFactoryTestsMixin.test_buildProtocolInstallsBootstrapsc C s | j �d�}| �| j |j � dS )z@
The protocol factory is saved in the protocol.
N)r: rH ZassertIdentical�r rI r r r � test_buildProtocolStoresFactory� s zAGenericXmlStreamFactoryTestsMixin.test_buildProtocolStoresFactoryN)r0 r1 r2 r8 r rJ rL r r r r rG � s rG c @ s e Zd ZdZdd� Zdd� ZdS )�XmlStreamFactoryMixinTestsz7
Tests for L{xmlstream.XmlStreamFactoryMixin}.
c C s t jd d d�| _t| j_d S )N)�test)r ZXmlStreamFactoryMixinr: r3 r r r r r r � s z XmlStreamFactoryMixinTests.setUpc C s0 | j �d�}| �d|j� | �ddi|j� dS )zb
Arguments passed to the factory are passed to protocol on
instantiation.
Nr rN )r: rH r r4 r5 rK r r r �"test_buildProtocolFactoryArguments� s z=XmlStreamFactoryMixinTests.test_buildProtocolFactoryArgumentsN)r0 r1 r2 r8 r rO r r r r rM � s rM N)r8 Z
__future__r r Ztwisted.internetr Ztwisted.pythonr Z
twisted.trialr Ztwisted.words.xishr r r ZTestCaser
ZProtocolr6 r3 r9 rG rM r r r r �<module> s \1#