CasperSecurity
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<unittest.TestCase>}.
c C s$ | � t�|��� t�|��� � dS )z�
Verify that two strings represent the same XML document.
@param first: An XML string.
@type first: L{bytes}
@param second: An XML string that should match C{first}.
@type second: L{bytes}
N)� assertEqual�dom� parseString�toxml)r
�first�secondr r r �assertXMLEqualR s
�z XMLAssertionMixin.assertXMLEqualN)r r r r r( r r r r r! K s r! c @ �" e Zd ZdZdedefdd�ZdS )�_EqualzN
A class the instances of which are equal to anything and everything.
�other�returnc C � dS )NTr �r
r+ r r r �__eq__f � z
_Equal.__eq__N�r r r r �object�boolr/ r r r r r* a � r* c @ r) )� _NotEqualz>
A class the instances of which are equal to nothing.
r+ r, c C r- )NFr r. r r r r/ o r0 z_NotEqual.__eq__Nr1 r r r r r5 j r4 r5 c @ r )�ComparisonTestsMixina�
A mixin which defines a method for making assertions about the correctness
of an implementation of C{==} and C{!=}.
Use this to unit test objects which follow the common convention for C{==}
and C{!=}:
- The object compares equal to itself
- The object cooperates with unrecognized types to allow them to
implement the comparison
- The object implements not-equal as the opposite of equal
c C s� | � ||k� | � ||k� | �||k� | �||k� | �||k� | � ||k� | � |t� k� | �|t� k� | �|t� k� | � |t� k� dS )aU
Assert that C{firstValueOne} is equal to C{secondValueOne} but not
equal to C{valueOne} and that it defines equality cooperatively with
other types it doesn't know about.
@param firstValueOne: An object which is expected to compare as equal
to C{secondValueOne} and not equal to C{valueTwo}.
@param secondValueOne: A different object than C{firstValueOne} but
which is expected to compare equal to that object.
@param valueTwo: An object which is expected to compare as not equal to
C{firstValueOne}.
N)�
assertTrue� assertFalser* r5 )r
�
firstValueOne�secondValueOne�valueTwor r r �"assertNormalEqualityImplementation� s z7ComparisonTestsMixin.assertNormalEqualityImplementationN)r r r r r< r r r r r6 s s
r6 N)r �ior �xml.domr r# �twisted.internet.protocolr r r r! r* r5 r6 r r r r �<module> s (