U Acjó§ã@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âeZdZdZdZddddddd d d d d ddg Zdd„ZdZdddgZedgZ dd„Z dZ ddddd gZ ddddddgZ dd„ZdZdddgZd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Zd,d-„Zd.d/„Zd0d1„Zd2d3„Zd4S)5Ú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 aslen 10ó 0123456789slen 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 stwiddle1stwiddle2r0stwiddle3cCs¦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. rrGrN)rHrrIrÚClockr(rJrrKr>ÚpauseBufrMrNÚ pauseOutput1r-ÚadvanceÚ pauseOutput2©rrQrRr*rSrTrUrrrÚ test_pausingÊs  zLineReceiverTests.test_pausings/twiddle1 twiddle2 len 5 rawpause 12345twiddle3 r1s12345cCs¦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. rrGrN)rHrrIrrWr(rJrrKr>Ú rawpauseBufrMrNÚrawpauseOutput1r-rZÚrawpauseOutput2r\rrrÚtest_rawPausingçs  z!LineReceiverTests.test_rawPausings"twiddle1 twiddle2 stop more stuff r2cCs‚tddƒD]r}t ¡}tƒ}| t |¡¡tt|jƒ|dƒD](}|j|||d|…}|  |¡qB|  |j |j ¡q dS)z- Test stop inside producing. rrGN) rHrrIr(rJrrKr>Ústop_bufrMrNÚ stop_outputr-rPrrrÚtest_stopProducingûs z$LineReceiverTests.test_stopProducingcCsBtƒ}t ¡}| t |¡¡| d¡| |jddddg¡dS)z6 Test produce/unproduce in receiving. s&produce hello world unproduce goodbye r4s hello worldr5ógoodbyeN) r(rrIrJrrKrMrNr-)rrSrRrrrÚ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ÚClearingReceiversrgs foo bar bazófoosbar bazsquux squuxr/N)r Ú LineReceiverrMrNrr?)rrgrrrrÚ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) rrrIrJrrKÚsysÚgetrecursionlimitrMrNÚ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}. rkró rN© r rirÚStringTransportrJrMr@Ú assertTrueÚ disconnecting©rror<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). r3rkrNr=r) r(r@rrurJrMr'Ú assertFalserwrNr>r-)rrorRrrrrÚ&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 rkNr) r rir'rrurJrMr@r>rzrwrxrrrÚ(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 rir"rrurJrrMZassertIsInstancer})rror<ÚwhyrrrÚ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 riÚ assertRaisesÚNotImplementedErrorr"©rrorrrÚ"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 rirƒr„rr…rrrÚtest_lineReceivedNotImplemented~sz1LineReceiverTests.test_lineReceivedNotImplementedN)r#r$r%r&rLrOrVrXrYr[r]r^r_r`rarbrcrdrfrjrqryr{r|rr†r‡rrrrrDsVó  ú       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ˆror@rrur<rJrrrrÚ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}. rkéN)ror@rMrNr‰©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. rkrN)ror@rMr'rNr‰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}. rkrN)ror'rnr@rMrNr‰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}. rkrN)ror@r'rMrNr‰rŽrrrÚtest_multipleLongLinesËs  z:LineReceiverLineLengthExceededTests.test_multipleLongLinescCsBt ¡}t ¡}| |¡| d|jdd¡| |j¡dSrrrtrxrrrryÖ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}. rkN) r rirrurJrMr@r>r'rvrwrxrrrÚtest_maximumLineLengthRemainingás  zCLineReceiverLineLengthExceededTests.test_maximumLineLengthRemainingN) r#r$r%r&rŒrrr‘r’ryr“rrrrrŠžs     rŠc@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=) rrurBrJr rLrMrNr-Úsplit)rrRrSÚcrrrrVø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). rkrrsN) rBrrurJrMr@r>r'rvrwrxrrrÚ!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&rLrVr—r‡rrrrr”ís   r”c@s8eZdZdd„Zdd„ZdZdZefeddœd d „Z dS) Ú TestMixincCs g|_dSrr,rrrrr.szTestMixin.connectionMadecCs|j |¡dSrrC©rrUrrrÚstringReceivedszTestMixin.stringReceivedé2rN)ÚreasonÚreturncCs d|_dS)Nr)Úclosed)rrrrrÚ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<Úwriteršrrrr›(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}. )rrurrJ)rrRrSrrrÚ getProtocol1s zLPTestCaseMixin.getProtocolcCs>|jD]2}| ¡}t|ƒD]}| |¡q| |jj¡qdS)zO Assert that illegal strings cause the transport to be closed. N)r¤r¦r rMrvr<rw©rrUÚrr–rrrÚ test_illegal;s    zLPTestCaseMixin.test_illegal) r#r$r%r¤rrÚbytesÚ__annotations__rr¦r©rrrrr£-s   r£c@sâeZdZdZdddddddgZd d d d gZeZd d„Zdd„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zd'd(„Zd)d*„Zd+d,„Zd-d.„Zd/d0„Zd1d2„Zd3d4„Zd5d6„Zd7d8„Zd9S):ÚNetstringReceiverTestszA Tests for L{twisted.protocols.basic.NetstringReceiver}. óhelloóworldshowsaresyou123s:todaysaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaas9999999999999999999999sabcs4:abcdes751:aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaab,cCs$t ¡|_tƒ|_|j |j¡dSr)rrur<r¡ÚnetstringReceiverrJrrrrrŒVs 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. rrGi»N) rHrrur¡r@rJÚstringsÚ sendStringÚvaluer>rMrNr-)rrQrRrSrUÚoutrTrrrrV[s    z"NetstringReceiverTests.test_buffercCs"|j d¡| |jjdg¡dS)zE Empty netstrings (with length '0') can be received. s0:,r/N©r¯rMrNr-rrrrÚtest_receiveEmptyNetstringms z1NetstringReceiverTests.test_receiveEmptyNetstringcCs"|j d¡| |jjdg¡dS)z; One-character netstrings can be received. ó1:a,rFNr´rrrrÚtest_receiveOneCharacterts 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¯rMrvr<rwrrrrÚ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¯rMrNr-rvÚ_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ó:rFó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,1rFs:b,rËN)r¯rMrvrÂrNr-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ÊrFN)r¯r@rMrnrªrvr<rw)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Ú_consumeLengthrNÚ_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ÒrNrÓ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¡rrŒrVrµr·r¹rºr¼r½r¾r¿rÀrÁrÃrÈrÌrÍrÎrÔrÖrÙrÚrÜrrrrr¬Fs<ü         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>rMrNr-)rr¨rUr–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 rMrNr-r§rrrÚ test_partial2s    zIntNTestCaseMixin.test_partialcCs6| ¡}| d¡| |j ¡t |jd¡d¡dS)z2 Test sending data over protocol. óbbbbbbbbbbbbbbbbéN)r¦r±rNr<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. rGr×N) r¦rÚlengthLimitExceededr@rMrßràrárN)rr:r¨rrrÚtest_lengthLimitExceededFs 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}. rGr×s xxxxxxxxxxxN)r¦r@rMrßràrárNr-rærrrÚtest_longStringNotDeliveredSsz-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Ú makeMessagenszRecvdAttributeMixin.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árMrN)rr›ZcompleteMessageZincompleteMessagerrðrÚtest_recvdContainsRemainingDatausz3RecvdAttributeMixin.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›rMrN)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ï©ZSWITCHrorñrrr›­s z:RecvdAttributeMixin.test_switching..stringReceivedr/óN) r¦r°rrìr›rMrnrNrî)rZmixrUr›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:©rorñrrrèÇs zPRecvdAttributeMixin.test_recvdInLengthLimitExceeded..lengthLimitExceededrN)r¦r>r@rìrèrMrN)rZDATAÚmessagerèrrúrÚtest_recvdInLengthLimitExceededºs  z3RecvdAttributeMixin.test_recvdInLengthLimitExceededN) r#r$r%r&rìrôrörùrürrrrrëgs 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ýÑsrýc@s4eZdZdZeZddgZdgZdddgZdd „Z d S) Ú Int32Testsz/ Test case for int32-prefixed protocol rFräs aaaaaasó hello therer/cCsB| ¡}| d¡| |j ¡d¡| d¡| |jdg¡dS)z? Test specific behavior of the 32-bits length. rhsfoosubaróubarN©r¦r±rNr<r²rMr-rærrrÚ test_dataãs   zInt32Tests.test_dataN) r#r$r%r&rýrr°r¤rÞrrrrrrÿÙs  rÿc@seZdZdZdS)Ú TestInt16zƒ A L{basic.Int16StringReceiver} storing received strings in an array. @ivar received: array holding received strings. Nrþrrrrrîsrc@s<eZdZdZeZddgZdgZdddgZdd „Z d d „Z d S) Ú Int16Testsz/ Test case for int16-prefixed protocol rFräsaaaaaaórr/cCsB| ¡}| d¡| |j ¡d¡| d¡| |jdg¡dS)z? Test specific behavior of the 16-bits length. rhsfoosubarrNrræ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Þrr rrrrrös  rc@seZdZdZdS)ÚTestInt8z‚ A L{basic.Int8StringReceiver} storing received strings in an array. @ivar received: array holding received strings. Nrþrrrrr sr c@s<eZdZdZeZddgZdgZdddgZdd „Z d d „Z d S) Ú Int8Testsz. Test case for int8-prefixed protocol rFräsaaaaaaósdzadzr/cCsB| ¡}| d¡| |j ¡d¡| d¡| |jdg¡dS)z> Test specific behavior of the 8-bits length. rhsfoosubarrNrrærrrr%s   zInt8Tests.test_datacCs2| ¡}dd|jdd}| t|j|¡dSrr r rrrr /szInt8Tests.test_tooLongSendN) r#r$r%r&r rr°r¤rÞrr rrrrrs  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~rrrrrAszOnlyProducerTransport.__init__cCs d|_dS)NT©Úpausedrrrrr6Dsz$OnlyProducerTransport.pauseProducingcCs d|_dS)NFrrrrrr7Gsz%OnlyProducerTransport.resumeProducingcCs|j |¡dSr)r!r)rrªrrrr¢JszOnlyProducerTransport.writeN) r#r$r%r&rrwrr6r7r¢rrrrr8src@seZdZdZdd„ZdS)ÚConsumingProtocolzC Protocol that really, really doesn't want any more bytes. cCs|j |¡| ¡dSr)r<r¢r6rrrrrSs zConsumingProtocol.lineReceivedN)r#r$r%r&rrrrrrNsrc@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¡|  ¡| |jdddg¡| |j¡| |j¡| d¡| |jdddg¡| |j¡| |j¡|  ¡| |jddddg¡| |j¡| |j¡|  ¡| |jddddg¡| |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 s hello, worldshello world r­r®s goodbye reN) rrrJrMrNr!rzrrvr7)rÚprRrrrÚtest_pauseResume]sF                     zProducerTests.test_pauseResumeN)r#r$r%r&rrrrrrXsrc@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 Ú FileSenderrvrr )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) rrrur rÚbeginFileTransferrNÚproducerrzZ 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) rrrur rrr7Z assertIsNonerrNÚsuccessResultOfr²©rrrrÚ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) rrrur rZ CHUNK_SIZErr7rNr²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) rrrur rrr7rNr"r²)rr)rrrr$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) rrrur rrr8ZfailureResultOfZtrapÚ ExceptionrNÚstrr²)rrrrr$ZfailurerrrÚtest_abortedTransferös   z$FileSenderTests.test_abortedTransferN) r#r$r%r&rr r%r&r+r.rrrrr¡s r);r&rßrlÚiorÚtypingrrrZzope.interface.verifyrZtwisted.internetrrZtwisted.internet.interfacesr Ztwisted.internet.protocolr Ztwisted.protocolsr Ztwisted.python.compatr Ztwisted.python.failurer Z twisted.testrZ twisted.trialrrirr(r˜rBZSynchronousTestCaserDrˆrŠr”r™rÛr¡r£r¬rÝrëZInt32StringReceiverrýrÿZInt16StringReceiverrrZInt8StringReceiverr rrrrZTestCaserrrrrÚsL         HyO,YIj I