← Back
Editing: testutils.cpython-38.pyc
U W[� � @ s� d Z ddlmZ ddlmZ ddlmZ G dd� d�Zdd� Z G d d � d e �Z G d d � d e �Z G d d� de �Z G dd� de �ZdS )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 q 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�|� qr|s�|r�dS dS dS )zOMove data back and forth. Returns whether any data was moved. r � N)r �seekr �read�truncater Z dataReceivedr )r ZcDataZsDataZbyter 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 )z2Take two Protocol instances and connect them. )r ZmakeConnectionr r r )r r ZcioZsior r r r �returnConnected? s r c @ s 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)Z assertEqual�domZ parseStringZtoxml)r �first�secondr r r �assertXMLEqualV s �z XMLAssertionMixin.assertXMLEqualN)r r r r r r r r r r O s r c @ s e Zd ZdZdd� Zdd� ZdS )�_EqualzN A class the instances of which are equal to anything and everything. c C s dS �NTr �r �otherr r r �__eq__j s z _Equal.__eq__c C s dS �NFr r r r r �__ne__n s z _Equal.__ne__N�r r r r r" r$ r r r r r f s r c @ s e Zd ZdZdd� Zdd� ZdS )� _NotEqualz> A class the instances of which are equal to nothing. c C s dS r# r r r r r r" w s z_NotEqual.__eq__c C s dS r r r r r r r$ { s z_NotEqual.__ne__Nr% r r r r r& s s r& c @ s e Zd ZdZdd� ZdS )�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)Z assertTrueZ assertFalser r& )r Z firstValueOneZsecondValueOneZvalueTwor r r �"assertNormalEqualityImplementation� s z7ComparisonTestsMixin.assertNormalEqualityImplementationN)r r r r r( r r r r r' � s r' N)r �ior Zxml.domr r Ztwisted.internet.protocolr r r �objectr r r& r' r r r r �<module> s (
Save File
Cancel