tfdZddlZddlZddlmZddlmZmZmZddl m Z ddl m Z m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZGddejZGddejZGddejZ Gddej!Z"GddejZ#Gddej!Z$Gddej!Z%GddZ&Gdde&ej'Z(Gd d!Z)Gd"d#ej!e)Z*Gd$d%e)Z+Gd&d'Z,Gd(d)e&ej-Z.Gd*d+ej!e+e,Z/Gd,d-e&ej0Z1Gd.d/ej!e+e,Z2Gd0d1e&ej3Z4Gd2d3ej!e+e,Z5Gd4d5Z6Gd6d7ejZ7Gd8d9ej!Z8Gd:d;ej9Z:dS)?? " "KK1q5K2G GHq!!!!   T[!* 5 5 5 5 6 6rs!twiddle1 twiddle2 pause twiddle3 )twiddle1twiddle2r:twiddle3c@tddD] }tj}tj}t |}|tj|tt|j |zdzD]/}|j ||z|dz|z}| |0| |j |j|d| |j|j dS)zo Test pause inside data receiving. It uses fake clock to see if pausing/resuming work. r#r\rN)r]rr^r Clockr/r_rr`rLpauseBufrbrc pauseOutput1r6advance pauseOutput2rrfrgr2rhrirjs r test_pausingzLineReceiverTests.test_pausings !B<< < > > MM!      T11: > > > > ? ?rs"twiddle1 twiddle2 stop more stuff )rlrmr<ctddD]}tj}t}|t j|tt|j|zdzD]/}|j||z|dz|z}| |0| |j |j dS)z- Test stop inside producing. r#r\N) r]rr^r/r_rr`rLstop_bufrbrc stop_outputr6res rtest_stopProducingz$LineReceiverTests.test_stopProducings!B<< ; ;K466A A  X1!44 5 5 53t}--s hello worldr?goodbyeN) r/rr^r_rr`rbrcr6)rrhrgs rtest_lineReceiverAsProducerz-LineReceiverTests.test_lineReceiverAsProducers} LL  0 2 2 -a00111 DEEE  JNNN     rcbGddtj}|}|jd||jd||jd|jd||jd||jdd S) z L{LineReceiver.clearLineBuffer} removes all buffered data and returns it as a C{bytes} and can be called from beneath C{dataReceived}. ceZdZdZdS)@LineReceiverTests.test_clearLineBuffer..ClearingReceivercF||_||_dSr)r clearLineBufferrMrs rr!zMLineReceiverTests.test_clearLineBuffer..ClearingReceiver.lineReceiveds    0022 rN)r(r)r*r!r-rrClearingReceiverrs# 3 3 3 3 3rrs foo bar bazfoosbar bazsquux squuxrN)r LineReceiverrbrcr rM)rrrs rtest_clearLineBufferz&LineReceiverTests.test_clearLineBuffers  3 3 3 3 3u1 3 3 3 $#%%2333 ///  666 k*** 000 ,,,,,rcNt}tj}|t j|t j}|d|z| d|zd |j dS)zC Test switching modes many times on the same data. sx xxrN) rrr^r_rr`sysgetrecursionlimitrbrcjoinr)rprotorGlimits rtest_stackRecursionz%LineReceiverTests.test_stackRecursion*s#$$!8::  X1)<<===%'' 7U?+++ sxx '<'<=====rctj}tj}|||d|jdzzdz||jdSzx C{LineReceiver} disconnects the transport if it receives a line longer than its C{MAX_LENGTH}. rr# rN r rrStringTransportr_rbrO assertTrue disconnectingrrrGs rtest_maximumLineLengthz(LineReceiverTests.test_maximumLineLength5u "$$!133  Y''' 45#3a#788CDDD  /00000rct}d|_tj}||d|jdz z}||||jdd||jdd|z||j| t|j d| ||j ddS)a~ C{LineReceiver} doesn't disconnect the transport when it receives a finished line as long as its C{MAX_LENGTH}, when the second-to-last packet ended with a pattern that could have been -- and turns out to have been -- the start of a delimiter, and that packet causes the total input to exceed C{MAX_LENGTH} + len(delimiter). r=rr#NrKr) r/rOrrr_rbr, assertFalserrcrLr6)rrrgr s r&test_maximumLineLengthPartialDelimiterz8LineReceiverTests.test_maximumLineLengthPartialDelimiter@s   ) + + Qu'!+, 4    5?3B3/000 5?233/$6777 ))) U^,,a000 u~a011111rcBtj}d|_tj}|||d|jz|jdt|jdz z| |j dS)aO C{LineReceiver} doesn't disconnect the transport it if receives a non-finished line whose length, counting the delimiter, is longer than its C{MAX_LENGTH} but shorter than its C{MAX_LENGTH} + len(delimiter). (When the first part that exceeds the max is the beginning of the delimiter.) s rNr#) r rr,rrr_rbrOrLrrrs r(test_notQuiteMaximumLineLengthUnfinishedz:LineReceiverTests.test_notQuiteMaximumLineLengthUnfinishedVs"$$"!133  Y'''  E$ $8R#eo:N:NQR:R8R(S S    011111rctj}d|_tj}||||d}||tdS)zf C{LineReceiver.dataReceived} forwards errors returned by C{rawDataReceived}. c tdS)Noops) RuntimeErrorr&s rz5LineReceiverTests.test_rawDataError..os\&-A-ArsdataN) r rr'rrr_rrbassertIsInstancer)rrrGwhys rtest_rawDataErrorz#LineReceiverTests.test_rawDataErroris "$$ A A!133  Y'''   )) c<00000rcntj}|t|jddS)z When L{LineReceiver.rawDataReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. fooN)r r assertRaisesNotImplementedErrorr'rrs r"test_rawDataReceivedNotImplementedz4LineReceiverTests.test_rawDataReceivedNotImplementedvs4 "$$ -u/DeLLLLLrcntj}|t|jddS)z When L{LineReceiver.lineReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. rN)r rrrr!rs rtest_lineReceivedNotImplementedz1LineReceiverTests.test_lineReceivedNotImplemented~s4 "$$ -u/A5IIIIIrN)r(r)r*r+rardrkrqrrrtrvrxryrzr{r}r~rrrrrrrrrrr-rrrWrWsU F$F 6 6 68H777L;-/L<<<"JKLLLOO??? :H555K ; ; ;    ---, > > > 1 1 1222,222& 1 1 1MMMJJJJJrrWc$eZdZdZdZdZdZdS)ExcessivelyLargeLineCatcherz Helper for L{LineReceiverLineLengthExceededTests}. @ivar longLines: A L{list} of L{bytes} giving the values C{lineLengthExceeded} has been called with. cg|_dSr) longLinesrs rr7z*ExcessivelyLargeLineCatcher.connectionMades rcdS)z/ Disregard any received lines. Nr-rs rr!z(ExcessivelyLargeLineCatcher.lineReceivedsrc:|j|dS)zF Record any data that exceeds the line length limits. N)rrr%s rrPz.ExcessivelyLargeLineCatcher.lineLengthExceededs  d#####rN)r(r)r*r+r7r!rPr-rrrrsK   $$$$$rrc<eZdZdZdZdZdZdZdZdZ dZ d S) #LineReceiverLineLengthExceededTestszO Tests for L{twisted.protocols.basic.LineReceiver.lineLengthExceeded}. ct|_d|j_tj|_|j|jdS)N)rrrOrrrGr_rs rsetUpz)LineReceiverLineLengthExceededTests.setUpsG022 ! &688 !!$.11111rcd|jjdzdzz}|j|||g|jjdS)z If more bytes than C{LineReceiver.MAX_LENGTH} arrive containing no line delimiter, all of the bytes are passed as a single string to L{LineReceiver.lineLengthExceeded}. rN)rrOrbrcrr excessives rtest_longUnendedLinez8LineReceiverLineLengthExceededTests.test_longUnendedLinesV DJ1A59:   *** )dj&:;;;;;rcd|jjdzdzz}|jd|jjz|z||g|jjdS)a* If L{LineReceiver.dataReceived} is called with bytes representing a short line followed by bytes that exceed the length limit without a line delimiter, L{LineReceiver.lineLengthExceeded} is called with all of the bytes following the short line's delimiter. rrN)rrOrbr,rcrrs rtest_longLineAfterShortLinez?LineReceiverLineLengthExceededTests.test_longLineAfterShortLineseDJ1A59:  tz'; ;i GHHH )dj&:;;;;;rc|jjd|jjdzdzzgdz}|j|||g|jjdS)a If L{LineReceiver.dataReceived} is called with more than C{LineReceiver.MAX_LENGTH} bytes containing a line delimiter somewhere not in the first C{MAX_LENGTH} bytes, the entire byte string is passed to L{LineReceiver.lineLengthExceeded}. rrN)rr,rrOrbrcrrs rtest_longLineWithDelimiterz>LineReceiverLineLengthExceededTests.test_longLineWithDelimitersxJ(-- TZ*Q.2 3 4q 8    *** )dj&:;;;;;rcd|jjdzdzz|jjzdz}|j|||g|jjdS)a If L{LineReceiver.dataReceived} is called with more than C{LineReceiver.MAX_LENGTH} bytes containing multiple line delimiters somewhere not in the first C{MAX_LENGTH} bytes, the entire byte string is passed to L{LineReceiver.lineLengthExceeded}. rrN)rrOr,rbrcrrs rtest_multipleLongLinesz:LineReceiverLineLengthExceededTests.test_multipleLongLinessfTZ2Q6:;dj>RRVWW   *** )dj&:;;;;;rctj}tj}|||d|jdzzdz||jdSrrrs rrz:LineReceiverLineLengthExceededTests.test_maximumLineLengthrrctj}tj}|||d|jt|jzz| |j dS)z C{LineReceiver} disconnects the transport it if receives a non-finished line longer than its C{MAX_LENGTH}. rN) r rrrr_rbrOrLr,rrrs rtest_maximumLineLengthRemainingzCLineReceiverLineLengthExceededTests.test_maximumLineLengthRemainingsz "$$!133  Y''' 45#3c%/6J6J#JKLLL  /00000rN) r(r)r*r+rrrrrrrr-rrrrs222 <<< < < < < < < < < < 1 1 1 1 1 1 1 1rrc(eZdZdZdZdZdZdZdS)LineOnlyReceiverTestsz@ Tests for L{twisted.protocols.basic.LineOnlyReceiver}. s7foo bleakness desolation plastic forks c@tj}t}||t |jD]}||||j|j ddddS)zW Test buffering over line protocol: data received should match buffer. rNrK) rrrRr_r rarbrcr6split)rrgrhcs rrkz!LineOnlyReceiverTests.test_buffers  ) + +    4;''  A NN1     T[%6%6u%=%=crc%BCCCCCrct}tj}|||d|jt |jzdzzdz||j dS)z C{LineOnlyReceiver} disconnects the transport if it receives a line longer than its C{MAX_LENGTH} + len(delimiter). rr#rN) rRrrr_rbrOrLr,rrrs r!test_greaterThanMaximumLineLengthz7LineOnlyReceiverTests.test_greaterThanMaximumLineLengths   !133  Y'''  E$s5?';';;a? @8 K     /00000rcntj}|t|jddS)z When L{LineOnlyReceiver.lineReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. rN)r LineOnlyReceiverrrr!rs rrz5LineOnlyReceiverTests.test_lineReceivedNotImplementeds4 &(( -u/A5IIIIIrN)r(r)r*r+rarkrrr-rrrrsZF D D D 1 1 1JJJJJrrc6eZdZdZdZdZdZefdeddfdZ dS) TestMixincg|_dSrr5rs rr7zTestMixin.connectionMades  rc:|j|dSrrUrrjs rstringReceivedzTestMixin.stringReceiveds Qr2rreasonreturnNcd|_dS)Nr#)closed)rrs rconnectionLostzTestMixin.connectionLost#s  r) r(r)r*r7rrOrr rrr-rrrrsh   J F/=W$rrceZdZdZdS) TestNetstringcn|j||j|dSr)r6rrGwriters rrzTestNetstring.stringReceived(s4 Q QrN)r(r)r*rr-rrrr's#     rrcPeZdZUgZeeeed<dZded<dZ dZ dS)LPTestCaseMixinillegalStringsN!Optional[Type[protocol.Protocol]]rc~tj}|}|||S)z Return a new instance of C{self.protocol} connected to a new instance of L{proto_helpers.StringTransport}. )rrrr_)rrgrhs r getProtocolzLPTestCaseMixin.getProtocol1s8  ) + + MMOO rc|jD]\}|}t|D]}||||jj]dS)zO Assert that illegal strings cause the transport to be closed. N)rrr rbrrGrrrjrrs r test_illegalzLPTestCaseMixin.test_illegal;su$ 7 7A  ""Aq\\ " "q!!!! OOAK5 6 6 6 6  7 7r) r(r)r*rrrbytes__annotations__rrrr-rrrr-s^,.NHT%[)...48H188877777rrceZdZdZgdZgdZeZdZdZ dZ dZ dZ d Z d Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdS)NetstringReceiverTestszA Tests for L{twisted.protocols.basic.NetstringReceiver}. )helloworldshowsaresyou123s:todaysaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)s9999999999999999999999sabcs4:abcdes751:aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaab,ctj|_t|_|j|jdSr)rrrGrnetstringReceiverr_rs rrzNetstringReceiverTests.setUpVs=&688!. --dn=====rctddD]}tj}t}d|_|||jD]}|||}tt||zdzD],}|||z|dz|z}|r| |-| |j |jdS)zI Strings can be received in chunks of different lengths. r#r\iN) r]rrrrOr_strings sendStringvaluerLrbrcr6)rrfrgrhrjoutris rrkz"NetstringReceiverTests.test_buffer[s!B<< 7 7K-//AAAL  Q   \   Q''))C3s88{2Q677 & &K1q5K*??@&NN1%%%   QZ 6 6 6 6 7 7rc||jd||jjdgdS)zE Empty netstrings (with length '0') can be received. s0:,rNrrbrcr6rs rtest_receiveEmptyNetstringz1NetstringReceiverTests.test_receiveEmptyNetstringms? ++F333 /83%@@@@@rc||jd||jjdgdS)z; One-character netstrings can be received. 1:a,rZNrrs rtest_receiveOneCharacterz/NetstringReceiverTests.test_receiveOneCharacterts? ++G444 /84&AAAAArc||jd||jjdgdS)z; Two-character netstrings can be received. s2:ab,abNrrs rtest_receiveTwoCharactersz0NetstringReceiverTests.test_receiveTwoCharacters{s? ++H555 /85'BBBBBrc||jd||jjdgdS)z Netstrings with embedded netstrings. This test makes sure that the parser does not become confused about the ',' and ':' characters appearing inside the data portion of the netstring. s4:1:a,,rNrrs rtest_receiveNestedNetstringz2NetstringReceiverTests.test_receiveNestedNetstrings? ++J777 /87)DDDDDrcx|jd||jjdS)zL Netstrings containing more data than expected are refused. s2:aaa,NrrbrrGrrs rtest_moreDataThanSpecifiedz1NetstringReceiverTests.test_moreDataThanSpecifieds8 ++I666 455555rcx|jd||jjdS)z Netstrings that should be empty according to their length specification are refused if they contain data. s0:a,Nr rs r$test_moreDataThanSpecifiedBorderCasez;NetstringReceiverTests.test_moreDataThanSpecifiedBorderCase8 ++G444 455555rcx|jd||jjdS)z` Netstrings without leading digits that specify the length are refused. s:aaa,Nr rs rtest_missingNumberz)NetstringReceiverTests.test_missingNumber8 ++H555 455555rcx|jd||jjdS)zg Netstrings without a colon between length specification and data are refused. s3aaa,Nr rs rtest_missingColonz(NetstringReceiverTests.test_missingColonrrcx|jd||jjdS)zY Netstrings that have no leading digits nor a colon are refused. saaa,Nr rs rtest_missingNumberAndColonz1NetstringReceiverTests.test_missingNumberAndColonr rcx|jd||jjdS)zA Netstrings consisting only of data are refused. saaaNr rs r test_onlyDataz$NetstringReceiverTests.test_onlyDatas8 ++F333 455555rc|jd|jd||jjdg||jdS)z= Netstrings can be received in two portions. s4:aasaa,saaaaN)rrbrcr6r_payloadCompleters rtest_receiveNetstringPortions_1z6NetstringReceiverTests.test_receiveNetstringPortions_1sw ++G444 ++F333 /87)DDD .??AABBBBBrcdD]}|j|||jjdgdS)z Netstrings can be received in more than two portions, even if the length specification is split across two portions. )1s0:01234s56789,rXNrrparts rtest_receiveNetstringPortions_2z6NetstringReceiverTests.test_receiveNetstringPortions_2sQ 7 6 6D  " / / 5 5 5 5 /8=/JJJJJrcdD]}|j|||jjdgdS)zE Netstrings can be received one character at a time. )2:rZbrrNrrs rtest_receiveNetstringPortions_3z6NetstringReceiverTests.test_receiveNetstringPortions_3sQ3 6 6D  " / / 5 5 5 5 /85'BBBBBrcL|jd||j||jjdg|jd||jjddgdS)z A stream of two netstrings can be received in two portions, where the first portion contains the complete first netstring and the length specification of the second netstring. s1:a,1rZs:b,r#N)rrbrrrcr6rs rtest_receiveTwoNetstringsz0NetstringReceiverTests.test_receiveTwoNetstringss ++H555 .??AABBB /84&AAA ++F333 /84,GGGGGrc|jjdz}|jdt |dd|zf||jjdS)zs Netstrings with a length specification exceeding the specified C{MAX_LENGTH} are refused. r#rr"rZN)rrOrbrrrrGr)rtooLongs rtest_maxReceiveLimitz+NetstringReceiverTests.test_maxReceiveLimitsn (3a7 ++ HHeGnndD7N; < <    455555rcd|j_|j||jjddS)zw C{_consumeLength} returns the expected length of the netstring, including the trailing comma. 12: N)r_remainingData_consumeLengthrc_expectedPayloadSizers rtest_consumeLengthz)NetstringReceiverTests.test_consumeLengthsF 17- --/// /DbIIIIIrcd|j_d|j_|j||jjddS)z C{_consumeLength} works as expected if the length specification contains the value of C{MAX_LENGTH} (border case). r+ r,N)rr-rOr.rcr/rs rtest_consumeLengthBorderCase1z4NetstringReceiverTests.test_consumeLengthBorderCase1sQ 17-,.) --/// /DbIIIIIrcd|j_d|j_|tj|jjdS)z C{_consumeLength} raises a L{basic.NetstringParseError} if the length specification exceeds the value of C{MAX_LENGTH} by 1 (border case). r+ Nrr-rOrr NetstringParseErrorr.rs rtest_consumeLengthBorderCase2z4NetstringReceiverTests.test_consumeLengthBorderCase2sJ 17-,.)   %t'='L     rcd|j_d|j_|tj|jjdS)z C{_consumeLength} raises a L{basic.NetstringParseError} if the length specification exceeds the value of C{MAX_LENGTH} by more than 1. s1000:r5Nr6rs rtest_consumeLengthBorderCase3z4NetstringReceiverTests.test_consumeLengthBorderCase3 sJ 19-,.)   %t'='L     rcntj}|t|jddS)z When L{NetstringReceiver.stringReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. rN)r NetstringReceiverrrrrs r!test_stringReceivedNotImplementedz8NetstringReceiverTests.test_stringReceivedNotImplementeds4 ')) -u/CUKKKKKrN)r(r)r*r+rrrrrrkrrrrr r rrrrrrr$r&r)r0r3r8r:r=r-rrrrFsUTTGNH>>> 777$AAABBBCCCEEE666666666666666666CCCKKKCCC H H H 6 6 6JJJJJJ        LLLLLrrceZdZUdZdZded<dZeee ed<dZ eee ed<dZ eee ed<dZ d Z d Zd Zd Zd ZdS)IntNTestCaseMixinz4 TestCase mixin for int-prefixed protocols. NrrrrpartialStringsc  |}|jD]Q}ttj|jt ||zD]}||R||j |jdS)z> Test receiving data find the same data send. N) rrr structpack structFormatrLrbrcr6)rrrjrs r test_receivezIntNTestCaseMixin.test_receive(s      " "Av{1>3q66BBQFGG " "q!!!! " T\22222rc|jD]X}|}t|D]}||||jgYdS)zK Send partial data, nothing should be definitely received. N)r@rr rbrcr6rs r test_partialzIntNTestCaseMixin.test_partial2sv$ - -A  ""Aq\\ " "q!!!!   QZ , , , ,  - -rc|}|d||jt j|jddzdS)z2 Test sending data over protocol. bbbbbbbbbbbbbbbbN)rrrcrGrrBrCrDrrs r test_sendzIntNTestCaseMixin.test_send<so      Y  K    Q^R!@!@9!L     rcg}|}|j|_d|_|t j|jd||dgdS)z When a length prefix is received which is greater than the protocol's C{MAX_LENGTH} attribute, the C{lengthLimitExceeded} method is called with the received length prefix. r\r5N) rrlengthLimitExceededrOrbrBrCrDrc)rrErs rtest_lengthLimitExceededz*IntNTestCaseMixin.test_lengthLimitExceededFsi       &   v{1>266777 "&&&&&rc|}d|_|tj|jddz||jgdS)z If a length prefix for a string longer than C{MAX_LENGTH} is delivered to C{dataReceived} at the same time as the entire string, the string is not passed to C{stringReceived}. r\r5s xxxxxxxxxxxN)rrOrbrBrCrDrcr6rKs rtest_longStringNotDeliveredz-IntNTestCaseMixin.test_longStringNotDeliveredSs_       v{1>266BCCC R(((((rcntj}|t|jddS)z When L{IntNStringReceiver.stringReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. rN)r IntNStringReceiverrrrrs rr=z3IntNTestCaseMixin.test_stringReceivedNotImplemented^s4 (** -u/CUKKKKKr)r(r)r*r+rrrrrrrr@rErGrLrOrQr=r-rrr?r?s59H1888%)GXd5k "))),0NHT%[)000,0NHT%[)000333---    ' ' ' ) ) )LLLLLrr?c0eZdZdZdZdZdZdZdZdS)RecvdAttributeMixinz Mixin defining tests for string receiving protocols with a C{recvd} attribute which should be settable by application code, to be combined with L{IntNTestCaseMixin} on a L{TestCase} subclass cVtj|jt||zS)zg Return C{data} prefixed with message length in C{protocol.structFormat} form. )rBrCrDrL)rrr&s r makeMessagezRecvdAttributeMixin.makeMessagens$ {80#d))<.stringReceived}s MM!' " " " " "raaaaasbbbbN)rrrBrCrDrbrc)rrcompleteMessageincompleteMessagerr]s @@rtest_recvdContainsRemainingDataz3RecvdAttributeMixin.test_recvdContainsRemainingDataus       # # # # # #* +ana88HE"K::hG )::;;; "3!455555rcD |g d}|| fd}|_d}d}||}||}||z| ||gdS)z In stringReceived, if recvd is changed, messages should be parsed from it rather than the input to dataReceived. sccccccDs_|dSr)r[r)r\messageCrr]s rrz=RecvdAttributeMixin.test_recvdChanged..stringReceiveds* #" MM. ) ) ) ) )rr_sbbbbbN)rrWrrbrc) rpayloadCrpayloadApayloadBmessageAmessageBrerr]s @@@rtest_recvdChangedz%RecvdAttributeMixin.test_recvdChangeds     ##Ax00 * * * * * * * *##Ax00##Ax00 x(*+++ (H!566666rc|g}d|jD]@}||||Agfd}|_d|d||j|jddS)z Data already parsed by L{IntNStringReceiver.dataReceived} is not reparsed if C{stringReceived} consumes some of the L{IntNStringReceiver.recvd} buffer. sct|jtd_dSr)rr[rL)r\SWITCHrr]s rrz:RecvdAttributeMixin.test_switching..stringReceiveds1 MM. ) ) )+c&kkmm4EKKKrrN) rrrrWrrbrrcr[)rmixrjrrnrr]s @@@rtest_switchingz"RecvdAttributeMixin.test_switchings   ""$  A JJt''q11 2 2 2 JJv     5 5 5 5 5 5 5 . 388C==))) 7### ... g.....rcb|d}t|dz _||}gfd}|_||dt||d|dS)z The L{IntNStringReceiver.recvd} buffer contains all data not yet processed by L{IntNStringReceiver.dataReceived} if the C{lengthLimitExceeded} event occurs. stoo longr#cf|jdSrrZ)rErr]s rrNzPRecvdAttributeMixin.test_recvdInLengthLimitExceeded..lengthLimitExceededs/ MM& ! ! ! MM%+ & & & & &rrN)rrLrOrWrNrbrc)rDATAmessagerNrr]s @@rtest_recvdInLengthLimitExceededz3RecvdAttributeMixin.test_recvdInLengthLimitExceededs   ""t99q=""5$// ' ' ' ' ' '%8! 7### CII... G,,,,,rN) r(r)r*r+rWrbrkrqrvr-rrrUrUgsl DDD666$777.///8-----rrUceZdZdZdS) TestInt32z A L{basic.Int32StringReceiver} storing received strings in an array. @ivar received: array holding received strings. Nr(r)r*r+r-rrrxrxrrxc2eZdZdZeZddgZdgZgdZdZ dS) Int32Testsz/ Test case for int32-prefixed protocol rZrIs aaaaaa)s hello thererc|}|d||jd|d||jdgdS)z? Test specific behavior of the 32-bits length. rsfoosubarubarNrrrcrGrrbr6rKs r test_datazInt32Tests.test_datas      V **,,.DEEE ./// gY/////rN) r(r)r*r+rxrrrr@rr-rrr|r|sQHYG/0N;;;N00000rr|ceZdZdZdS) TestInt16z A L{basic.Int16StringReceiver} storing received strings in an array. @ivar received: array holding received strings. Nryr-rrrrrzrrc8eZdZdZeZddgZdgZgdZdZ dZ dS) Int16Testsz/ Test case for int16-prefixed protocol rZrIsaaaaaa)r}rc|}|d||jd|d||jdgdS)z? Test specific behavior of the 16-bits length. rsfoosubarrNrrKs rrzInt16Tests.test_datas}      V **,,n=== ''' gY/////rc|}dd|jdzzdzz}|t|j|dSzA Send too much data: that should cause an error. r#rr#Nr prefixLengthrAssertionErrorrrrtooSends rtest_tooLongSendzInt16Tests.test_tooLongSend P     ! 23a78 .!,@@@@@rN) r(r)r*r+rrrrr@rrr-rrrrseHYG'(N333N000AAAAArrceZdZdZdS)TestInt8z A L{basic.Int8StringReceiver} storing received strings in an array. @ivar received: array holding received strings. Nryr-rrrrrzrrc8eZdZdZeZddgZdgZgdZdZ dZ dS) Int8Testsz. Test case for int8-prefixed protocol rZrIsaaaaaa)sdzadzrc|}|d||jd|d||jdgdS)z> Test specific behavior of the 8-bits length. rsfoosubarrNrrKs rrzInt8Tests.test_data%s}      V **,,j999 {### gY/////rc|}dd|jdzzdzz}|t|j|dSrrrs rrzInt8Tests.test_tooLongSend/rrN) r(r)r*r+rrrrr@rrr-rrrrseHYG'(N---N000AAAAArrc2eZdZdZdZdZdZdZdZdZ dS)OnlyProducerTransportzm Transport which isn't really a transport, just looks like one to someone not looking very hard. Fcg|_dSrrrs rrzOnlyProducerTransport.__init__As  rcd|_dS)NTpausedrs rr@z$OnlyProducerTransport.pauseProducingDs  rcd|_dS)NFrrs rrBz%OnlyProducerTransport.resumeProducingGs  rc:|j|dSr)r&r)rrs rrzOnlyProducerTransport.writeJs rN) r(r)r*r+rrrr@rBrr-rrrr8sf FM     rrceZdZdZdZdS)ConsumingProtocolzC Protocol that really, really doesn't want any more bytes. cb|j||dSr)rGrr@rs rr!zConsumingProtocol.lineReceivedSs0 T""" rN)r(r)r*r+r!r-rrrrNs-rrceZdZdZdZdS) ProducerTestszM Tests for L{basic._PausableMixin} and L{basic.LineReceiver.paused}. ct}t}|||d||jg||j||j|d||jdg||j||j| ||jdg||j||j|d||jddg||j||j| ||jgd||j||j|d||jgd||j||j| ||jgd||j||j| ||jgd||j||jd S) a? When L{basic.LineReceiver} is paused, it doesn't deliver lines to L{basic.LineReceiver.lineReceived} and delivers them immediately upon being resumed. L{ConsumingProtocol} is a L{LineReceiver} that pauses itself after every line, and writes that line to its transport. shello, sworld hello, worldshello world r)rrrs goodbye )rrrrN) rrr_rbrcr&rrrrB)rprgs rtest_pauseResumezProducerTests.test_pauseResume]s    ! # #  z""" $$$ """ """ |$$$ /!2333 !!! !!!  /!2333 """ """ ,--- /8!<=== !!! !!!  !F!F!FGGG !!! !!! ~&&& !F!F!FGGG !!! !!!  !R!R!RSSS !!! !!!  !R!R!RSSS """ """""rN)r(r)r*r+rr-rrrrXs2A#A#A#A#A#rrc6eZdZdZdZdZdZdZdZdZ dS) FileSenderTestsz( Tests for L{basic.FileSender}. c|tj}|tt|dS)zP L{basic.FileSender} implements the L{IPullProducer} interface. N)r FileSenderrrr )rsenders rtest_interfacezFileSenderTests.test_interfaces4!##  Y7788888rctd}tj}tj}|||||j|||j dS)z When L{basic.FileSender.beginFileTransfer} is called, it registers itself with provided consumer, as a non-streaming producer. Test contentN) rrrr rbeginFileTransferrcproducerr streaming)rsourceconsumerrs rtest_producerRegisteredz'FileSenderTests.test_producerRegisteredsy )) 022!##  222 *F333 +,,,,,rctd}tj}tj}|||}||||j| d| || d| dS)z L{basic.FileSender} sends the content of the given file using a C{IConsumer} interface via C{beginFileTransfer}. It returns a L{Deferred} which fires with the last byte sent. rtN) rrrr rrrB assertIsNonerrcsuccessResultOfrrrrrds r test_transferzFileSenderTests.test_transfers )) 022!##  $ $VX 6 6       (+,,, t33A66777 (..*:*:;;;;;rctd}tj}tj}d|_|||}||d| ||d| ||d| ||d| ||d| dS)zj L{basic.FileSender} reads at most C{CHUNK_SIZE} every time it resumes producing. rr=sTestsTest conrN) rrrr r CHUNK_SIZErrBrcrrrs rtest_transferMultipleChunksz+FileSenderTests.test_transferMultipleChunkss< )) 022!##  $ $VX 6 6    (.."2"2333    hnn&6&6777    (..*:*:;;;    t33A66777 (..*:*:;;;;;rcd}td}tj}tj}||||}|||d|||d| dS)z L{basic.FileSender.beginFileTransfer} takes a C{transform} argument which allows to manipulate the data on the fly. c*|Sr)swapcase)chunks r transformz=FileSenderTests.test_transferWithTransform..transforms>>## #rrTs tEST CONTENTN) rrrr rrrBrcrr)rrrrrrs rtest_transferWithTransformz*FileSenderTests.test_transferWithTransforms  $ $ $)) 022!##  $ $VXy A A       t33A66777 (..*:*:;;;;;rcrtd}tj}tj}|||}|||}|t| dt|j dS)z The C{Deferred} returned by L{basic.FileSender.beginFileTransfer} fails with an C{Exception} if C{stopProducing} when the transfer is not complete. rz#Consumer asked us to stop producingN) rrrr rrrCfailureResultOftrap Exceptionrcstrr)rrrrrfailures rtest_abortedTransferz$FileSenderTests.test_abortedTransfers )) 022!##  $ $VX 6 6&&q)) Y >GM@R@RSSSSSrN) r(r)r*r+rrrrrrr-rrrrs}999 - - -<<<$<<<.<<<(TTTTTrr);r+rBriortypingrrrzope.interface.verifyrtwisted.internetrr twisted.internet.interfacesr twisted.internet.protocolr twisted.protocolsr twisted.python.compatr twisted.python.failurer twisted.testr twisted.trialrrrr/rrRSynchronousTestCaserWrrrrr<rrrr?rUInt32StringReceiverrxr|Int16StringReceiverrrInt8StringReceiverrrrrrTestCaserr-rrrs  ''''''''''......++++++++111111444444######++++++******&&&&&&""""""#####+###0E:E:E:E:E:#E:E:E:P#####U+###*uJuJuJuJuJ4uJuJuJp$$$$$%"4$$$.L1L1L1L1L1(*FL1L1L1^)J)J)J)J)JH8)J)J)JX             Iu6   777777772ULULULULULX9?ULULULpFLFLFLFLFLFLFLFLRg-g-g-g-g-g-g-g-T 5400000-/@BU000* 54AAAAA-/@BUAAA:y%2AAAAA,.?ATAAA:        ,*F#F#F#F#F#H0F#F#F#RdTdTdTdTdTh'dTdTdTdTdTr