a )¶(bô§ã@sJdZddlZddlZddlmZddlmZmZmZddl m Z ddl m Z m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZGdd„dejƒZGdd„dejƒZGdd„dejƒZ Gdd„dej!ƒZ"Gdd„dejƒZ#Gdd„dej!ƒZ$Gdd„dej!ƒZ%Gdd„dƒZ&Gdd„de&ej'ƒZ(Gd d!„d!ƒZ)Gd"d#„d#ej!e)ƒZ*Gd$d%„d%e)ƒZ+Gd&d'„d'ƒZ,Gd(d)„d)e&ej-ƒZ.Gd*d+„d+ej!e+e,ƒZ/Gd,d-„d-e&ej0ƒZ1Gd.d/„d/ej!e+e,ƒZ2Gd0d1„d1e&ej3ƒZ4Gd2d3„d3ej!e+e,ƒZ5Gd4d5„d5ƒZ6Gd6d7„d7ejƒZ7Gd8d9„d9ej!ƒZ8Gd:d;„d;ej9ƒZ:dS)Ú MAX_LENGTHrrrrrÚlineLengthExceededrszLineTester.lineLengthExceeded)N) r#r$r%r&r'r@rr.rr"rArrrrr(2s   r(c@s(eZdZdZdZdZdd„Zdd„ZdS) ÚLineOnlyTesterz) A buffering line only receiver. rr)cCs g|_dSr+r,rrrrr.‚szLineOnlyTester.connectionMadecCs|j |¡dS)z% Save received data. N©r-rrrrrrˆszLineOnlyTester.lineReceivedN)r#r$r%r&r'r@r.rrrrrrBzs rBc@sºeZdZdZdZgd¢Zdd„ZdZgd¢ZedgZ d d „Z d Z gd ¢Z gd ¢Z dd„ZdZgd¢Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&S)'ÚLineReceiverTestsz\ Test L{twisted.protocols.basic.LineReceiver}, using the C{LineTester} wrapper. sžlen 10 0123456789len 5 1234 len 20 foo 123 0123456789 012345678len 0 foo 5 1234567890123456789012345678901234567890123456789012345678901234567890 len 1 a) slen 10ó 0123456789ólen 5s1234 slen 20sfoo 123s0123456789 012345678slen 0sfoo 5r/s67890slen 1óacCs‚tddƒD]r}t ¡}tƒ}| t |¡¡tt|jƒ|dƒD](}|j|||d|…}|  |¡qB|  |j |j ¡q dS)zl Test buffering for different packet size, checking received matches expected data. ré N) ÚrangerÚStringIOWithoutClosingr(ÚmakeConnectionrÚ FileWrapperr>ÚbufferÚ dataReceivedÚ assertEqualÚoutputr-©rÚ packet_sizeÚtÚaÚiÚsrrrÚ test_buffer·s zLineReceiverTests.test_buffers!twiddle1 twiddle2 pause twiddle3 )ótwiddle1ótwiddle2r0ótwiddle3cCs¦tddƒD]–}t ¡}t ¡}t|ƒ}| t |¡¡tt |j ƒ|dƒD](}|j |||d|…}|  |¡qL|  |j |j¡| d¡|  |j|j¡q dS)zo Test pause inside data receiving. It uses fake clock to see if pausing/resuming work. rrHrN)rIrrJrÚClockr(rKrrLr>ÚpauseBufrNrOÚ pauseOutput1r-ÚadvanceÚ pauseOutput2©rrRrSr*rTrUrVrrrÚ test_pausingÊs  zLineReceiverTests.test_pausings/twiddle1 twiddle2 len 5 rawpause 12345twiddle3 )rXrYrFr1r/)rXrYrFr1s12345rZcCs¦tddƒD]–}t ¡}t ¡}t|ƒ}| t |¡¡tt |j ƒ|dƒD](}|j |||d|…}|  |¡qL|  |j |j¡| d¡|  |j|j¡q dS)z7 Test pause inside raw date receiving. rrHrN)rIrrJrr[r(rKrrLr>Ú rawpauseBufrNrOÚrawpauseOutput1r-r^ÚrawpauseOutput2r`rrrÚtest_rawPausingçs  z!LineReceiverTests.test_rawPausings"twiddle1 twiddle2 stop more stuff )rXrYr2cCs‚tddƒD]r}t ¡}tƒ}| t |¡¡tt|jƒ|dƒD](}|j|||d|…}|  |¡qB|  |j |j ¡q dS)z- Test stop inside producing. rrHN) rIrrJr(rKrrLr>Ústop_bufrNrOÚ stop_outputr-rQrrrÚtest_stopProducingûs z$LineReceiverTests.test_stopProducingcCs>tƒ}t ¡}| t |¡¡| d¡| |jgd¢¡dS)z6 Test produce/unproduce in receiving. s&produce hello world unproduce goodbye )r4s hello worldr5ógoodbyeN) r(rrJrKrrLrNrOr-)rrTrSrrrÚtest_lineReceiverAsProducers  ÿz-LineReceiverTests.test_lineReceiverAsProducercCshGdd„dtjƒ}|ƒ}| d¡| |jd¡| |jd¡| d¡| |jd¡| |jd¡d S) z  L{LineReceiver.clearLineBuffer} removes all buffered data and returns it as a C{bytes} and can be called from beneath C{dataReceived}. c@seZdZdd„ZdS)z@LineReceiverTests.test_clearLineBuffer..ClearingReceivercSs||_| ¡|_dSr)rZclearLineBufferr?rrrrrszMLineReceiverTests.test_clearLineBuffer..ClearingReceiver.lineReceivedN)r#r$r%rrrrrÚClearingReceiversrks foo bar bazófoosbar bazsquux squuxr/N)r Ú LineReceiverrNrOrr?)rrkrrrrÚtest_clearLineBuffers  z&LineReceiverTests.test_clearLineBuffercCsPtƒ}t ¡}| t |¡¡t ¡}| d|¡|  d|d  |j ¡¡dS)zC Test switching modes many times on the same data. sx xóxr/N) rrrJrKrrLÚsysÚgetrecursionlimitrNrOÚjoinr)rÚprotor<ÚlimitrrrÚtest_stackRecursion*s z%LineReceiverTests.test_stackRecursioncCsBt ¡}t ¡}| |¡| d|jdd¡| |j¡dS©zx C{LineReceiver} disconnects the transport if it receives a line longer than its C{MAX_LENGTH}. roró rN© r rmrÚStringTransportrKrNr@Ú assertTrueÚ disconnecting©rrsr<rrrÚtest_maximumLineLength5s  z(LineReceiverTests.test_maximumLineLengthcCs–tƒ}d|_t ¡}| |¡d|jd}| |¡| |jdd…¡| |jdd…|¡| |j¡|  t |j ƒd¡|  ||j d¡dS)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). r3rorNr=r) r(r@rryrKrNr'Ú assertFalser{rOr>r-)rrsrSrrrrÚ&test_maximumLineLengthPartialDelimiter@s    z8LineReceiverTests.test_maximumLineLengthPartialDelimitercCsXt ¡}d|_t ¡}| |¡| d|j|jdt|jƒd…¡|  |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 roNr) r rmr'rryrKrNr@r>r~r{r|rrrÚ(test_notQuiteMaximumLineLengthUnfinishedVs  ÿz:LineReceiverTests.test_notQuiteMaximumLineLengthUnfinishedcCsFt ¡}dd„|_t ¡}| |¡| ¡| d¡}| |t ¡dS)zf C{LineReceiver.dataReceived} forwards errors returned by C{rawDataReceived}. cSstdƒS)NZoops)Ú RuntimeError©r!rrrÚor/z5LineReceiverTests.test_rawDataError..sdataN) r rmr"rryrKrrNZassertIsInstancer)rrsr<ZwhyrrrÚtest_rawDataErroris   z#LineReceiverTests.test_rawDataErrorcCst ¡}| t|jd¡dS)z‰ When L{LineReceiver.rawDataReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. ÚfooN)r rmÚ assertRaisesÚNotImplementedErrorr"©rrsrrrÚ"test_rawDataReceivedNotImplementedvsz4LineReceiverTests.test_rawDataReceivedNotImplementedcCst ¡}| t|jd¡dS)z† When L{LineReceiver.lineReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. r…N)r rmr†r‡rrˆrrrÚtest_lineReceivedNotImplemented~sz1LineReceiverTests.test_lineReceivedNotImplementedN)r#r$r%r&rMrPrWr\r]r_rarbrcrdrerfrgrhrjrnrur}rr€r„r‰rŠrrrrrDs0       rDc@s(eZdZdZdd„Zdd„Zdd„ZdS) ÚExcessivelyLargeLineCatcherz® Helper for L{LineReceiverLineLengthExceededTests}. @ivar longLines: A L{list} of L{bytes} giving the values C{lineLengthExceeded} has been called with. cCs g|_dSr)Ú longLinesrrrrr.sz*ExcessivelyLargeLineCatcher.connectionMadecCsdS)z/ Disregard any received lines. Nrrrrrr’sz(ExcessivelyLargeLineCatcher.lineReceivedcCs|j |¡dS)zF Record any data that exceeds the line length limits. N)rŒrr rrrrA—sz.ExcessivelyLargeLineCatcher.lineLengthExceededN)r#r$r%r&r.rrArrrrr‹‡sr‹c@sHeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dS)Ú#LineReceiverLineLengthExceededTestszO Tests for L{twisted.protocols.basic.LineReceiver.lineLengthExceeded}. cCs,tƒ|_d|j_t ¡|_|j |j¡dS)Né)r‹rsr@rryr<rKrrrrÚsetUp£s z)LineReceiverLineLengthExceededTests.setUpcCs6d|jjdd}|j |¡| |g|jj¡dS)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}. roéN)rsr@rNrOrŒ©rZ excessiverrrÚtest_longUnendedLine©s z8LineReceiverLineLengthExceededTests.test_longUnendedLinecCsBd|jjdd}|j d|jj|¡| |g|jj¡dS)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. rorN)rsr@rNr'rOrŒr‘rrrÚtest_longLineAfterShortLine³sz?LineReceiverLineLengthExceededTests.test_longLineAfterShortLinecCsF|jj d|jjddgd¡}|j |¡| |g|jj¡dS)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}. rorN)rsr'rrr@rNrOrŒr‘rrrÚtest_longLineWithDelimiter¾s ÿ z>LineReceiverLineLengthExceededTests.test_longLineWithDelimitercCsBd|jjdd|jjd}|j |¡| |g|jj¡dS)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}. rorN)rsr@r'rNrOrŒr‘rrrÚtest_multipleLongLinesËs  z:LineReceiverLineLengthExceededTests.test_multipleLongLinescCsBt ¡}t ¡}| |¡| d|jdd¡| |j¡dSrvrxr|rrrr}Ös  z:LineReceiverLineLengthExceededTests.test_maximumLineLengthcCsDt ¡}t ¡}| |¡| d|jt|jƒ¡|  |j ¡dS)z… C{LineReceiver} disconnects the transport it if receives a non-finished line longer than its C{MAX_LENGTH}. roN) r rmrryrKrNr@r>r'rzr{r|rrrÚtest_maximumLineLengthRemainingás  zCLineReceiverLineLengthExceededTests.test_maximumLineLengthRemainingN) r#r$r%r&rr’r“r”r•r}r–rrrrržs     rc@s,eZdZdZdZdd„Zdd„Zdd„Zd S) ÚLineOnlyReceiverTestsz@ Tests for L{twisted.protocols.basic.LineOnlyReceiver}. s7foo bleakness desolation plastic forks cCsTt ¡}tƒ}| |¡t|jƒD]}| |¡q"| |j|j  d¡dd…¡dS)zW Test buffering over line protocol: data received should match buffer. rNr=) rryrBrKr rMrNrOr-Úsplit)rrSrTÚcrrrrWøs   z!LineOnlyReceiverTests.test_buffercCsJtƒ}t ¡}| |¡| d|jt|jƒdd¡| |j ¡dS)z C{LineOnlyReceiver} disconnects the transport if it receives a line longer than its C{MAX_LENGTH} + len(delimiter). rorrwN) rBrryrKrNr@r>r'rzr{r|rrrÚ!test_greaterThanMaximumLineLengths ÿz7LineOnlyReceiverTests.test_greaterThanMaximumLineLengthcCst ¡}| t|jd¡dS)zŠ When L{LineOnlyReceiver.lineReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. r…N)r ÚLineOnlyReceiverr†r‡rrˆrrrrŠsz5LineOnlyReceiverTests.test_lineReceivedNotImplementedN)r#r$r%r&rMrWršrŠrrrrr—ís   r—c@s8eZdZdd„Zdd„ZdZdZefeddœd d „Z dS) Ú TestMixincCs g|_dSrr,rrrrr.szTestMixin.connectionMadecCs|j |¡dSrrC©rrVrrrÚstringReceivedszTestMixin.stringReceivedé2rN)ÚreasonÚreturncCs d|_dS)Nr)Úclosed)rr rrrÚconnectionLost#szTestMixin.connectionLost) r#r$r%r.ržr@r¢r r r£rrrrrœs rœc@seZdZdd„ZdS)Ú TestNetstringcCs|j |¡|j |¡dSr)r-rr<Úwriterrrrrž(s zTestNetstring.stringReceivedN)r#r$r%ržrrrrr¤'sr¤c@s>eZdZUgZeeeed<dZded<dd„Z dd„Z dS) ÚLPTestCaseMixinÚillegalStringsNú!Optional[Type[protocol.Protocol]]rcCst ¡}| ¡}| |¡|S)z„ Return a new instance of C{self.protocol} connected to a new instance of L{proto_helpers.StringTransport}. )rryrrK)rrSrTrrrÚ getProtocol2s zLPTestCaseMixin.getProtocolcCs>|jD]2}| ¡}t|ƒD]}| |¡q| |jj¡qdS)zO Assert that illegal strings cause the transport to be closed. N)r§r©r rNrzr<r{©rrVÚrr™rrrÚ test_illegal<s    zLPTestCaseMixin.test_illegal) r#r$r%r§rrÚbytesÚ__annotations__rr©r¬rrrrr¦-s   r¦c@sÔeZdZdZgd¢Zgd¢ZeZdd„Zdd„Z dd „Z d d „Z d d „Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Zd,d-„Zd.d/„Zd0S)1ÚNetstringReceiverTestszA Tests for L{twisted.protocols.basic.NetstringReceiver}. )óhelloóworldshowsaresyou123s:todaysaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)s9999999999999999999999sabcs4:abcdes751:aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaab,cCs$t ¡|_tƒ|_|j |j¡dSr)rryr<r¤ÚnetstringReceiverrKrrrrrWs zNetstringReceiverTests.setUpcCs tddƒD]}t ¡}tƒ}d|_| |¡|jD]}| |¡q2| ¡}tt |ƒ|dƒD]*}||||d|…}|r^|  |¡q^|  |j |j¡q dS)zI Strings can be received in chunks of different lengths. rrHi»N) rIrryr¤r@rKÚstringsÚ sendStringÚvaluer>rNrOr-)rrRrSrTrVÚoutrUrrrrW\s    z"NetstringReceiverTests.test_buffercCs"|j d¡| |jjdg¡dS)zE Empty netstrings (with length '0') can be received. s0:,r/N©r²rNrOr-rrrrÚtest_receiveEmptyNetstringns z1NetstringReceiverTests.test_receiveEmptyNetstringcCs"|j d¡| |jjdg¡dS)z; One-character netstrings can be received. ó1:a,rGNr·rrrrÚtest_receiveOneCharacterus z/NetstringReceiverTests.test_receiveOneCharactercCs"|j d¡| |jjdg¡dS)z; Two-character netstrings can be received. s2:ab,óabNr·rrrrÚtest_receiveTwoCharacters|s z0NetstringReceiverTests.test_receiveTwoCharacterscCs"|j d¡| |jjdg¡dS)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,,r¹Nr·rrrrÚtest_receiveNestedNetstringƒs z2NetstringReceiverTests.test_receiveNestedNetstringcCs|j d¡| |jj¡dS)zL Netstrings containing more data than expected are refused. s2:aaa,N©r²rNrzr<r{rrrrÚtest_moreDataThanSpecifiedŒs z1NetstringReceiverTests.test_moreDataThanSpecifiedcCs|j d¡| |jj¡dS)zƒ Netstrings that should be empty according to their length specification are refused if they contain data. s0:a,Nr¾rrrrÚ$test_moreDataThanSpecifiedBorderCase“s z;NetstringReceiverTests.test_moreDataThanSpecifiedBorderCasecCs|j d¡| |jj¡dS)z` Netstrings without leading digits that specify the length are refused. s:aaa,Nr¾rrrrÚtest_missingNumber›s z)NetstringReceiverTests.test_missingNumbercCs|j d¡| |jj¡dS)zg Netstrings without a colon between length specification and data are refused. s3aaa,Nr¾rrrrÚtest_missingColon£s z(NetstringReceiverTests.test_missingColoncCs|j d¡| |jj¡dS)zY Netstrings that have no leading digits nor a colon are refused. saaa,Nr¾rrrrÚtest_missingNumberAndColon«s z1NetstringReceiverTests.test_missingNumberAndColoncCs|j d¡| |jj¡dS)zA Netstrings consisting only of data are refused. saaaNr¾rrrrÚ test_onlyData³s z$NetstringReceiverTests.test_onlyDatacCs>|j d¡|j d¡| |jjdg¡| |j ¡¡dS)z= Netstrings can be received in two portions. s4:aasaa,saaaaN)r²rNrOr-rzÚ_payloadCompleterrrrÚtest_receiveNetstringPortions_1ºs  z6NetstringReceiverTests.test_receiveNetstringPortions_1cCs,dD]}|j |¡q| |jjdg¡dS)zŽ Netstrings can be received in more than two portions, even if the length specification is split across two portions. )ó1s0:01234s56789ó,rENr·©rÚpartrrrÚtest_receiveNetstringPortions_2Ãsz6NetstringReceiverTests.test_receiveNetstringPortions_2cCs,dD]}|j |¡q| |jjdg¡dS)zE Netstrings can be received one character at a time. )ó2ó:rGóbrÈr»Nr·rÉrrrÚtest_receiveNetstringPortions_3Ìsz6NetstringReceiverTests.test_receiveNetstringPortions_3cCsR|j d¡| |j ¡¡| |jjdg¡|j d¡| |jjddg¡dS)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,1rGs:b,rÎN)r²rNrzrÅrOr-rrrrÚtest_receiveTwoNetstringsÔs   z0NetstringReceiverTests.test_receiveTwoNetstringscCs>|jjd}|j d t|ƒdd|f¡¡| |jj¡dS)zs Netstrings with a length specification exceeding the specified C{MAX_LENGTH} are refused. rr/rÍrGN)r²r@rNrrr­rzr<r{)rZtooLongrrrÚtest_maxReceiveLimitàs  ÿz+NetstringReceiverTests.test_maxReceiveLimitcCs&d|j_|j ¡| |jjd¡dS)zw C{_consumeLength} returns the expected length of the netstring, including the trailing comma. ó12:é N)r²Ú_remainingDataÚ_consumeLengthrOÚ_expectedPayloadSizerrrrÚtest_consumeLengthës z)NetstringReceiverTests.test_consumeLengthcCs.d|j_d|j_|j ¡| |jjd¡dS)zŒ C{_consumeLength} works as expected if the length specification contains the value of C{MAX_LENGTH} (border case). rÒé rÓN)r²rÔr@rÕrOrÖrrrrÚtest_consumeLengthBorderCase1ôs z4NetstringReceiverTests.test_consumeLengthBorderCase1cCs&d|j_d|j_| tj|jj¡dS)z¬ C{_consumeLength} raises a L{basic.NetstringParseError} if the length specification exceeds the value of C{MAX_LENGTH} by 1 (border case). rÒé N©r²rÔr@r†r ZNetstringParseErrorrÕrrrrÚtest_consumeLengthBorderCase2þs  ÿz4NetstringReceiverTests.test_consumeLengthBorderCase2cCs&d|j_d|j_| tj|jj¡dS)z¨ C{_consumeLength} raises a L{basic.NetstringParseError} if the length specification exceeds the value of C{MAX_LENGTH} by more than 1. s1000:rÚNrÛrrrrÚtest_consumeLengthBorderCase3 s  ÿz4NetstringReceiverTests.test_consumeLengthBorderCase3cCst ¡}| t|jd¡dS)z When L{NetstringReceiver.stringReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. r…N)r ÚNetstringReceiverr†r‡ržrˆrrrÚ!test_stringReceivedNotImplementedsz8NetstringReceiverTests.test_stringReceivedNotImplementedN)r#r$r%r&r³r§r¤rrrWr¸rºr¼r½r¿rÀrÁrÂrÃrÄrÆrËrÏrÐrÑr×rÙrÜrÝrßrrrrr¯Gs4         r¯c@sŠeZdZUdZdZded<dZeee ed<dZ eee ed<dZ eee ed<dd „Z d d „Z d d „Zdd„Zdd„Zdd„ZdS)ÚIntNTestCaseMixinz4 TestCase mixin for int-prefixed protocols. Nr¨rr³r§ÚpartialStringscCsR| ¡}|jD].}tt |jt|ƒ¡|ƒD]}| |¡q,q| |j |j¡dS)z> Test receiving data find the same data send. N) r©r³r ÚstructÚpackÚ structFormatr>rNrOr-)rr«rVr™rrrÚ test_receive)s  zIntNTestCaseMixin.test_receivecCs>|jD]2}| ¡}t|ƒD]}| |¡q| |jg¡qdS)zK Send partial data, nothing should be definitely received. N)rár©r rNrOr-rªrrrÚ test_partial3s    zIntNTestCaseMixin.test_partialcCs6| ¡}| d¡| |j ¡t |jd¡d¡dS)z2 Test sending data over protocol. óbbbbbbbbbbbbbbbbéN)r©r´rOr<rµrârãrä©rr«rrrÚ test_send=s  ÿzIntNTestCaseMixin.test_sendcCs@g}| ¡}|j|_d|_| t |jd¡¡| |dg¡dS)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. rHrÚN) r©rÚlengthLimitExceededr@rNrârãrärO)rr:r«rrrÚtest_lengthLimitExceededGs z*IntNTestCaseMixin.test_lengthLimitExceededcCs8| ¡}d|_| t |jd¡d¡| |jg¡dS)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}. rHrÚs xxxxxxxxxxxN)r©r@rNrârãrärOr-rérrrÚtest_longStringNotDeliveredTsz-IntNTestCaseMixin.test_longStringNotDeliveredcCst ¡}| t|jd¡dS)zŽ When L{IntNStringReceiver.stringReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. r…N)r ZIntNStringReceiverr†r‡ržrˆrrrrß_sz3IntNTestCaseMixin.test_stringReceivedNotImplemented)r#r$r%r&rr®r³rrr­r§rárårærêrìrírßrrrrràs       ràc@s8eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S) Ú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 cCst |jt|ƒ¡|S)zg Return C{data} prefixed with message length in C{protocol.structFormat} form. )rârãrär>)rrr!rrrÚ makeMessageoszRecvdAttributeMixin.makeMessagecsdg‰| ¡‰‡‡fdd„}|ˆ_t ˆjd¡d}t ˆjd¡d}ˆ ||¡| ˆ|g¡dS)z˜ In stringReceived, recvd contains the remaining data that was passed to dataReceived that was not part of the current message. csˆ ˆj¡dSr©rÚrecvd©ZreceivedString©r«Úresultrrrž~szKRecvdAttributeMixin.test_recvdContainsRemainingData..stringReceivedéóaaaaasbbbbN)r©ržrârãrärNrO)rržZcompleteMessageZincompleteMessagerrórÚtest_recvdContainsRemainingDatavsz3RecvdAttributeMixin.test_recvdContainsRemainingDatacst| ¡‰g‰d}| ˆ|¡‰‡‡‡fdd„}|ˆ_d}d}| ˆ|¡}| ˆ|¡}ˆ ||¡| ˆ||g¡dS)zŠ In stringReceived, if recvd is changed, messages should be parsed from it rather than the input to dataReceived. sccccccsˆs ˆˆ_ˆ |¡dSr)rñrrò©ZmessageCr«rôrrrž’sz=RecvdAttributeMixin.test_recvdChanged..stringReceivedrösbbbbbN)r©rïržrNrO)rZpayloadCržZpayloadAZpayloadBZmessageAZmessageBrrørÚtest_recvdChangedˆs   z%RecvdAttributeMixin.test_recvdChangedcsŒ| ¡‰g}d‰|jD] }| | ˆ|¡¡| ˆ¡qg‰‡‡‡fdd„}|ˆ_ˆ d |¡¡ˆ d¡| ˆ|j¡| ˆjd¡dS)z¹ Data already parsed by L{IntNStringReceiver.dataReceived} is not reparsed if C{stringReceived} consumes some of the L{IntNStringReceiver.recvd} buffer. scs"ˆ |¡ˆjtˆƒd…ˆ_dSr)rrñr>rò©ZSWITCHrsrôrrrž®s z:RecvdAttributeMixin.test_switching..stringReceivedr/óN) r©r³rrïržrNrrrOrñ)rZmixrVržrrúrÚtest_switchingŸs   z"RecvdAttributeMixin.test_switchingcsp| ¡‰d}t|ƒdˆ_| ˆ|¡}g‰‡‡fdd„}|ˆ_ˆ |¡| ˆ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 longrcsˆ |¡ˆ ˆj¡dSrrð)r:©rsrôrrrëÈs zPRecvdAttributeMixin.test_recvdInLengthLimitExceeded..lengthLimitExceededrN)r©r>r@rïrërNrO)rZDATAÚmessagerërrýrÚtest_recvdInLengthLimitExceeded»s  z3RecvdAttributeMixin.test_recvdInLengthLimitExceededN) r#r$r%r&rïr÷rùrürÿrrrrrîhs rîc@seZdZdZdS)Ú TestInt32zƒ A L{basic.Int32StringReceiver} storing received strings in an array. @ivar received: array holding received strings. N©r#r$r%r&rrrrrÒsrc@s2eZdZdZeZddgZdgZgd¢Zdd„Z dS) Ú Int32Testsz/ Test case for int32-prefixed protocol rGrçs aaaaaa)só hello therer/cCsB| ¡}| d¡| |j ¡d¡| d¡| |jdg¡dS)z? Test specific behavior of the 32-bits length. rlsfoosubaróubarN©r©r´rOr<rµrNr-rérrrÚ test_dataäs   zInt32Tests.test_dataN) r#r$r%r&rrr³r§rárrrrrrÚs rc@seZdZdZdS)Ú TestInt16zƒ A L{basic.Int16StringReceiver} storing received strings in an array. @ivar received: array holding received strings. Nrrrrrrïsrc@s:eZdZdZeZddgZdgZgd¢Zdd„Z dd „Z d S) Ú Int16Testsz/ Test case for int16-prefixed protocol rGrçsaaaaaa)órr/cCsB| ¡}| d¡| |j ¡d¡| d¡| |jdg¡dS)z? Test specific behavior of the 16-bits length. rlsfoosubarrNrrérrrrs   zInt16Tests.test_datacCs2| ¡}dd|jdd}| t|j|¡dS©zA Send too much data: that should cause an error. rÎrérN©r©Z prefixLengthr†ÚAssertionErrorr´©rr«ZtooSendrrrÚtest_tooLongSend szInt16Tests.test_tooLongSendN) r#r$r%r&rrr³r§rárrrrrrr÷s rc@seZdZdZdS)ÚTestInt8z‚ A L{basic.Int8StringReceiver} storing received strings in an array. @ivar received: array holding received strings. Nrrrrrrsrc@s:eZdZdZeZddgZdgZgd¢Zdd„Z dd „Z d S) Ú Int8Testsz. Test case for int8-prefixed protocol rGrçsaaaaaa)ósdzadzr/cCsB| ¡}| d¡| |j ¡d¡| d¡| |jdg¡dS)z> Test specific behavior of the 8-bits length. rlsfoosubarrNrrérrrr&s   zInt8Tests.test_datacCs2| ¡}dd|jdd}| t|j|¡dSr r rrrrr0szInt8Tests.test_tooLongSendN) r#r$r%r&rrr³r§rárrrrrrrs rc@s8eZdZdZdZdZdd„Zdd„Zdd„Zd d „Z d S) ÚOnlyProducerTransportzm Transport which isn't really a transport, just looks like one to someone not looking very hard. FcCs g|_dSrr‚rrrrrBszOnlyProducerTransport.__init__cCs d|_dS)NT©Úpausedrrrrr6Esz$OnlyProducerTransport.pauseProducingcCs d|_dS)NFrrrrrr7Hsz%OnlyProducerTransport.resumeProducingcCs|j |¡dSr)r!r)rr­rrrr¥KszOnlyProducerTransport.writeN) r#r$r%r&rr{rr6r7r¥rrrrr9src@seZdZdZdd„ZdS)ÚConsumingProtocolzC Protocol that really, really doesn't want any more bytes. cCs|j |¡| ¡dSr)r<r¥r6rrrrrTs zConsumingProtocol.lineReceivedN)r#r$r%r&rrrrrrOsrc@seZdZdZdd„ZdS)Ú ProducerTestszM Tests for L{basic._PausableMixin} and L{basic.LineReceiver.paused}. cCsªtƒ}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¡| |j¡d 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°)rr°r±s goodbye )rr°r±riN) rrrKrNrOr!r~rrzr7)rÚprSrrrÚtest_pauseResume^sF                     zProducerTests.test_pauseResumeN)r#r$r%r&rrrrrrYsrc@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)ÚFileSenderTestsz( Tests for L{basic.FileSender}. cCst ¡}| tt|ƒ¡dS)zP L{basic.FileSender} implements the L{IPullProducer} interface. N)r Ú FileSenderrzrr )rÚsenderrrrÚtest_interface§szFileSenderTests.test_interfacecCsBtdƒ}t ¡}t ¡}| ||¡| |j|¡| |j ¡dS)z˜ When L{basic.FileSender.beginFileTransfer} is called, it registers itself with provided consumer, as a non-streaming producer. ó Test contentN) rrryr rÚbeginFileTransferrOÚproducerr~Z streaming)rÚsourceÚconsumerrrrrÚtest_producerRegistered®s  z'FileSenderTests.test_producerRegisteredcCsftdƒ}t ¡}t ¡}| ||¡}| ¡| ¡| |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. rótN) rrryr rr r7Z assertIsNoner!rOÚsuccessResultOfrµ©rr"r#rÚdrrrÚ test_transferºs  zFileSenderTests.test_transfercCs tdƒ}t ¡}t ¡}d|_| ||¡}| ¡| d|  ¡¡| ¡| d|  ¡¡| ¡| d|  ¡¡| ¡| d|  |¡¡| d|  ¡¡dS)zj L{basic.FileSender} reads at most C{CHUNK_SIZE} every time it resumes producing. rr3sTestsTest conr%N) rrryr rZ CHUNK_SIZEr r7rOrµr&r'rrrÚtest_transferMultipleChunksÌs z+FileSenderTests.test_transferMultipleChunkscCsddd„}tdƒ}t ¡}t ¡}| |||¡}| ¡| ¡| d| |¡¡| d|  ¡¡dS)z L{basic.FileSender.beginFileTransfer} takes a C{transform} argument which allows to manipulate the data on the fly. cSs| ¡Sr)Úswapcase)ÚchunkrrrÚ transformész=FileSenderTests.test_transferWithTransform..transformróTs tEST CONTENTN) rrryr rr r7rOr&rµ)rr-r"r#rr(rrrÚtest_transferWithTransformãsz*FileSenderTests.test_transferWithTransformcCsVtdƒ}t ¡}t ¡}| ||¡}| ¡| |¡}| 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) rrryr rr r8ZfailureResultOfZtrapÚ ExceptionrOÚstrrµ)rr"r#rr(ZfailurerrrÚtest_abortedTransfer÷s   z$FileSenderTests.test_abortedTransferN) r#r$r%r&rr$r)r*r/r2rrrrr¢s r);r&rârpÚiorÚtypingrrrZzope.interface.verifyrZtwisted.internetrrZtwisted.internet.interfacesr Ztwisted.internet.protocolr Ztwisted.protocolsr Ztwisted.python.compatr Ztwisted.python.failurer Z twisted.testrZ twisted.trialrrmrr(r›rBZSynchronousTestCaserDr‹rr—rœrÞr¤r¦r¯ràrîZInt32StringReceiverrrZInt16StringReceiverrrZInt8StringReceiverrrrrrZTestCaserrrrrÚsL         HyO,YIj I