← Back
Editing: test_iosim.cpython-38.pyc
U W[d# � @ s� d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl m Z m Z ddl mZ G dd � d e�Zee�G d d � d e��ZG d d � d e�ZG dd� de�ZdS )z" Tests for L{twisted.test.iosim}. � )�absolute_import�division)� implementer)� IPushProducer)�Protocol)� FakeTransport�connect)�TestCasec @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �FakeTransportTestsz% Tests for L{FakeTransport}. c C sH t t� d�}t t� d�}| �|jt� | �|jt� | �|j|j� dS )ze Each L{FakeTransport} receives a serial number that uniquely identifies it. TFN)r �objectZassertIsInstance�serial�intZassertNotEqual)�self�a�b� r �9/usr/lib/python3/dist-packages/twisted/test/test_iosim.py�test_connectionSerial s z(FakeTransportTests.test_connectionSerialc C s> t t� d�}|�d� |�dddg� | �d�|j�d� dS ) zl L{FakeTransport.writeSequence} will write a sequence of L{bytes} to the transport. F� a� b� c� d� s abcdN)r r �writeZ writeSequence� assertEqual�join�stream�r r r r r �test_writeSequence$ s z%FakeTransportTests.test_writeSequencec C s@ t t� d�}|�d� |�� |�d� | �d�|j�d� dS )z� L{FakeTransport.write} will accept writes after transport was closed, but the data will be silently discarded. Fs befores afterr N)r r r ZloseConnectionr r r r r r r �test_writeAfterClose1 s z'FakeTransportTests.test_writeAfterCloseN)�__name__� __module__� __qualname__�__doc__r r r r r r r r s r c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S ) �StrictPushProducerz� An L{IPushProducer} implementation which produces nothing but enforces preconditions on its state transition methods. �runningc C s | j dkrtd��d| _ d S )N�stoppedz)Cannot stop already-stopped IPushProducer)�_state� ValueError�r r r r � stopProducingG s z StrictPushProducer.stopProducingc C s$ | j dkrtd�| j ���d| _ d S )Nr% z Cannot pause {} IPushProducer�paused�r' r( �formatr) r r r �pauseProducingM s �z!StrictPushProducer.pauseProducingc C s$ | j dkrtd�| j ���d| _ d S )Nr+ z Cannot resume {} IPushProducerr% r, r) r r r �resumeProducingU s �z"StrictPushProducer.resumeProducingN)r r! r" r# r' r* r. r/ r r r r r$ ? s r$ c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(S ))�StrictPushProducerTestsz* Tests for L{StrictPushProducer}. c C s t � S )zp @return: A new L{StrictPushProducer} which has not been through any state changes. )r$ r) r r r �_initialb s z StrictPushProducerTests._initialc C s t � }|�� |S )z@ @return: A new, stopped L{StrictPushProducer}. )r$ r* �r �producerr r r �_stoppedj s z StrictPushProducerTests._stoppedc C s t � }|�� |S )z? @return: A new, paused L{StrictPushProducer}. )r$ r. r2 r r r �_pauseds s z StrictPushProducerTests._pausedc C s t � }|�� |�� |S )zY @return: A new L{StrictPushProducer} which has been paused and resumed. )r$ r. r/ r2 r r r �_resumed| s z StrictPushProducerTests._resumedc C s | � |jd� dS )z� Assert that the given producer is in the stopped state. @param producer: The producer to verify. @type producer: L{StrictPushProducer} r& N�r r' r2 r r r � assertStopped� s z%StrictPushProducerTests.assertStoppedc C s | � |jd� dS )z� Assert that the given producer is in the paused state. @param producer: The producer to verify. @type producer: L{StrictPushProducer} r+ Nr7 r2 r r r � assertPaused� s z$StrictPushProducerTests.assertPausedc C s | � |jd� dS )z� Assert that the given producer is in the running state. @param producer: The producer to verify. @type producer: L{StrictPushProducer} r% Nr7 r2 r r r � assertRunning� s z%StrictPushProducerTests.assertRunningc C s | � t| �� j� dS )zz L{StrictPushProducer.stopProducing} raises L{ValueError} if called when the producer is stopped. N)� assertRaisesr( r4 r* r) r r r �test_stopThenStop� s z)StrictPushProducerTests.test_stopThenStopc C s | � t| �� j� dS )z{ L{StrictPushProducer.pauseProducing} raises L{ValueError} if called when the producer is stopped. N)r; r( r4 r. r) r r r �test_stopThenPause� s z*StrictPushProducerTests.test_stopThenPausec C s | � t| �� j� dS )z| L{StrictPushProducer.resumeProducing} raises L{ValueError} if called when the producer is stopped. N)r; r( r4 r/ r) r r r �test_stopThenResume� s z+StrictPushProducerTests.test_stopThenResumec C s | � � }|�� | �|� dS )zn L{StrictPushProducer} is stopped if C{stopProducing} is called on a paused producer. N)r5 r* r8 r2 r r r �test_pauseThenStop� s z*StrictPushProducerTests.test_pauseThenStopc C s | � � }| �t|j� dS )zs L{StrictPushProducer.pauseProducing} raises L{ValueError} if called on a paused producer. N)r5 r; r( r. r2 r r r �test_pauseThenPause� s z+StrictPushProducerTests.test_pauseThenPausec C s | � � }|�� | �|� dS )zp L{StrictPushProducer} is resumed if C{resumeProducing} is called on a paused producer. N)r5 r/ r: r2 r r r �test_pauseThenResume� s z,StrictPushProducerTests.test_pauseThenResumec C s | � � }|�� | �|� dS )zo L{StrictPushProducer} is stopped if C{stopProducing} is called on a resumed producer. N)r6 r* r8 r2 r r r �test_resumeThenStop� s z+StrictPushProducerTests.test_resumeThenStopc C s | � � }|�� | �|� dS )zo L{StrictPushProducer} is paused if C{pauseProducing} is called on a resumed producer. N)r6 r. r9 r2 r r r �test_resumeThenPause� s z,StrictPushProducerTests.test_resumeThenPausec C s | � � }| �t|j� dS )zu L{StrictPushProducer.resumeProducing} raises L{ValueError} if called on a resumed producer. N)r6 r; r( r/ r2 r r r �test_resumeThenResume� s z-StrictPushProducerTests.test_resumeThenResumec C s | � � }|�� | �|� dS )zn L{StrictPushProducer} is stopped if C{stopProducing} is called in the initial state. N)r1 r* r8 r2 r r r � test_stop� s z!StrictPushProducerTests.test_stopc C s | � � }|�� | �|� dS )zn L{StrictPushProducer} is paused if C{pauseProducing} is called in the initial state. N)r1 r. r9 r2 r r r � test_pause s z"StrictPushProducerTests.test_pausec C s | � � }| �t|j� dS )zz L{StrictPushProducer} raises L{ValueError} if C{resumeProducing} is called in the initial state. N)r1 r; r( r/ r2 r r r � test_resume s z#StrictPushProducerTests.test_resumeN)r r! r" r# r1 r4 r5 r6 r8 r9 r: r< r= r> r? r@ rA rB rC rD rE rF rG r r r r r0 ^ s( r0 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � IOPumpTestsz Tests for L{IOPump}. c C sr t � }t|dd�}t � }t|dd�}t||||dd�}t� }||d�| }|j|dd� |�� | �d|j� dS ) a� Connect a couple protocol/transport pairs to an L{IOPump} and then pump it. Verify that a streaming producer registered with one of the transports does not receive invalid L{IPushProducer} method calls and ends in the right state. @param mode: C{u"server"} to test a producer registered with the server transport. C{u"client"} to test a producer registered with the client transport. T)ZisServerF)Zgreet)�server�client)Z streamingr% N)r r r r$ ZregisterProducer�pumpr r' ) r �modeZ serverProtoZserverTransportZ clientProtoZclientTransportrK r3 Zvictimr r r �_testStreamingProducer s( ���z"IOPumpTests._testStreamingProducerc C s | j dd� dS )z� L{IOPump.pump} does not call C{resumeProducing} on a L{IPushProducer} (stream producer) registered with the server transport. rI �rL N�rM r) r r r � test_serverStreamingProducer: s z(IOPumpTests.test_serverStreamingProducerc C s | j dd� dS )z� L{IOPump.pump} does not call C{resumeProducing} on a L{IPushProducer} (stream producer) registered with the client transport. rJ rN NrO r) r r r � test_clientStreamingProducerB s z(IOPumpTests.test_clientStreamingProducerN)r r! r" r# rM rP rQ r r r r rH s "rH N)r# Z __future__r r Zzope.interfacer Ztwisted.internet.interfacesr Ztwisted.internet.protocolr Ztwisted.test.iosimr r Ztwisted.trial.unittestr r r r$ r0 rH r r r r �<module> s , 7
Save File
Cancel