U jb*@sdZddlmZddlmZddlmZddlmZddl m Z m Z m Z ddl mZddlmZdd lmZmZmZmZmZdd lmZGd d d eZeeGd ddZGdddeZGdddeZGdddeZdS)z" Tests for L{twisted.test.iosim}. ) annotations)Literal) implementer) IPushProducer) ClientFactoryFactoryProtocol)Clock) MemoryReactor)ConnectionCompleter FakeTransportIOPumpconnectconnectedServerAndClient)TestCasec@s:eZdZdZddddZddddZdddd Zd S) FakeTransportTestsz% Tests for L{FakeTransport}. NonereturncCsHttd}ttd}||jt||jt||j|jdS)ze Each L{FakeTransport} receives a serial number that uniquely identifies it. TFN)r objectassertIsInstanceserialintZassertNotEqual)selfabr;/tmp/pip-unpacked-wheel-j5h_c8_q/twisted/test/test_iosim.pytest_connectionSerial s   z(FakeTransportTests.test_connectionSerialcCs>ttd}|d|dddg|d|jddS) zl L{FakeTransport.writeSequence} will write a sequence of L{bytes} to the transport. FabcdsabcdN)r rwriteZ writeSequence assertEqualjoinstreamrrrrrtest_writeSequence+s  z%FakeTransportTests.test_writeSequencecCs@ttd}|d||d|d|jddS)z L{FakeTransport.write} will accept writes after transport was closed, but the data will be silently discarded. Fsbeforesafterr#N)r rr$ZloseConnectionr%r&r'r(rrrtest_writeAfterClose7s    z'FakeTransportTests.test_writeAfterCloseN)__name__ __module__ __qualname____doc__rr)r*rrrrrs  rc@s>eZdZdZdZddddZddddZddd d Zd S) StrictPushProducerz An L{IPushProducer} implementation which produces nothing but enforces preconditions on its state transition methods. runningrrcCs|jdkrtdd|_dS)Nstoppedz)Cannot stop already-stopped IPushProducer_state ValueErrorrrrr stopProducingMs z StrictPushProducer.stopProducingcCs&|jdkrtd|jdd|_dS)Nr0z Cannot pause  IPushProducerpausedr2r5rrrpauseProducingRs z!StrictPushProducer.pauseProducingcCs&|jdkrtd|jdd|_dS)Nr8zCannot resume r7r0r2r5rrrresumeProducingWs z"StrictPushProducer.resumeProducingN)r+r,r-r.r3r6r9r:rrrrr/Ds r/c@s eZdZdZddddZddddZdddd Zddd d Zdd d ddZdd d ddZ dd d ddZ d dddZ d dddZ d dddZ d dddZd dddZd dddZd dd d!Zd dd"d#Zd dd$d%Zd dd&d'Zd dd(d)Zd dd*d+Zd,S)-StrictPushProducerTestsz* Tests for L{StrictPushProducer}. r/rcCstS)zp @return: A new L{StrictPushProducer} which has not been through any state changes. )r/r5rrr_initialbsz StrictPushProducerTests._initialcCst}||S)z@ @return: A new, stopped L{StrictPushProducer}. )r/r6rproducerrrr_stoppedisz StrictPushProducerTests._stoppedcCst}||S)z? @return: A new, paused L{StrictPushProducer}. )r/r9r=rrr_pausedqszStrictPushProducerTests._pausedcCst}|||S)zY @return: A new L{StrictPushProducer} which has been paused and resumed. )r/r9r:r=rrr_resumedysz StrictPushProducerTests._resumedr)r>rcCs||jddS)z Assert that the given producer is in the stopped state. @param producer: The producer to verify. @type producer: L{StrictPushProducer} r1Nr%r3r=rrr assertStoppedsz%StrictPushProducerTests.assertStoppedcCs||jddS)z Assert that the given producer is in the paused state. @param producer: The producer to verify. @type producer: L{StrictPushProducer} r8NrBr=rrr assertPausedsz$StrictPushProducerTests.assertPausedcCs||jddS)z Assert that the given producer is in the running state. @param producer: The producer to verify. @type producer: L{StrictPushProducer} r0NrBr=rrr assertRunningsz%StrictPushProducerTests.assertRunningcCs|t|jdS)zz L{StrictPushProducer.stopProducing} raises L{ValueError} if called when the producer is stopped. N) assertRaisesr4r?r6r5rrrtest_stopThenStopsz)StrictPushProducerTests.test_stopThenStopcCs|t|jdS)z{ L{StrictPushProducer.pauseProducing} raises L{ValueError} if called when the producer is stopped. N)rFr4r?r9r5rrrtest_stopThenPausesz*StrictPushProducerTests.test_stopThenPausecCs|t|jdS)z| L{StrictPushProducer.resumeProducing} raises L{ValueError} if called when the producer is stopped. N)rFr4r?r:r5rrrtest_stopThenResumesz+StrictPushProducerTests.test_stopThenResumecCs|}|||dS)zn L{StrictPushProducer} is stopped if C{stopProducing} is called on a paused producer. N)r@r6rCr=rrrtest_pauseThenStopsz*StrictPushProducerTests.test_pauseThenStopcCs|}|t|jdS)zs L{StrictPushProducer.pauseProducing} raises L{ValueError} if called on a paused producer. N)r@rFr4r9r=rrrtest_pauseThenPausesz+StrictPushProducerTests.test_pauseThenPausecCs|}|||dS)zp L{StrictPushProducer} is resumed if C{resumeProducing} is called on a paused producer. N)r@r:rEr=rrrtest_pauseThenResumesz,StrictPushProducerTests.test_pauseThenResumecCs|}|||dS)zo L{StrictPushProducer} is stopped if C{stopProducing} is called on a resumed producer. N)rAr6rCr=rrrtest_resumeThenStopsz+StrictPushProducerTests.test_resumeThenStopcCs|}|||dS)zo L{StrictPushProducer} is paused if C{pauseProducing} is called on a resumed producer. N)rAr9rDr=rrrtest_resumeThenPausesz,StrictPushProducerTests.test_resumeThenPausecCs|}|t|jdS)zu L{StrictPushProducer.resumeProducing} raises L{ValueError} if called on a resumed producer. N)rArFr4r:r=rrrtest_resumeThenResumesz-StrictPushProducerTests.test_resumeThenResumecCs|}|||dS)zn L{StrictPushProducer} is stopped if C{stopProducing} is called in the initial state. N)r<r6rCr=rrr test_stopsz!StrictPushProducerTests.test_stopcCs|}|||dS)zn L{StrictPushProducer} is paused if C{pauseProducing} is called in the initial state. N)r<r9rDr=rrr test_pausesz"StrictPushProducerTests.test_pausecCs|}|t|jdS)zz L{StrictPushProducer} raises L{ValueError} if C{resumeProducing} is called in the initial state. N)r<rFr4r:r=rrr test_resumesz#StrictPushProducerTests.test_resumeN)r+r,r-r.r<r?r@rArCrDrErGrHrIrJrKrLrMrNrOrPrQrRrrrrr;]s(          r;c@sJeZdZdZdddddZdddd Zddd d Zddd d ZdS) IOPumpTestsz Tests for L{IOPump}. zLiteral[('server', 'client')]r)moderc Csrt}t|dd}t}t|dd}t||||dd}t}||d|}|j|dd||d|jdS) 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)serverclient)Z streamingr0N)rr rr/ZregisterProducerpumpr%r3) rrTZ serverProtoZserverTransportZ clientProtoZclientTransportrWr>Zvictimrrr_testStreamingProducers(   z"IOPumpTests._testStreamingProducerrcCs|jdddS)z L{IOPump.pump} does not call C{resumeProducing} on a L{IPushProducer} (stream producer) registered with the server transport. rUrTNrXr5rrrtest_serverStreamingProducer)sz(IOPumpTests.test_serverStreamingProducercCs|jdddS)z L{IOPump.pump} does not call C{resumeProducing} on a L{IPushProducer} (stream producer) registered with the client transport. rVrYNrZr5rrrtest_clientStreamingProducer0sz(IOPumpTests.test_clientStreamingProducercsRgt}ttt|d\}}}|dfdd|||dS)zE L{IOPump.pump} advances time in the given L{Clock}. )clockrcs dS)NT)appendrZ time_passedrr>r#z/IOPumpTests.test_timeAdvances..N)r rrZ callLaterZ assertFalserW assertTrue)rr]_rWrr_rtest_timeAdvances7s zIOPumpTests.test_timeAdvancesN)r+r,r-r.rXr[r\rcrrrrrSs #rSc@s,eZdZdZddddZddddZdS) ConnectionCompleterTestszr L{ConnectionCompleter} can connect a connecting client and a listening server in a L{MemoryReactor}. rrcCstt}||ddS)z| When no connection for its reactor is pending, L{ConnectionCompleter.succeedOnce} returns C{None}. N)r r ZassertIs succeedOnce)rccrrrtest_noConnectionJs z*ConnectionCompleterTests.test_noConnectioncCsHt}|dtt|ddttt|}|| t dS)z When a connection to a matched host and port is pending, L{ConnectionCompleter} returns an L{IOPump} connecting them. Pz 127.0.0.1N) r Z listenTCPrZ forProtocolrZ connectTCPrr rrer )rZmemrfrrrtest_someConnectionRs z,ConnectionCompleterTests.test_someConnectionN)r+r,r-r.rgrirrrrrdDsrdN)r. __future__rtypingrZzope.interfacerZtwisted.internet.interfacesrZtwisted.internet.protocolrrrZtwisted.internet.taskr Ztwisted.internet.testingr Ztwisted.test.iosimr r r rrZtwisted.trial.unittestrrr/r;rSrdrrrrs        )%C