YfAXdZddlmZddlmZddlmZddlmZddl m Z ddl m Z m Z mZmZmZmZmZmZmZddlmZmZdd lmZmZmZmZdd lmZdd lm Z Gd d e Z!Gdde Z"GddZ#Gdde Z$GddeZ%Gdde Z&dS)z( Tests for L{twisted.internet.testing}. ) annotations)Callable) verifyObject)Protocol) IPv4Address) IAddress IConnector IConsumerIListeningPort IPushProducer IReactorSSL IReactorTCP IReactorUNIX ITransport) ClientFactoryFactory) MemoryReactorNonStreamingProducerRaisingMemoryReactorStringTransport)namedAny)TestCaseceZdZdZddZddZddZddZddZdd Z dd Z dd Z dd Z dd Z ddZddZddZddZddZddZddZddZddZdS)StringTransportTestsz@ Tests for L{twisted.internet.testing.StringTransport}. returnNonec,t|_dSN)r transportselfs T/var/www/html/env/lib/python3.11/site-packages/twisted/internet/test/test_testing.pysetUpzStringTransportTests.setUp+s(**c|tt|j|tt|j|tt |jdS)zq L{StringTransport} instances provide L{ITransport}, L{IPushProducer}, and L{IConsumer}. N) assertTruerrrr r r s r"test_interfacesz$StringTransportTests.test_interfaces.sd  Z@@AAA  ]DNCCDDD  Y??@@@@@r$ct}t}|j||||jj|||jj|dS)zz L{StringTransport.registerProducer} records the arguments supplied to it as instance attributes. N)objectrregisterProducerassertIsproducer streamingr!r,r-s r"test_registerProducerz*StringTransportTests.test_registerProducer7sf 88HH  '')<<< dn-x888 dn. :::::r$c<t}|j|d|t|jjtd||jj|||jjdS)zy L{StringTransport.registerProducer} raises L{RuntimeError} if a producer is already registered. TFN) r)rr* assertRaises RuntimeErrorr+r,r&r-)r!r,s r"test_disallowedRegisterProducerz4StringTransportTests.test_disallowedRegisterProducerBs 88 ''$777  $.9688U    dn-x888 011111r$ct}t}|j|d|j||jj|j|d||jj|||jjdS)z L{StringTransport.unregisterProducer} causes the transport to forget about the registered producer and makes it possible to register a new one. FTN) r)rr*unregisterProducer assertIsNoner,r+r&r-)r! oldProducer newProducers r"test_unregisterProducerz,StringTransportTests.test_unregisterProducerOs hh hh  '' U;;; ))+++ $.1222 '' T::: dn-{;;; 011111r$cP|t|jjdS)z L{StringTransport.unregisterProducer} raises L{RuntimeError} if called when no producer is registered. N)r1r2rr5r s r"test_invalidUnregisterProducerz3StringTransportTests.test_invalidUnregisterProducer^s% ,(IJJJJJr$cF||jjddS)zO L{StringTransport.producerState} is initially C{'producing'}. producingN) assertEqualr producerStater s r"test_initialProducerStatez.StringTransportTests.test_initialProducerStatees% 5{CCCCCr$cx|j||jjddS)zy L{StringTransport.pauseProducing} changes the C{producerState} of the transport to C{'paused'}. pausedN)rpauseProducingr>r?r s r"test_pauseProducingz(StringTransportTests.test_pauseProducingks9 %%''' 5x@@@@@r$c|j|j||jjddS)z} L{StringTransport.resumeProducing} changes the C{producerState} of the transport to C{'producing'}. r=N)rrCresumeProducingr>r?r s r"test_resumeProducingz)StringTransportTests.test_resumeProducingssM %%''' &&((( 5{CCCCCr$cx|j||jjddS)z{ L{StringTransport.stopProducing} changes the C{'producerState'} of the transport to C{'stopped'}. stoppedN)r stopProducingr>r?r s r"test_stopProducingz'StringTransportTests.test_stopProducing|s9 $$&&& 5yAAAAAr$c|j|t|jjdS)zu L{StringTransport.pauseProducing} raises L{RuntimeError} if the transport has been stopped. N)rrJr1r2rCr s r" test_stoppedTransportCannotPausez5StringTransportTests.test_stoppedTransportCannotPauses9 $$&&& ,(EFFFFFr$c|j|t|jjdS)zv L{StringTransport.resumeProducing} raises L{RuntimeError} if the transport has been stopped. N)rrJr1r2rFr s r"!test_stoppedTransportCannotResumez6StringTransportTests.test_stoppedTransportCannotResumes9 $$&&& ,(FGGGGGr$c|j|t|jjdS)zz L{StringTransport.pauseProducing} raises L{RuntimeError} if the transport is being disconnected. N)rloseConnectionr1r2rCr s r"&test_disconnectingTransportCannotPausez;StringTransportTests.test_disconnectingTransportCannotPauses9 %%''' ,(EFFFFFr$c|j|t|jjdS)z{ L{StringTransport.resumeProducing} raises L{RuntimeError} if the transport is being disconnected. N)rrQr1r2rFr s r"'test_disconnectingTransportCannotResumez (( o':::BBDDgNNNNNr$c|t}||tdS)z If no host address is passed to L{StringTransport.__init__}, an L{IPv4Address} is returned from L{StringTransport.getHost}. N)rrZassertIsInstancerr[s r"test_defaultHostAddressz,StringTransportTests.test_defaultHostAddress7 "##++-- g{33333r$c|t}||tdS)z If no peer address is passed to L{StringTransport.__init__}, an L{IPv4Address} is returned from L{StringTransport.getPeer}. N)rr`rcrr[s r"test_defaultPeerAddressz,StringTransportTests.test_defaultPeerAddressrer$Nrr)__name__ __module__ __qualname____doc__r#r'r/r3r9r;r@rDrGrKrMrOrRrTrXr]rardrgr$r"rr&s++++AAAA ; ; ; ; 2 2 2 2 2 2 2 2KKKKDDDD AAAADDDDBBBBGGGGHHHHGGGGHHHH6666CCCCOOOO4444444444r$rcBeZdZdZd dZd dZd dZd dZd dZd d Z d S) ReactorTestszA Tests for L{MemoryReactor} and L{RaisingMemoryReactor}. rrct}tt|tt|tt|dS)zt L{MemoryReactor} provides all of the attributes described by the interfaces it advertises. N)rrrr r)r! memoryReactors r"test_memoryReactorProvidesz'ReactorTests.test_memoryReactorProvidessC & [-000[-000\=11111r$ct}tt|tt|tt|dS)z{ L{RaisingMemoryReactor} provides all of the attributes described by the interfaces it advertises. N)rrrr r)r!raisingReactors r"test_raisingReactorProvidesz(ReactorTests.test_raisingReactorProvidessE .//[.111[.111\>22222r$ct}|ddt|ddtdfD]v}t t ||}t t|||j d||j dw| dt}t t ||}t t|||j ddS)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) r connectTCPr connectSSLrr getDestinationrr>hostport connectUNIXname)r!rq connectorr\s r"test_connectDestinationz$ReactorTests.test_connectDestinations$&  $ $%7} O O  $ $%7}PT U U  1 1I Y / / /..00G 7 + + +   W\+= > > >   W\4 0 0 0 0!--m]__MM Z+++**,,Xw''' }55555r$ct}|dt|dtdfD]v}t t ||}t t|||j d||j dw| dt}t t ||}t t|||j ddS)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) r listenTCPr listenSSLrr rZrr>rzr{ listenUNIXr})r!rqr{r\s r"test_listenDefaultHostz#ReactorTests.test_listenDefaultHosts&  # #D')) 4 4  # #D'))T : :  1 1D  . . .llnnG 7 + + +   W\9 5 5 5   W\4 0 0 0 0''(:GIIFF^T***,,..Xw''' '9:::::r$c^t}t}|||||||g||||gdS)z> Adding, removing, and listing readers works. N)r)r addReaderr> getReaders removeReader)r!readerreactors r" test_readerszReactorTests.test_readers//&!!!&!!! ++--x888V$$$ ++--r22222r$c^t}t}|||||||g||||gdS)z> Adding, removing, and listing writers works. N)r)r addWriterr> getWriters removeWriter)r!writerrs r" test_writerszReactorTests.test_writers'rr$Nrh) rirjrkrlrrrurrrrrmr$r"roros222233336666.;;;;23333 333333r$roc2eZdZdZddZdd Zdd Zdd ZdS) TestConsumerzP A very basic test consumer for use with the NonStreamingProducerTests. rrc0g|_d|_d|_dSr)writesr,producerStreamingr s r"__init__zTestConsumer.__init__=s#% $ .2r$r,r)r-boolc"||_||_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. Nr,rr.s r"r*zTestConsumer.registerProducerBs! !*r$c"d|_d|_dS)zC Forget the producer we had previously registered. Nrr s r"r5zTestConsumer.unregisterProducerLs !%r$databytesc:|j|dS)zz Some data was written to the consumer: stores it for later use. @param data: The data to write. N)rappend)r!rs r"writezTestConsumer.writeSs 4     r$Nrh)r,r)r-rrr)rrrr)rirjrkrlrr*r5rrmr$r"rr8sn3333 ++++&&&&!!!!!!r$rc"eZdZdZddZddZdS)NonStreamingProducerTestszF Tests for the L{NonStreamingProducer} to validate behaviour. rrcrt}t|}||d||j|||j|||jtdD]}| gd}| |j| |j| |j| |j || t|j dS)z When the L{NonStreamingProducer} has resumeProducing called 10 times, it writes the counter each time and then fails. F ) 0123456789N)rrr*r+r,consumerrVrrangerFr6r>rr1r2)r!rr,_expectedWritess r"test_producesOnly10Timesz2NonStreamingProducerTests.test_producesOnly10Timesas+  >>'11!!(E222 h'222 h'222 3444r ' 'A  $ $ & & & &VUU (+,,, (4555 (+,,, .999 ,(@AAAAAr$ct}t|}||d||t |jdS)zb When the L{NonStreamingProducer} is paused, it raises a L{RuntimeError}. FN)rrr*rFr1r2rC)r!rr,s r"test_cannotPauseProductionz4NonStreamingProducerTests.test_cannotPauseProduction{sc  >>'11!!(E222   """ ,(?@@@@@r$Nrh)rirjrkrlrrrmr$r"rr\sPBBBB4 A A A A A Ar$rc&eZdZeddZdS) _SupportsNamerstrcdSrrmr s r"riz_SupportsName.__name__s r$N)rr)rirjrkpropertyrmr$r"rrs2    X   r$rceZdZdZddZdd Zdd Zdd Zdd Zdd Z ddZ ddZ ddZ ddZ ddZddZddZddZddZdS)DeprecationTestsz8 Deprecations in L{twisted.test.proto_helpers}. testCallable[..., object]objrrrcdd|j}||g}|t|dd|dt ||||dd||t|dS)Nztwisted.internet.testing.rcategorymessage)ri flushWarningsr>DeprecationWarninglenassertInr+r)r!rrnew_pathwarningss r"helperzDeprecationTests.helpers=s|==%%tf-- +Xa[-DEEE CMM*** h I 6777 c8H--.....r$cHddlm}||j|dS)Nr)AccumulatingProtocol)twisted.test.proto_helpersrrtest_accumulatingProtocol)r!rs r"rz*DeprecationTests.test_accumulatingProtocol2CCCCCC D24HIIIIIr$cHddlm}||j|dS)Nr)LineSendingProtocol)rrrtest_lineSendingProtocol)r!rs r"rz)DeprecationTests.test_lineSendingProtocols2BBBBBB D13FGGGGGr$cHddlm}||j|dS)Nr)FakeDatagramTransport)rrrtest_fakeDatagramTransport)r!rs r"rz+DeprecationTests.test_fakeDatagramTransports2DDDDDD D35JKKKKKr$cHddlm}||j|dS)Nr)r)rrrtest_stringTransport)r!rs r"rz%DeprecationTests.test_stringTransports1>>>>>> D-?????r$cHddlm}||j|dS)Nr) StringTransportWithDisconnection)rrr%test_stringTransportWithDisconnection)r!rs r"rz6DeprecationTests.test_stringTransportWithDisconnections=OOOOOO  68X     r$cHddlm}||j|dS)Nr)StringIOWithoutClosing)rrrtest_stringIOWithoutClosing)r!rs r"rz,DeprecationTests.test_stringIOWithoutClosings2EEEEEE D46LMMMMMr$cHddlm}||j|dS)Nr)_FakeConnector)rrrtest__fakeConnector)r!rs r"rz$DeprecationTests.test__fakeConnectors1====== D,n=====r$cHddlm}||j|dS)Nr) _FakePort)rrrtest__fakePort)r!rs r"rzDeprecationTests.test__fakePorts1888888 D'33333r$cHddlm}||j|dS)Nr)r)rrrtest_memoryReactor)r!rs r"rz#DeprecationTests.test_memoryReactors1<<<<<< D+];;;;;r$cHddlm}||j|dS)Nr)MemoryReactorClock)rrrtest_memoryReactorClock)r!rs r"rz(DeprecationTests.test_memoryReactorClocks2AAAAAA D02DEEEEEr$cHddlm}||j|dS)Nr)r)rrrtest_raisingMemoryReactor)r!rs r"rz*DeprecationTests.test_raisingMemoryReactorrr$cHddlm}||j|dS)Nr)r)rrrtest_nonStreamingProducer)r!rs r"rz*DeprecationTests.test_nonStreamingProducerrr$cHddlm}||j|dS)Nr)waitUntilAllDisconnected)rrrtest_waitUntilAllDisconnected)r!rs r"rz.DeprecationTests.test_waitUntilAllDisconnecteds2GGGGGG D68PQQQQQr$cHddlm}||j|dS)Nr)EventLoggingObserver)rrrtest_eventLoggingObserver)r!rs r"rz*DeprecationTests.test_eventLoggingObserverrr$N)rrrrrrrh)rirjrkrlrrrrrrrrrrrrrrrrmr$r"rrst////JJJJ HHHH LLLL @@@@     NNNN >>>> 4444 <<<< FFFF JJJJ JJJJ RRRR JJJJJJr$rN)'rl __future__rtypingrzope.interface.verifyrtyping_extensionsrtwisted.internet.addressrtwisted.internet.interfacesrr r r r r rrrtwisted.internet.protocolrrtwisted.internet.testingrrrrtwisted.python.reflectrtwisted.trial.unittestrrrorrrrrmr$r"rs#"""""......&&&&&&000000                      =<<<<<<< ,+++++++++++e4e4e4e4e48e4e4e4Pg3g3g3g3g38g3g3g3T!!!!!!!!!!!!!!!!H+A+A+A+A+A+A+A+A\     H   SJSJSJSJSJxSJSJSJSJSJr$