o ¯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)| ¡| ¡|j d¡|j d|j¡dS|dkrH| ¡dS|dd…dkr[t|dd…ƒ|_ dS|  d ¡ri|j   |d ¡dS|  d ¡ru|j   ¡dSdS) z| Receive line and make some action for some tokens: pause, rawpause, stop, len, produce, unproduce. óópauserórawpauseóstopNéslen óproduceFó unproduce)r2rrÚpauseProducingr-Ú callLaterÚresumeProducingÚ stopProducingÚintÚlengthÚ startswithÚ transportÚregisterProducerÚunregisterProducerrrrrr Ns(      ÿzLineTester.lineReceivedcCs^|d|j…||jd…}}|jt|ƒ|_|jd||jd<|jdkr-| |¡dSdS)zj Read raw data, until the quantity specified by a previous 'len' line is reached. Néÿÿÿÿr)r@Úlenr2r")rr$Úrestrrrr%gs  ÿzLineTester.rawDataReceivedcCs2t|ƒ|jdkr| ||jdd…¡dSdS)z< Adjust line mode when long lines received. r!N)rFÚ MAX_LENGTHr"rrrrÚlineLengthExceededrsÿzLineTester.lineLengthExceededr) r&r'r(r)r*rHrr3r r%rIrrrrr+2s   r+c@s(eZdZdZdZdZdd„Zdd„ZdS) ÚLineOnlyTesterz) A buffering line only receiver. rr,cCr/r0r1rrrrr3‚r.zLineOnlyTester.connectionMadecCó|j |¡dS)z% Save received data. N©r2rrrrrr ˆózLineOnlyTester.lineReceivedN)r&r'r(r)r*rHr3r rrrrrJzs  rJc@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 5r4s67890slen 1óacCó‚tddƒD]9}t ¡}tƒ}| t |¡¡tt|jƒ|dƒD]}|j|||d|…}|  |¡q!|  |j |j ¡qdS)zl Test buffering for different packet size, checking received matches expected data. r!é N) ÚrangerÚStringIOWithoutClosingr+ÚmakeConnectionrÚ FileWrapperrFÚbufferÚ dataReceivedÚ assertEqualÚoutputr2©rÚ packet_sizeÚtÚaÚiÚsrrrÚ test_buffer·s ùzLineReceiverTests.test_buffers!twiddle1 twiddle2 pause twiddle3 )ótwiddle1ótwiddle2r5ótwiddle3cCó¦tddƒD]K}t ¡}t ¡}t|ƒ}| t |¡¡tt |j ƒ|dƒD]}|j |||d|…}|  |¡q&|  |j |j¡| d¡|  |j|j¡qdS)zo Test pause inside data receiving. It uses fake clock to see if pausing/resuming work. r!rSrN)rTrrUrÚClockr+rVrrWrFÚpauseBufrYrZÚ pauseOutput1r2ÚadvanceÚ pauseOutput2©rr]r^r-r_r`rarrrÚ test_pausingÊs  özLineReceiverTests.test_pausings/twiddle1 twiddle2 len 5 rawpause 12345twiddle3 )rcrdrPr6r4)rcrdrPr6s12345recCrf)z7 Test pause inside raw date receiving. r!rSrN)rTrrUrrgr+rVrrWrFÚ rawpauseBufrYrZÚrawpauseOutput1r2rjÚrawpauseOutput2rlrrrÚtest_rawPausingçs  öz!LineReceiverTests.test_rawPausings"twiddle1 twiddle2 stop more stuff )rcrdr7cCrR)z- Test stop inside producing. r!rSN) rTrrUr+rVrrWrFÚstop_bufrYrZÚ stop_outputr2r\rrrÚ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 )r9s hello worldr:ógoodbyeN) r+rrUrVrrWrYrZr2)rr_r^rrrÚ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@óeZdZdd„ZdS)z@LineReceiverTests.test_clearLineBuffer..ClearingReceivercSs||_| ¡|_dSr)rÚclearLineBufferrGrrrrr szMLineReceiverTests.test_clearLineBuffer..ClearingReceiver.lineReceivedN)r&r'r(r rrrrÚClearingReceiveró rys foo bar bazófoosbar bazsquux squuxr4N)r Ú LineReceiverrYrZrrG)rryrrrrÚ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óxr4N) rrrUrVrrWÚsysÚgetrecursionlimitrYrZÚjoinr)rÚprotorBÚlimitrrrÚtest_stackRecursion*s z%LineReceiverTests.test_stackRecursioncCóBt ¡}t ¡}| |¡| d|jdd¡| |j¡dS©zx C{LineReceiver} disconnects the transport if it receives a line longer than its C{MAX_LENGTH}. r~r!ó rN© r r|rÚStringTransportrVrYrHÚ assertTrueÚ disconnecting©rr‚rBrrrÚtest_maximumLineLength5ó  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). r8r~r!NrEr) r+rHrr‰rVrYr*Ú assertFalser‹rZrFr2)rr‚r^rrrrÚ&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 r~Nr!) r r|r*rr‰rVrYrHrFrr‹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)NÚoops)Ú RuntimeError©r$rrrÚosz5LineReceiverTests.test_rawDataError..sdataN) r r|r%rr‰rVrrYÚassertIsInstancer“)rr‚rBÚwhyrrrÚtest_rawDataErroris   z#LineReceiverTests.test_rawDataErrorcCót ¡}| t|jd¡dS)z‰ When L{LineReceiver.rawDataReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. ÚfooN)r r|Ú assertRaisesÚNotImplementedErrorr%©rr‚rrrÚ"test_rawDataReceivedNotImplementedvóz4LineReceiverTests.test_rawDataReceivedNotImplementedcCr™)z† When L{LineReceiver.lineReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. ršN)r r|r›rœr rrrrÚtest_lineReceivedNotImplemented~rŸz1LineReceiverTests.test_lineReceivedNotImplementedN)r&r'r(r)rXr[rbrhrirkrmrnrorprqrrrsrtrvr}r„rrr‘r˜ržr rrrrrNs2       rNc@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. cCrr)Ú longLinesrrrrr3rz*ExcessivelyLargeLineCatcher.connectionMadecCsdS)z/ Disregard any received lines. Nrrrrrr ’sz(ExcessivelyLargeLineCatcher.lineReceivedcCrK)zF Record any data that exceeds the line length limits. N)r¢rr#rrrrI—rMz.ExcessivelyLargeLineCatcher.lineLengthExceededN)r&r'r(r)r3r rIrrrrr¡‡s  r¡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¡r‚rHrr‰rBrVrrrrÚ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}. r~éN)r‚rHrYrZr¢©rÚ 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. r~r¦N)r‚rHrYr*rZr¢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}. r~r¦N)r‚r*rrHrYrZr¢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}. r~r¦N)r‚rHr*rYrZr¢r§rrrÚtest_multipleLongLinesËs  z:LineReceiverLineLengthExceededTests.test_multipleLongLinescCr…r†rˆrŒrrrrÖrŽ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}. r~N) r r|rr‰rVrYrHrFr*rŠr‹rŒrrrÚtest_maximumLineLengthRemainingás  zCLineReceiverLineLengthExceededTests.test_maximumLineLengthRemainingN) r&r'r(r)r¥r©rªr«r¬rr­rrrrr£žs    r£c@r) Ú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. rNrE) rr‰rJrVr rXrYrZr2Úsplit)rr^r_Úcrrrrbø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). r~r!r‡N) rJrr‰rVrYrHrFr*rŠr‹rŒrrrÚ!test_greaterThanMaximumLineLengths ÿz7LineOnlyReceiverTests.test_greaterThanMaximumLineLengthcCr™)zŠ When L{LineOnlyReceiver.lineReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. ršN)r ÚLineOnlyReceiverr›rœr rrrrr rŸz5LineOnlyReceiverTests.test_lineReceivedNotImplementedN)r&r'r(r)rXrbr±r rrrrr®ís  r®c@s:eZdZdd„Zdd„ZdZdZefdedd fd d „Z d S) Ú TestMixincCrrr1rrrrr3rzTestMixin.connectionMadecCó|j |¡dSrrL©rrarrrÚstringReceivedózTestMixin.stringReceivedé2rÚreasonÚreturnNcCó d|_dS)Nr!)Úclosed)rr¹rrrÚconnectionLost#rzTestMixin.connectionLost) r&r'r(r3r¶rHr¼r r r½rrrrr³s r³c@rw)Ú TestNetstringcCs|j |¡|j |¡dSr)r2rrBÚwriterµrrrr¶(s zTestNetstring.stringReceivedN)r&r'r(r¶rrrrr¾'rzr¾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}. )rr‰rrV)rr^r_rrrÚ getProtocol2s zLPTestCaseMixin.getProtocolcCs>|jD]}| ¡}t|ƒD]}| |¡q | |jj¡qdS)zO Assert that illegal strings cause the transport to be closed. N)rÁrÃr rYrŠrBr‹©rraÚrr°rrrÚ test_illegal<ó   ü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)rr‰rBr¾ÚnetstringReceiverrVrrrrr¥Ws zNetstringReceiverTests.setUpcCs tddƒD]H}t ¡}tƒ}d|_| |¡|jD]}| |¡q| ¡}tt |ƒ|dƒD]}||||d|…}|rD|  |¡q/|  |j |j¡qdS)zI Strings can be received in chunks of different lengths. r!rSi»N) rTrr‰r¾rHrVÚstringsÚ sendStringÚvaluerFrYrZr2)rr]r^r_raÚoutr`rrrrb\s    €ôz"NetstringReceiverTests.test_buffercCó"|j d¡| |jjdg¡dS)zE Empty netstrings (with length '0') can be received. s0:,r4N©rÍrYrZr2rrrrÚtest_receiveEmptyNetstringnó z1NetstringReceiverTests.test_receiveEmptyNetstringcCrÒ)z; One-character netstrings can be received. ó1:a,rQNrÓrrrrÚtest_receiveOneCharacterurÕz/NetstringReceiverTests.test_receiveOneCharactercCrÒ)z; Two-character netstrings can be received. s2:ab,óabNrÓrrrrÚtest_receiveTwoCharacters|rÕz0NetstringReceiverTests.test_receiveTwoCharacterscCrÒ)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_receiveNestedNetstringcCó|j d¡| |jj¡dS)zL Netstrings containing more data than expected are refused. s2:aaa,N©rÍrYrŠrBr‹rrrrÚtest_moreDataThanSpecifiedŒó z1NetstringReceiverTests.test_moreDataThanSpecifiedcCrÛ)zƒ Netstrings that should be empty according to their length specification are refused if they contain data. s0:a,NrÜrrrrÚ$test_moreDataThanSpecifiedBorderCase“ó z;NetstringReceiverTests.test_moreDataThanSpecifiedBorderCasecCrÛ)z` Netstrings without leading digits that specify the length are refused. s:aaa,NrÜrrrrÚtest_missingNumber›ràz)NetstringReceiverTests.test_missingNumbercCrÛ)zg Netstrings without a colon between length specification and data are refused. s3aaa,NrÜrrrrÚtest_missingColon£ràz(NetstringReceiverTests.test_missingColoncCrÛ)zY Netstrings that have no leading digits nor a colon are refused. saaa,NrÜrrrrÚtest_missingNumberAndColon«ràz1NetstringReceiverTests.test_missingNumberAndColoncCrÛ)zA Netstrings consisting only of data are refused. saaaNrÜrrrrÚ test_onlyData³rÞz$NetstringReceiverTests.test_onlyDatacCs>|j d¡|j d¡| |jjdg¡| |j ¡¡dS)z= Netstrings can be received in two portions. s4:aasaa,saaaaN)rÍrYrZr2rŠÚ_payloadCompleterrrrÚtest_receiveNetstringPortions_1ºs  z6NetstringReceiverTests.test_receiveNetstringPortions_1cCó,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ó,rONrÓ©rÚpartrrrÚtest_receiveNetstringPortions_2Ãsz6NetstringReceiverTests.test_receiveNetstringPortions_2cCrç)zE Netstrings can be received one character at a time. )ó2ó:rQó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,1rQs:b,rïN)rÍrYrŠrårZr2rrrrÚ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. r!r4rîrQN)rÍrHrYrrÈrŠrBr‹)rÚtooLongrrrÚ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Ú_consumeLengthrZÚ_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örHr÷rZrørrrrÚtest_consumeLengthBorderCase1ôs z4NetstringReceiverTests.test_consumeLengthBorderCase1cCó&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örHr›r ÚNetstringParseErrorr÷rrrrÚtest_consumeLengthBorderCase2þó  ÿz4NetstringReceiverTests.test_consumeLengthBorderCase2cCrü)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 rz4NetstringReceiverTests.test_consumeLengthBorderCase3cCr™)z When L{NetstringReceiver.stringReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. ršN)r ÚNetstringReceiverr›rœr¶rrrrÚ!test_stringReceivedNotImplementedrŸz8NetstringReceiverTests.test_stringReceivedNotImplementedN)r&r'r(r)rÎrÁr¾rr¥rbrÔr×rÙrÚrÝrßrárârãrärærìrðrñrórùrûrrrrrrrrÊGs6        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]}| |¡qq| |j |j¡dS)z> Test receiving data find the same data send. N) rÃrÎr ÚstructÚpackÚ structFormatrFrYrZr2)rrÅrar°rrrÚ test_receive)s   ÿzIntNTestCaseMixin.test_receivecCs>|jD]}| ¡}t|ƒD]}| |¡q | |jg¡qdS)zK Send partial data, nothing should be definitely received. N)rrÃr rYrZr2rÄrrrÚ test_partial3rÇzIntNTestCaseMixin.test_partialcCs6| ¡}| d¡| |j ¡t |jd¡d¡dS)z2 Test sending data over protocol. óbbbbbbbbbbbbbbbbéN)rÃrÏrZrBrÐrrr ©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. rSrýN) rÃrÚlengthLimitExceededrHrYrrr rZ)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}. rSrýs xxxxxxxxxxxN)rÃrHrYrrr rZr2rrrrÚtest_longStringNotDeliveredTsz-IntNTestCaseMixin.test_longStringNotDeliveredcCr™)zŽ When L{IntNStringReceiver.stringReceived} is not overridden in a subclass, calling it raises C{NotImplementedError}. ršN)r ÚIntNStringReceiverr›rœr¶rrrrr_rŸz3IntNTestCaseMixin.test_stringReceivedNotImplemented)r&r'r(r)rrÉrÎrrrÈrÁrr r rrrrrrrrrs      rc@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. )rrr rF)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©ÚreceivedString©rÅÚresultrrr¶~r·zKRecvdAttributeMixin.test_recvdContainsRemainingData..stringReceivedéóaaaaasbbbbN)rÃr¶rrr rYrZ)rr¶ÚcompleteMessageÚincompleteMessagerrrÚ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)rrr©ÚmessageCrÅrrrr¶’sz=RecvdAttributeMixin.test_recvdChanged..stringReceivedrsbbbbbN)rÃrr¶rYrZ)rÚpayloadCr¶ÚpayloadAÚpayloadBÚmessageAÚmessageBrr!rÚtest_recvdChangedˆs   z%RecvdAttributeMixin.test_recvdChangedcsŒ| ¡‰g}d‰|jD]}| | ˆ|¡¡| ˆ¡q g‰‡‡‡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)rrrFr©ÚSWITCHr‚rrrr¶®s z:RecvdAttributeMixin.test_switching..stringReceivedr4óN) rÃrÎrrr¶rYrrZr)rÚmixrar¶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 longr!csˆ |¡ˆ ˆj¡dSrr)r@©r‚rrrrÈs zPRecvdAttributeMixin.test_recvdInLengthLimitExceeded..lengthLimitExceededrN)rÃrFrHrrrYrZ)rÚDATAÚmessagerrr.rÚtest_recvdInLengthLimitExceeded»s  z3RecvdAttributeMixin.test_recvdInLengthLimitExceededN) r&r'r(r)rr r(r-r1rrrrrhs rc@óeZdZdZdS)Ú TestInt32zƒ A L{basic.Int32StringReceiver} storing received strings in an array. @ivar received: array holding received strings. N©r&r'r(r)rrrrr3Òór3c@s2eZdZdZeZddgZdgZgd¢Zdd„Z dS) Ú Int32Testsz/ Test case for int32-prefixed protocol rQr s aaaaaa)só hello therer4cCóB| ¡}| d¡| |j ¡d¡| d¡| |jdg¡dS)z? Test specific behavior of the 32-bits length. r{sfoosubaróubarN©rÃrÏrZrBrÐrYr2rrrrÚ test_dataäó   zInt32Tests.test_dataN) r&r'r(r)r3rrÎrÁrr;rrrrr6Ús r6c@r2)Ú TestInt16zƒ A L{basic.Int16StringReceiver} storing received strings in an array. @ivar received: array holding received strings. Nr4rrrrr=ïr5r=c@ó:eZdZdZeZddgZdgZgd¢Zdd„Z dd „Z d S) Ú Int16Testsz/ Test case for int16-prefixed protocol rQr saaaaaa)ór7r4cCr8)z? Test specific behavior of the 16-bits length. r{sfoosubarr9Nr:rrrrr;r<zInt16Tests.test_datacCó2| ¡}dd|jdd}| t|j|¡dS©zA Send too much data: that should cause an error. rïr¦ér!N©rÃÚ prefixLengthr›ÚAssertionErrorrÏ©rrÅÚtooSendrrrÚtest_tooLongSend ózInt16Tests.test_tooLongSendN) r&r'r(r)r=rrÎrÁrr;rIrrrrr?÷ó r?c@r2)ÚTestInt8z‚ A L{basic.Int8StringReceiver} storing received strings in an array. @ivar received: array holding received strings. Nr4rrrrrLr5rLc@r>) Ú Int8Testsz. Test case for int8-prefixed protocol rQr saaaaaa)ósdzadzr4cCr8)z> Test specific behavior of the 8-bits length. r{sfoosubarr9Nr:rrrrr;&r<zInt8Tests.test_datacCrArBrDrGrrrrI0rJzInt8Tests.test_tooLongSendN) r&r'r(r)rLrrÎrÁrr;rIrrrrrMrKrMc@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. FcCrrr”rrrrrBrzOnlyProducerTransport.__init__cCr»)NT©Úpausedrrrrr;Erz$OnlyProducerTransport.pauseProducingcCr»)NFrPrrrrr=Hrz%OnlyProducerTransport.resumeProducingcCr´r)r$r)rrÈrrrr¿Kr·zOnlyProducerTransport.writeN) r&r'r(r)rQr‹rr;r=r¿rrrrrO9s rOc@óeZdZdZdd„ZdS)ÚConsumingProtocolzC Protocol that really, really doesn't want any more bytes. cCs|j |¡| ¡dSr)rBr¿r;rrrrr Ts  zConsumingProtocol.lineReceivedN)r&r'r(r)r rrrrrSOó rSc@rR)Ú 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Ë)rVrËrÌs goodbye )rVrËrÌruN) rSrOrVrYrZr$rrQrŠr=)rÚpr^rrrÚtest_pauseResume^sF                     zProducerTests.test_pauseResumeN)r&r'r(r)rXrrrrrUYrTrUc@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 Ú FileSenderrŠrr )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) rrr‰r rZÚbeginFileTransferrZÚproducerrÚ streaming)rÚsourceÚconsumerr[rrrÚ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) rrr‰r rZr^r=Ú assertIsNoner_rZÚsuccessResultOfrЩrrarbr[Ú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. r]r8sTestsTest conrdN) rrr‰r rZÚ CHUNK_SIZEr^r=rZrÐrfrgrrrÚ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) rrr‰r rZr^r=rZrfrÐ)rrnrarbr[rhrrrÚ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. r]z#Consumer asked us to stop producingN) rrr‰r rZr^r>ÚfailureResultOfÚtrapÚ ExceptionrZÚstrrÐ)rrarbr[rhÚfailurerrrÚtest_abortedTransfer÷s   z$FileSenderTests.test_abortedTransferN) r&r'r(r)r\rcrirkrprvrrrrrY¢s  rY);r)rrÚiorÚtypingrrrÚzope.interface.verifyrÚtwisted.internetrrÚtwisted.internet.interfacesr Útwisted.internet.protocolr Útwisted.protocolsr Útwisted.python.compatr Útwisted.python.failurer Ú twisted.testrÚ twisted.trialrr|rr+r²rJÚSynchronousTestCaserNr¡r£r®r³rr¾rÀrÊrrÚInt32StringReceiverr3r6ÚInt16StringReceiverr=r?ÚInt8StringReceiverrLrMrOrSrUÚTestCaserYrrrrÚsN         HyO,YIj I