U 2ܟj7@sddlZddlmZddlmZddlmZmZddlm Z ddl m Z ddl m Z ddlmZdd lmZd ZGd d d eZd dZddZGdddeZGdddeZGdddeZGdddeZdS)N)partial)BytesIO)mainprotocol)StringTransport)failure) iterbytes)banana)TestCaselc@seZdZddZdS) MathTestscCspttddttddttdddg}|D]6}t}t||j|}t|}|||q4dS)NrdiiLi@BiBl) listrangerr Zint2b128writegetvalueZb1282int assertEqual)selfZ funkylistixvyrC/tmp/pip-unpacked-wheel-4b529toj/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)rdialectrrr 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}. )rmakeConnectionclear sendEncodedvalue)Z bananaFactoryobj transportr rrrencode2s   r%c@s.eZdZdZejZddZddZddZ dS) BananaTestBasezn The base for test classes. It defines commonly used things and sets up a connection for testing. cCsNt|_||_|jt|jt|jd|j|j_ t t |j|_ dS)Nsnone) rioencClassencrrZ FileWrapperr putResultZexpressionReceivedrr%rrrrsetUpPs    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|jttj|`dSN)r)ZconnectionLostrZFailurerZCONNECTION_DONEr+rrrtearDownaszBananaTestBase.tearDownN) rrr__doc__r Bananar(r,r*r/rrrrr&Hs  r&c@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0S)1 BananaTestsz General banana tests. cCs0|jd|j|j|jdks,tdS)Nshello)r)r! dataReceivedr'rr-AssertionErrorr+rrr test_stringks zBananaTests.test_stringcCs|dddS)z Banana does not support unicode. ``Banana.sendEncoded`` raises ``BananaError`` if called with an instance of ``unicode``. Zhelloz builtins.strN)_unsupportedTypeTestr+rrrtest_unsupportedUnicodepsz#BananaTests.test_unsupportedUnicodecCs|tddS)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)r6typer+rrrtest_unsupportedBuiltinTypewsz'BananaTests.test_unsupportedBuiltinTypecCs|ttddS)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)r6r rr+rrrtest_unsupportedUserTypesz$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!ZassertInstr)rr#nameexcrrrr6s z BananaTests._unsupportedTypeTestcCs>|jd|j|j||jd||jtdS)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!r3r'rrr-ZassertIsInstanceintr+rrrtest_ints zBananaTests.test_intcCs(|jj}|d}d|d}|d}|S)N)r) prefixLimit)rbytesbitslargestsmallestrrr _getSmallests  zBananaTests._getSmallestcCs |}|tj|jj|dS)zz Test that a long above the implementation-specific limit is rejected as too large to be encoded. N)rJr;r r<r)r!)rrIrrrtest_encodeTooLargeLongsz#BananaTests.test_encodeTooLargeLongcCsj|}|j|jjd|j||j}|jd|j|jjd|t j |jj |dS)zz Test that a long above the implementation specific limit is rejected as too large to be decoded. rCrN) rJr)setPrefixLimitrEr!r'rtruncater;r r<r3)rrIencodedrrrtest_decodeTooLargeLongs   z#BananaTests.test_decodeTooLargeLongcCs | Sr.)rJr+rrr _getLargestszBananaTests._getLargestcCs |}|tj|jj|dS)z Test that a negative long below the implementation-specific limit is rejected as too small to be encoded. N)rPr;r r<r)r!)rrHrrrtest_encodeTooSmallLongsz#BananaTests.test_encodeTooSmallLongcCsj|}|j|jjd|j||j}|jd|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. rCrN) rPr)rLrEr!r'rrMr;r r<r3)rrHrNrrrtest_decodeTooSmallLongs   z#BananaTests.test_decodeTooSmallLongcCs0|jd|j|j||jddS)Nir)r!r3r'rrr-r+rrr test_integers zBananaTests.test_integercCs0|jd|j|j||jddS)Ni rSr+rrr test_negatives zBananaTests.test_negativecCs0|jd|j|j||jddS)Ng@rSr+rrr test_floats zBananaTests.test_floatc CsXddddgddgddd d d ggd ggg}|j||j|j||j|dS) NrDrC>@皙D@sixseveneight rSrZfoorrr test_lists  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. rDrCrWrXrYrZr[r\r]r^r_r`N) sysmaxsizer)r!feedr'rrr-rarrr test_partials  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)r3)rdatabyterrrrfs zBananaTests.feedcCsd}|tj|j|dS)Nsr;r r<rfrrhrrrtest_oversizedListszBananaTests.test_oversizedListcCsd}|tj|j|dS)Nsrjrkrrrtest_oversizedString!sz BananaTests.test_oversizedStringcCs0d}z|j|Wntjk r*YnXdS)Ns)r)r3r r<)rZ crashStringrrrtest_crashString&s  zBananaTests.test_crashStringcCs0|jd|j|j||jddS)NrSr+rrrtest_crashNegativeLong9s z"BananaTests.test_crashNegativeLongcCsPd}d}d}|||dd||||dd||||dd |d }|||dd ||||dd ||||d d|d}|||dd||||dd ||||ddd}|||dd ||||dd||||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}. irosrC}rD~rsrWsssN)rr%)rZ baseIntInZ baseNegInZ baseIntOutZbaseLongIntOutZ baseNegOutZbaseLongNegOutrrrtest_sizedIntegerTypesEs$z"BananaTests.test_sizedIntegerTypesN)rrrr0r5r7r9r:r6rArJrKrOrPrQrRrTrUrVrbrgrfrlrmrnrprxrrrrr2fs0     r2c@sZeZdZdZdZeejjefej Z edej Z ddZ ddZ dd Zd d Zd S) DialectTestsz2 Tests for Banana's handling of dialects. sremote)zcCs|t|jj|jdS)z| If no dialect has been selected and a PB VOCAB item is received, L{NotImplementedError} is raised. N)r;NotImplementedErrorr)r3 legalPbItemr+rrrtest_dialectNotSetlszDialectTests.test_dialectNotSetcCs.t|jd|j|j||j|jdS)zS If the PB dialect has been selected, a PB VOCAB item is accepted. pbN)rr)r3r|rr-vocabr+rrrtest_receivePbss zDialectTests.test_receivePbcCs$t|jd|t|jj|jdS)z If the PB dialect has been selected and an unrecognized PB VOCAB item is received, L{banana.Banana.dataReceived} raises L{KeyError}. r~N)rr)r;KeyErrorr3 illegalPbItemr+rrrtest_receiveIllegalPb{s z"DialectTests.test_receiveIllegalPbcCs2t|jd|j|j||j|jdS)zl if pb dialect is selected, the sender must be able to send things in that dialect. r~N)rr)r!rrr|r'rr+rrr test_sendPbs zDialectTests.test_sendPbN)rrrr0rrFr r1ZoutgoingVocabularyZVOCABr|rr}rrrrrrrrycsryc@seZdZdZddZdS)GlobalCoderTestszM Tests for the free functions L{banana.encode} and L{banana.decode}. cCs0d}|tjtj|d}|t|ddS)zJ Calls to L{banana.decode} are independent of each other. sBsrDN)r;r r<decoder)rZ undecodableZ decodablerrrtest_statelessDecodesz%GlobalCoderTests.test_statelessDecodeN)rrrr0rrrrrrsr)rd functoolsrr'rZtwisted.internetrrZtwisted.internet.testingrZtwisted.pythonrZtwisted.python.compatrZtwisted.spreadr Ztwisted.trial.unittestr Z_maxintr rr%r&r2ryrrrrrs"       ~*