a )¶(bï7ã@sÔddlZddlmZddlmZddlmZmZddlm Z ddl m Z ddl m Z ddlmZdd lmZd ZGd d „d eƒZd d„Zdd„ZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZdS)éN)Úpartial)ÚBytesIO)ÚmainÚprotocol)Úfailure)Ú iterbytes)Úbanana)ÚStringTransport)ÚTestCaselÿÿÿÿc@seZdZdd„ZdS)Ú MathTestscCspttddƒƒttddƒƒttddƒƒdg}|D]6}tƒ}t ||j¡| ¡}t |¡}| ||¡q4dS)NrédièiLi@Bi¤Bl) ÚlistÚrangerrZint2b128ÚwriteÚgetvalueZb1282intÚ assertEqual)ÚselfZ funkylistÚiÚxÚvÚy©rú>lib/python3.9/site-packages/twisted/spread/test/test_banana.pyÚ test_int2b128s  ÿ þýÿ zMathTests.test_int2b128N)Ú__name__Ú __module__Ú __qualname__rrrrrr sr cCs| |¡dS)zÜ Dictate a Banana dialect to use. @param protocol: A L{banana.Banana} instance which has not yet had a dialect negotiated. @param dialect: A L{bytes} instance naming a Banana dialect to select. N)Z_selectDialect)rZdialectrrrÚ selectDialect#s rcCs0tƒ}|ƒ}| |¡| ¡| |¡| ¡S)a« Banana encode an object using L{banana.Banana.sendEncoded}. @param bananaFactory: A no-argument callable which will return a new, unconnected protocol instance to use to do the encoding (this should most likely be a L{banana.Banana} instance). @param obj: The object to encode. @type obj: Any type supported by Banana. @return: A L{bytes} instance giving the encoded form of C{obj}. )r ÚmakeConnectionÚclearÚ sendEncodedÚvalue)Z bananaFactoryÚobjZ transportrrrrÚencode2s   r#c@s.eZdZdZejZdd„Zdd„Zdd„Z dS) ÚBananaTestBasezn The base for test classes. It defines commonly used things and sets up a connection for testing. cCsNtƒ|_| ¡|_|j t |j¡¡t|jdƒ|j|j_ t t |jƒ|_ dS)Nsnone) rÚioÚencClassÚencrrZ FileWrapperrÚ putResultZexpressionReceivedrr#©rrrrÚsetUpPs    zBananaTestBase.setUpcCs ||_dS)z¨ Store an expression received by C{self.enc}. @param result: The object that was received. @type result: Any type supported by Banana. N)Úresult)rr+rrrr(XszBananaTestBase.putResultcCs|j t tj¡¡|`dS©N)r'ZconnectionLostrZFailurerZCONNECTION_DONEr)rrrÚtearDownaszBananaTestBase.tearDownN) rrrÚ__doc__rÚBananar&r*r(r-rrrrr$Hs  r$c@sÈeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Zd,d-„Zd.d/„Zd0S)1Ú BananaTestsz General banana tests. cCs0|j d¡|j |j ¡¡|jdks,J‚dS)Nshello)r'r Ú dataReceivedr%rr+r)rrrÚ test_stringks zBananaTests.test_stringcCs| dd¡dS)z” Banana does not support unicode. ``Banana.sendEncoded`` raises ``BananaError`` if called with an instance of ``unicode``. Zhelloz builtins.strN)Ú_unsupportedTypeTestr)rrrÚtest_unsupportedUnicodepsz#BananaTests.test_unsupportedUnicodecCs| td¡dS)zÀ Banana does not support arbitrary builtin types like L{type}. L{banana.Banana.sendEncoded} raises L{banana.BananaError} if called with an instance of L{type}. z builtins.typeN)r3Útyper)rrrÚtest_unsupportedBuiltinTypewsz'BananaTests.test_unsupportedBuiltinTypecCs| tƒtd¡dS)zæ Banana does not support arbitrary user-defined types (such as those defined with the ``class`` statement). ``Banana.sendEncoded`` raises ``BananaError`` if called with an instance of such a type. z .MathTestsN)r3r rr)rrrÚtest_unsupportedUserType€sz$BananaTests.test_unsupportedUserTypecCs0| tj|jj|¡}| d|›dt|ƒ¡dS)aû Assert that L{banana.Banana.sendEncoded} raises L{banana.BananaError} if called with the given object. @param obj: Some object that Banana does not support. @param name: The name of the type of the object. @raise: The failure exception is raised if L{Banana.sendEncoded} does not raise L{banana.BananaError} or if the message associated with the exception is not formatted to include the type of the unsupported object. zBanana cannot send z objectsN)Ú assertRaisesrÚ BananaErrorr'r ZassertInÚstr)rr"ÚnameÚexcrrrr3ˆs z BananaTests._unsupportedTypeTestcCs>|j d¡|j |j ¡¡| |jd¡| |jt¡dS)z× A positive integer less than 2 ** 32 should round-trip through banana without changing value and should come out represented as an C{int} (regardless of the type which was encoded). i§'N) r'r r1r%rrr+ZassertIsInstanceÚintr)rrrÚtest_int˜s zBananaTests.test_intcCs(|jj}|d}d|d}|d}|S)Nééé)r'Ú prefixLimit)rÚbytesÚbitsÚlargestÚsmallestrrrÚ _getSmallest£s  zBananaTests._getSmallestcCs | ¡}| tj|jj|¡dS)zz Test that a long above the implementation-specific limit is rejected as too large to be encoded. N)rGr8rr9r'r )rrFrrrÚtest_encodeTooLargeLong¯sz#BananaTests.test_encodeTooLargeLongcCsj| ¡}|j |jjd¡|j |¡|j ¡}|j d¡|j |jjd¡| t j |jj |¡dS)zz Test that a long above the implementation specific limit is rejected as too large to be decoded. r@rN) rGr'ÚsetPrefixLimitrBr r%rÚtruncater8rr9r1)rrFÚencodedrrrÚtest_decodeTooLargeLong·s   z#BananaTests.test_decodeTooLargeLongcCs | ¡ Sr,)rGr)rrrÚ _getLargestÅszBananaTests._getLargestcCs | ¡}| tj|jj|¡dS)zƒ Test that a negative long below the implementation-specific limit is rejected as too small to be encoded. N)rMr8rr9r'r )rrErrrÚtest_encodeTooSmallLongÈsz#BananaTests.test_encodeTooSmallLongcCsj| ¡}|j |jjd¡|j |¡|j ¡}|j d¡|j |jjd¡| t j |jj |¡dS)zƒ Test that a negative long below the implementation specific limit is rejected as too small to be decoded. r@rN) rMr'rIrBr r%rrJr8rr9r1)rrErKrrrÚtest_decodeTooSmallLongÐs   z#BananaTests.test_decodeTooSmallLongcCs0|j d¡|j |j ¡¡| |jd¡dS)Ni÷©r'r r1r%rrr+r)rrrÚ test_integerÞs zBananaTests.test_integercCs0|j d¡|j |j ¡¡| |jd¡dS)Ni üÿÿrPr)rrrÚ test_negativeãs zBananaTests.test_negativecCs0|j d¡|j |j ¡¡| |jd¡dS)Ng¸@rPr)rrrÚ test_floatès zBananaTests.test_floatc CsXddddgddgddd d d ggd ggg}|j |¡|j |j ¡¡| |j|¡dS) NrAr@ééç€>@çš™™™™D@éósixósevenóeighté é rP©rZfoorrrÚ test_listís ø zBananaTests.test_listc Cslddddgddgddd d d ggd gtjdtjdtjd g }|j |¡| |j ¡¡| |j|¡dS)zj Test feeding the data byte per byte to the receiver. Normally data is not split. rAr@rTrUrVrWrXrYrZr[r\r]éþÿÿÿN) ÚsysÚmaxsizer'r Úfeedr%rrr+r^rrrÚ test_partialüs õ zBananaTests.test_partialcCst|ƒD]}|j |¡qdS)z‡ Feed the data byte per byte to the receiver. @param data: The bytes to deliver. @type data: L{bytes} N)rr'r1)rÚdataZbyterrrrcs zBananaTests.feedcCsd}| tj|j|¡dS)Ns€©r8rr9rc©rrerrrÚtest_oversizedListszBananaTests.test_oversizedListcCsd}| tj|j|¡dS)Ns‚rfrgrrrÚtest_oversizedString!sz BananaTests.test_oversizedStringcCs.d}z|j |¡Wntjy(Yn0dS)Ns€)r'r1rr9)rZ crashStringrrrÚtest_crashString&s  zBananaTests.test_crashStringcCs0|j d¡|j |j ¡¡| |jd¡dS)Né€rPr)rrrÚtest_crashNegativeLong9s z"BananaTests.test_crashNegativeLongcCsPd}d}d}| | |d¡d|¡| | |d¡d|¡| | |d¡d |¡d }| | |d¡d |¡| | |d¡d |¡| | |d ¡d|¡d}| | |d¡d|¡| | |d¡d |¡| | |d¡d¡d}| | |d¡d |¡| | |d¡d|¡| | |d ¡d|¡dS)zÇ Test that integers below the maximum C{INT} token size cutoff are serialized as C{INT} or C{NEG} and that larger integers are serialized as C{LONGINT} or C{LONGNEG}. iÿÿÿrksr@ó}rAó~rós…óórTósƒsƒs†óN)rr#)rZ baseIntInZ baseNegInZ baseIntOutZbaseLongIntOutZ baseNegOutZbaseLongNegOutrrrÚtest_sizedIntegerTypesEs$z"BananaTests.test_sizedIntegerTypesN)rrrr.r2r4r6r7r3r>rGrHrLrMrNrOrQrRrSr_rdrcrhrirjrlrtrrrrr0fs0     r0c@sZeZdZdZdZeejjefƒej Z edƒej Z dd„Z dd„Z dd „Zd d „Zd S) Ú DialectTestsz2 Tests for Banana's handling of dialects. sremote)ézcCs| t|jj|j¡dS)z| If no dialect has been selected and a PB VOCAB item is received, L{NotImplementedError} is raised. N)r8ÚNotImplementedErrorr'r1Ú legalPbItemr)rrrÚtest_dialectNotSetlszDialectTests.test_dialectNotSetcCs.t|jdƒ|j |j¡| |j|j¡dS)zS If the PB dialect has been selected, a PB VOCAB item is accepted. ópbN)rr'r1rxrr+Úvocabr)rrrÚtest_receivePbss zDialectTests.test_receivePbcCs$t|jdƒ| t|jj|j¡dS)zž If the PB dialect has been selected and an unrecognized PB VOCAB item is received, L{banana.Banana.dataReceived} raises L{KeyError}. rzN)rr'r8ÚKeyErrorr1Ú illegalPbItemr)rrrÚtest_receiveIllegalPb{s z"DialectTests.test_receiveIllegalPbcCs2t|jdƒ|j |j¡| |j|j ¡¡dS)zl if pb dialect is selected, the sender must be able to send things in that dialect. rzN)rr'r r{rrxr%rr)rrrÚ test_sendPbƒs zDialectTests.test_sendPbN)rrrr.r{rCrr/ZoutgoingVocabularyZVOCABrxr~ryr|rr€rrrrrucsruc@seZdZdZdd„ZdS)ÚGlobalCoderTestszM Tests for the free functions L{banana.encode} and L{banana.decode}. cCs0d}| tjtj|¡d}| t |¡d¡dS)zJ Calls to L{banana.decode} are independent of each other. sB…srAN)r8rr9Údecoder)rZ undecodableZ decodablerrrÚtest_statelessDecode’sz%GlobalCoderTests.test_statelessDecodeN)rrrr.rƒrrrrrsr)raÚ functoolsrr%rZtwisted.internetrrZtwisted.pythonrZtwisted.python.compatrZtwisted.spreadrZtwisted.test.proto_helpersr Ztwisted.trial.unittestr Z_maxintr rr#r$r0rurrrrrÚs"       ~*