o
¯b: ã @ sz d Z ddlmZ ddlmZ ddlmZ G dd„ dƒZdd„ Z G d d
„ d
ƒZ
G d d „ d ƒZ G d
d„ dƒZ G dd„ dƒZ
dS )a¼
I{Private} test utilities for use throughout Twisted's test suite. Unlike
C{proto_helpers}, this is no exception to the
don't-use-it-outside-Twisted-we-won't-maintain-compatibility rule!
@note: Maintainers be aware: things in this module should be gradually promoted
to more full-featured test helpers and exposed as public API as your
maintenance time permits. In order to be public API though, they need
their own test cases.
é )ÚBytesIO)Úminidom)Ú FileWrapperc @ s( e Zd ZdZdd„ Zdd„ Zdd„ ZdS ) ÚIOPumpzŠUtility to pump data between clients and servers for protocol testing.
Perhaps this is a utility worthy of being in protocol.py?
c C s || _ || _|| _|| _d S )N)ÚclientÚserverÚclientIOÚserverIO)Úselfr r r r © r ú8/usr/lib/python3/dist-packages/twisted/test/testutils.pyÚ__init__ s
zIOPump.__init__c C s | ¡ r | ¡ sdS dS )z,Pump until there is no more input or output.N)Úpump)r
r r r Úflush! s ÿz IOPump.flushc C s” | j d¡ | j d¡ | j ¡ }| j ¡ }| j d¡ | j d¡ | j ¡ | j ¡ |D ]}| j |¡ q.|D ]}| j |¡ q9|sF|rHdS dS )zOMove data back and forth.
Returns whether any data was moved.
r é )r Úseekr ÚreadÚtruncater Ú dataReceivedr )r
ÚcDataÚsDataÚbyter r r r & s
z IOPump.pumpN)Ú__name__Ú
__module__Ú __qualname__Ú__doc__r
r r r r r r r s
r c C sJ t ƒ }t ƒ }| t|ƒ¡ | t|ƒ¡ t|| ||ƒ}| ¡ | ¡ |S )z-Take two Protocol instances and connect them.)r ÚmakeConnectionr r r )r r ÚcioÚsior r r r ÚreturnConnected= s r c @ ó e Zd ZdZdd„ ZdS )ÚXMLAssertionMixinzŠ
Test mixin defining a method for comparing serialized XML documents.
Must be mixed in to a L{test case