U  W[Ð3ã@sÀdZddlmZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZmZmZmZddlmZmZGdd „d eƒZGd d „d eƒZGd d „d eƒZGdd„deƒZdS)z* Tests for L{twisted.test.proto_helpers}. é)Ú verifyObject) Ú ITransportÚ IPushProducerÚ IConsumerÚ IReactorTCPÚ IReactorSSLÚ IReactorUNIXÚIAddressÚIListeningPortÚ IConnector)Ú IPv4Address)ÚTestCase)ÚStringTransportÚ MemoryReactorÚRaisingMemoryReactorÚNonStreamingProducer)Ú ClientFactoryÚFactoryc@s¨eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(S))ÚStringTransportTestszB Tests for L{twisted.test.proto_helpers.StringTransport}. cCs tƒ|_dS©N)rÚ transport©Úself©rúC/usr/lib/python3/dist-packages/twisted/test/test_stringtransport.pyÚsetUpszStringTransportTests.setUpcCs:| tt|jƒ¡| tt|jƒ¡| tt|jƒ¡dS)zq L{StringTransport} instances provide L{ITransport}, L{IPushProducer}, and L{IConsumer}. N)Ú assertTruerrrrrrrrrÚtest_interfacessz$StringTransportTests.test_interfacescCs>tƒ}tƒ}|j ||¡| |jj|¡| |jj|¡dS)zz L{StringTransport.registerProducer} records the arguments supplied to it as instance attributes. N)ÚobjectrÚregisterProducerÚassertIsÚproducerÚ streaming©rr!r"rrrÚtest_registerProducer&s z*StringTransportTests.test_registerProducercCsLtƒ}|j |d¡| t|jjtƒd¡| |jj|¡| |jj¡dS)zy L{StringTransport.registerProducer} raises L{RuntimeError} if a producer is already registered. TFN) rrrÚ assertRaisesÚ RuntimeErrorr r!rr")rr!rrrÚtest_disallowedRegisterProducer2sÿz4StringTransportTests.test_disallowedRegisterProducercCsbtƒ}tƒ}|j |d¡|j ¡| |jj¡|j |d¡| |jj|¡| |jj¡dS)z± L{StringTransport.unregisterProducer} causes the transport to forget about the registered producer and makes it possible to register a new one. FTN) rrrÚunregisterProducerÚ assertIsNoner!r rr")rZ oldProducerZ newProducerrrrÚtest_unregisterProducer?s z,StringTransportTests.test_unregisterProducercCs| t|jj¡dS)z€ L{StringTransport.unregisterProducer} raises L{RuntimeError} if called when no producer is registered. N)r%r&rr(rrrrÚtest_invalidUnregisterProducerOsz3StringTransportTests.test_invalidUnregisterProducercCs| |jjd¡dS)zO L{StringTransport.producerState} is initially C{'producing'}. Ú producingN)Ú assertEqualrÚ producerStaterrrrÚtest_initialProducerStateWsz.StringTransportTests.test_initialProducerStatecCs|j ¡| |jjd¡dS)zy L{StringTransport.pauseProducing} changes the C{producerState} of the transport to C{'paused'}. ZpausedN)rÚpauseProducingr-r.rrrrÚtest_pauseProducing^s z(StringTransportTests.test_pauseProducingcCs(|j ¡|j ¡| |jjd¡dS)z} L{StringTransport.resumeProducing} changes the C{producerState} of the transport to C{'producing'}. r,N)rr0ÚresumeProducingr-r.rrrrÚtest_resumeProducinggs  z)StringTransportTests.test_resumeProducingcCs|j ¡| |jjd¡dS)z{ L{StringTransport.stopProducing} changes the C{'producerState'} of the transport to C{'stopped'}. ZstoppedN)rÚ stopProducingr-r.rrrrÚtest_stopProducingqs z'StringTransportTests.test_stopProducingcCs|j ¡| t|jj¡dS)zu L{StringTransport.pauseProducing} raises L{RuntimeError} if the transport has been stopped. N)rr4r%r&r0rrrrÚ test_stoppedTransportCannotPausezs z5StringTransportTests.test_stoppedTransportCannotPausecCs|j ¡| t|jj¡dS)zv L{StringTransport.resumeProducing} raises L{RuntimeError} if the transport has been stopped. N)rr4r%r&r2rrrrÚ!test_stoppedTransportCannotResumeƒs z6StringTransportTests.test_stoppedTransportCannotResumecCs|j ¡| t|jj¡dS)zz L{StringTransport.pauseProducing} raises L{RuntimeError} if the transport is being disconnected. N)rÚloseConnectionr%r&r0rrrrÚ&test_disconnectingTransportCannotPauseŒs z;StringTransportTests.test_disconnectingTransportCannotPausecCs|j ¡| t|jj¡dS)z{ L{StringTransport.resumeProducing} raises L{RuntimeError} if the transport is being disconnected. N)rr8r%r&r2rrrrÚ'test_disconnectingTransportCannotResume•s zrrrÚtest_specifiedPeerAddress±s  ÿz.StringTransportTests.test_specifiedPeerAddresscCstƒ ¡}| |t¡dS)z• If no host address is passed to L{StringTransport.__init__}, an L{IPv4Address} is returned from L{StringTransport.getHost}. N)rr=ÚassertIsInstancer r>rrrÚtest_defaultHostAddress»s z,StringTransportTests.test_defaultHostAddresscCstƒ ¡}| |t¡dS)z• If no peer address is passed to L{StringTransport.__init__}, an L{IPv4Address} is returned from L{StringTransport.getPeer}. N)rrArCr r>rrrÚtest_defaultPeerAddressÄs z,StringTransportTests.test_defaultPeerAddressN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr$r'r*r+r/r1r3r5r6r7r9r:r<r@rBrDrErrrrrs(              rc@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)Ú ReactorTestszA Tests for L{MemoryReactor} and L{RaisingMemoryReactor}. cCs(tƒ}tt|ƒtt|ƒtt|ƒdS)zt L{MemoryReactor} provides all of the attributes described by the interfaces it advertises. N)rrrrr)rÚ memoryReactorrrrÚtest_memoryReactorProvidesÓs  z'ReactorTests.test_memoryReactorProvidescCs(tƒ}tt|ƒtt|ƒtt|ƒdS)z{ L{RaisingMemoryReactor} provides all of the attributes described by the interfaces it advertises. N)rrrrr)rZraisingReactorrrrÚtest_raisingReactorProvidesÞs  z(ReactorTests.test_raisingReactorProvidescCs¢tƒ}| ddtƒ¡| ddtƒd¡fD]<}tt|ƒ| ¡}tt|ƒ| |j d¡| |j d¡q(|  dtƒ¡}tt|ƒ| ¡}tt|ƒ| |j d¡dS)a L{MemoryReactor.connectTCP}, L{MemoryReactor.connectSSL}, and L{MemoryReactor.connectUNIX} will return an L{IConnector} whose C{getDestination} method returns an L{IAddress} with attributes which reflect the values passed. ztest.example.comi Ns /fake/path) rZ connectTCPrZ connectSSLrr ZgetDestinationr r-ÚhostÚportZ connectUNIXÚname)rrKZ connectorr?rrrÚtest_connectDestinationés.ÿþþ    z$ReactorTests.test_connectDestinationcCsžtƒ}| dtƒ¡| dtƒd¡fD]<}tt|ƒ| ¡}tt|ƒ| |j d¡| |j d¡q$|  dtƒ¡}tt|ƒ| ¡}tt|ƒ| |j d¡dS)a– L{MemoryReactor.listenTCP}, L{MemoryReactor.listenSSL} and L{MemoryReactor.listenUNIX} will return an L{IListeningPort} whose C{getHost} method returns an L{IAddress}; C{listenTCP} and C{listenSSL} will have a default host of C{'0.0.0.0'}, and a port that reflects the value passed, and C{listenUNIX} will have a name that reflects the path passed. i2 Nz0.0.0.0s/path/to/socket) rZ listenTCPrZ listenSSLrr r=r r-rNrOZ listenUNIXrP)rrKrOr?rrrÚtest_listenDefaultHosts  ÿ    z#ReactorTests.test_listenDefaultHostcCsPtƒ}tƒ}| |¡| |¡| | ¡|g¡| |¡| | ¡g¡dS)z> Adding, removing, and listing readers works. N)rrZ addReaderr-Z getReadersZ removeReader)rÚreaderÚreactorrrrÚ test_readerss   zReactorTests.test_readerscCsPtƒ}tƒ}| |¡| |¡| | ¡|g¡| |¡| | ¡g¡dS)z> Adding, removing, and listing writers works. N)rrZ addWriterr-Z getWritersZ removeWriter)rÚwriterrTrrrÚ test_writers+s   zReactorTests.test_writersN) rFrGrHrIrLrMrQrRrUrWrrrrrJÎs  rJc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) Ú TestConsumerzP A very basic test consumer for use with the NonStreamingProducerTests. cCsg|_d|_d|_dSr)Úwritesr!ÚproducerStreamingrrrrÚ__init__AszTestConsumer.__init__cCs||_||_dS)z× Registers a single producer with this consumer. Just keeps track of it. @param producer: The producer to register. @param streaming: Whether the producer is a streaming one or not. N©r!rZr#rrrrGszTestConsumer.registerProducercCsd|_d|_dS)zC Forget the producer we had previously registered. Nr\rrrrr(RszTestConsumer.unregisterProducercCs|j |¡dS)zz Some data was written to the consumer: stores it for later use. @param data: The data to write. N)rYÚappend)rÚdatarrrÚwriteZszTestConsumer.writeN)rFrGrHrIr[rr(r_rrrrrX=s  rXc@s eZdZdZdd„Zdd„ZdS)ÚNonStreamingProducerTestszF Tests for the L{NonStreamingProducer} to validate behaviour. c Cs´tƒ}t|ƒ}| |d¡| |j|¡| |j|¡| |j¡tdƒD] }|  ¡qJddddddd d d d g }|  |j¡|  |j¡|  |j¡|  |j |¡|  t|j ¡d S)z When the L{NonStreamingProducer} has resumeProducing called 10 times, it writes the counter each time and then fails. Fé ó0ó1ó2ó3ó4ó5ó6ó7ó8ó9N)rXrrr r!Úconsumerr;rZÚranger2r)r-rYr%r&)rrlr!Ú_ZexpectedWritesrrrÚtest_producesOnly10Timeshs0    ÿ   z2NonStreamingProducerTests.test_producesOnly10TimescCs4tƒ}t|ƒ}| |d¡| ¡| t|j¡dS)zb When the L{NonStreamingProducer} is paused, it raises a L{RuntimeError}. FN)rXrrr2r%r&r0)rrlr!rrrÚtest_cannotPauseProduction…s  z4NonStreamingProducerTests.test_cannotPauseProductionN)rFrGrHrIrorprrrrr`dsr`N)rIZzope.interface.verifyrZtwisted.internet.interfacesrrrrrrr r r Ztwisted.internet.addressr Ztwisted.trial.unittestr Ztwisted.test.proto_helpersrrrrZtwisted.internet.protocolrrrrJrrXr`rrrrÚs ,  ;o'