o ib*@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.eZdZdZd ddZd ddZd dd Zd S) FakeTransportTestsz% Tests for L{FakeTransport}. returnNonecCsHttd}ttd}||jt||jt||j|jdS)ze Each L{FakeTransport} receives a serial number that uniquely identifies it. TFN)r objectassertIsInstanceserialintZassertNotEqual)selfabr]C:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/test/test_iosim.pytest_connectionSerial s  z(FakeTransportTests.test_connectionSerialcCs<ttd}|d|gd|d|jddS)zl L{FakeTransport.writeSequence} will write a sequence of L{bytes} to the transport. Fa)bcdsabcdN)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_writeAfterCloseNrr)__name__ __module__ __qualname____doc__rr(r)rrrrrs    rc@s2eZdZdZdZd ddZd ddZd d d Zd S) StrictPushProducerz An L{IPushProducer} implementation which produces nothing but enforces preconditions on its state transition methods. runningrrcCs|jdkr tdd|_dS)Nstoppedz)Cannot stop already-stopped IPushProducer_state ValueErrorrrrr stopProducingMs  z StrictPushProducer.stopProducingcC&|jdkrtd|jdd|_dS)Nr0z Cannot pause  IPushProducerpausedr2r5rrrpauseProducingR  z!StrictPushProducer.pauseProducingcCr7)Nr9zCannot resume r8r0r2r5rrrresumeProducingWr;z"StrictPushProducer.resumeProducingNr*)r+r,r-r.r3r6r:r<rrrrr/Ds   r/c@seZdZdZd-ddZd-ddZd-dd Zd-d d Zd.ddZd.ddZ d.ddZ d/ddZ d/ddZ d/ddZ d/ddZd/ddZd/ddZd/d d!Zd/d"d#Zd/d$d%Zd/d&d'Zd/d(d)Zd/d*d+Zd,S)0StrictPushProducerTestsz* Tests for L{StrictPushProducer}. rr/cCstS)zp @return: A new L{StrictPushProducer} which has not been through any state changes. N)r/r5rrr_initialbsz StrictPushProducerTests._initialcCt}||S)z@ @return: A new, stopped L{StrictPushProducer}. N)r/r6rproducerrrr_stoppediz StrictPushProducerTests._stoppedcCr?)z? @return: A new, paused L{StrictPushProducer}. N)r/r:r@rrr_pausedqrCzStrictPushProducerTests._pausedcCst}|||S)zY @return: A new L{StrictPushProducer} which has been paused and resumed. N)r/r:r<r@rrr_resumedysz StrictPushProducerTests._resumedrArcC||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 assertStoppedz%StrictPushProducerTests.assertStoppedcCrF)z Assert that the given producer is in the paused state. @param producer: The producer to verify. @type producer: L{StrictPushProducer} r9NrGr@rrr assertPausedrIz$StrictPushProducerTests.assertPausedcCrF)z Assert that the given producer is in the running state. @param producer: The producer to verify. @type producer: L{StrictPushProducer} r0NrGr@rrr assertRunningrIz%StrictPushProducerTests.assertRunningcC|t|jdS)zz L{StrictPushProducer.stopProducing} raises L{ValueError} if called when the producer is stopped. N) assertRaisesr4rBr6r5rrrtest_stopThenStopz)StrictPushProducerTests.test_stopThenStopcCrL)z{ L{StrictPushProducer.pauseProducing} raises L{ValueError} if called when the producer is stopped. N)rMr4rBr:r5rrrtest_stopThenPauserOz*StrictPushProducerTests.test_stopThenPausecCrL)z| L{StrictPushProducer.resumeProducing} raises L{ValueError} if called when the producer is stopped. N)rMr4rBr<r5rrrtest_stopThenResumerOz+StrictPushProducerTests.test_stopThenResumecC|}|||dS)zn L{StrictPushProducer} is stopped if C{stopProducing} is called on a paused producer. N)rDr6rHr@rrrtest_pauseThenStopz*StrictPushProducerTests.test_pauseThenStopcC|}|t|jdS)zs L{StrictPushProducer.pauseProducing} raises L{ValueError} if called on a paused producer. N)rDrMr4r:r@rrrtest_pauseThenPausez+StrictPushProducerTests.test_pauseThenPausecCrR)zp L{StrictPushProducer} is resumed if C{resumeProducing} is called on a paused producer. N)rDr<rKr@rrrtest_pauseThenResumerTz,StrictPushProducerTests.test_pauseThenResumecCrR)zo L{StrictPushProducer} is stopped if C{stopProducing} is called on a resumed producer. N)rEr6rHr@rrrtest_resumeThenStoprTz+StrictPushProducerTests.test_resumeThenStopcCrR)zo L{StrictPushProducer} is paused if C{pauseProducing} is called on a resumed producer. N)rEr:rJr@rrrtest_resumeThenPauserTz,StrictPushProducerTests.test_resumeThenPausecCrU)zu L{StrictPushProducer.resumeProducing} raises L{ValueError} if called on a resumed producer. N)rErMr4r<r@rrrtest_resumeThenResumerWz-StrictPushProducerTests.test_resumeThenResumecCrR)zn L{StrictPushProducer} is stopped if C{stopProducing} is called in the initial state. N)r>r6rHr@rrr test_stoprTz!StrictPushProducerTests.test_stopcCrR)zn L{StrictPushProducer} is paused if C{pauseProducing} is called in the initial state. N)r>r:rJr@rrr test_pauserTz"StrictPushProducerTests.test_pausecCrU)zz L{StrictPushProducer} raises L{ValueError} if C{resumeProducing} is called in the initial state. N)r>rMr4r<r@rrr test_resumerWz#StrictPushProducerTests.test_resumeN)rr/)rAr/rrr*)r+r,r-r.r>rBrDrErHrJrKrNrPrQrSrVrXrYrZr[r\r]r^rrrrr=]s*           r=c@s8eZdZdZdddZddd Zdd d Zdd d ZdS) IOPumpTestsz Tests for L{IOPump}. modeLiteral['server', 'client']rrc 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) rr`Z serverProtoZserverTransportZ clientProtoZclientTransportrdrAZvictimrrr_testStreamingProducers(  z"IOPumpTests._testStreamingProducercC|jdddS)z L{IOPump.pump} does not call C{resumeProducing} on a L{IPushProducer} (stream producer) registered with the server transport. rbr`Nrer5rrrtest_serverStreamingProducer)z(IOPumpTests.test_serverStreamingProducercCrf)z L{IOPump.pump} does not call C{resumeProducing} on a L{IPushProducer} (stream producer) registered with the client transport. rcrgNrhr5rrrtest_clientStreamingProducer0rjz(IOPumpTests.test_clientStreamingProducercsRgt}ttt|d\}}}|dfdd|||dS)zE L{IOPump.pump} advances time in the given L{Clock}. )clockrcs dS)NT)appendrZ time_passedrr>s z/IOPumpTests.test_timeAdvances..N)r rrZ callLaterZ assertFalserd assertTrue)rrl_rdrrnrtest_timeAdvances7s zIOPumpTests.test_timeAdvancesN)r`rarrr*)r+r,r-r.rerirkrrrrrrr_s   # r_c@s$eZdZdZd ddZd ddZdS) 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 rrtr )rZmemrurrrtest_someConnectionRs z,ConnectionCompleterTests.test_someConnectionNr*)r+r,r-r.rvrxrrrrrsDs rsN)r. __future__rtypingrZzope.interfacerZtwisted.internet.interfacesrZtwisted.internet.protocolrrrZtwisted.internet.taskr Ztwisted.internet.testingr Ztwisted.test.iosimr r r rrZtwisted.trial.unittestrrr/r=r_rsrrrrs"       )%C