o b7@sddlZddlmZddlmZddlmZmZddlm Z ddl m Z ddl m Z ddlmZdd lmZd ZGd d d eZd dZddZGdddeZGdddeZGdddeZGdddeZdS)N)partial)BytesIO)mainprotocol)failure) iterbytes)banana)StringTransport)TestCaselc@seZdZddZdS) MathTestscCspttddttddttdddg}|D]}t}t||j|}t|}|||qdS)NrdiiLi@BiBl) listrangerrint2b128writegetvalueb1282int assertEqual)self funkylistixvyrA/usr/lib/python3/dist-packages/twisted/spread/test/test_banana.py test_int2b128s    zMathTests.test_int2b128N)__name__ __module__ __qualname__rrrrrr s r 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)_selectDialect)rdialectrrr selectDialect#s r"cCs0t}|}||||||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 makeConnectionclear sendEncodedvalue) bananaFactoryobj transportrrrrencode2s    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) rioencClassencr#r FileWrapperr" putResultexpressionReceivedrr*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)rr4rrrr0Xs zBananaTestBase.putResultcCs|jttj|`dSN)r.connectionLostrFailurerCONNECTION_DONEr2rrrtearDownaszBananaTestBase.tearDownN) rrr__doc__rBananar-r3r0r9rrrrr+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|jdksJdS)Nshello)r.r% dataReceivedr,rr4r2rrr test_stringk zBananaTests.test_stringcCs|dddS)z Banana does not support unicode. ``Banana.sendEncoded`` raises ``BananaError`` if called with an instance of ``unicode``. helloz builtins.strN)_unsupportedTypeTestr2rrrtest_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)rAtyper2rrrtest_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)rAr rr2rrrtest_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%assertInstr)rr(nameexcrrrrAs 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%r=r,rrr4assertIsInstanceintr2rrrtest_ints zBananaTests.test_intcCs(|jj}|d}d|d}|d}|S)N)r. prefixLimit)rbytesbitslargestsmallestrrr _getSmallests  zBananaTests._getSmallestcC |}|tj|jj|dS)zz Test that a long above the implementation-specific limit is rejected as too large to be encoded. N)rWrFrrGr.r%)rrVrrrtest_encodeTooLargeLongz#BananaTests.test_encodeTooLargeLongcCj|}|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. rPrN) rWr.setPrefixLimitrRr%r,rtruncaterFrrGr=)rrVencodedrrrtest_decodeTooLargeLong   z#BananaTests.test_decodeTooLargeLongcCs | Sr5)rWr2rrr _getLargests zBananaTests._getLargestcCrX)z Test that a negative long below the implementation-specific limit is rejected as too small to be encoded. N)rarFrrGr.r%)rrUrrrtest_encodeTooSmallLongrZz#BananaTests.test_encodeTooSmallLongcCr[)z Test that a negative long below the implementation specific limit is rejected as too small to be decoded. rPrN) rar.r\rRr%r,rr]rFrrGr=)rrUr^rrrtest_decodeTooSmallLongr`z#BananaTests.test_decodeTooSmallLongcC0|jd|j|j||jddS)Nir.r%r=r,rrr4r2rrr test_integerr?zBananaTests.test_integercCrd)Ni rer2rrr test_negativer?zBananaTests.test_negativecCrd)Ng@rer2rrr test_floatr?zBananaTests.test_floatc CsXddddgddgddd d d ggd ggg}|j||j|j||j|dS) NrQrP>@皙D@sixseveneight rerfoorrr 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. rQrPrirjrkrlrmrnrorprqrrN) sysmaxsizer.r%feedr,rrr4rsrrr 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.r=)rdatabyterrrrys zBananaTests.feedcCd}|tj|j|dS)NsrFrrGryrr{rrrtest_oversizedListzBananaTests.test_oversizedListcCr})Nsr~rrrrtest_oversizedString!rz BananaTests.test_oversizedStringcCs.d}z |j|WdStjyYdSw)Ns)r.r=rrG)r crashStringrrrtest_crashString&s  zBananaTests.test_crashStringcCrd)Nrer2rrrtest_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}. irsrP}rQ~rsrisssN)rr*)r baseIntIn baseNegIn baseIntOutbaseLongIntOut baseNegOutbaseLongNegOutrrrtest_sizedIntegerTypesEs$z"BananaTests.test_sizedIntegerTypesN)rrrr:r>rBrDrErArNrWrYr_rarbrcrfrgrhrurzryrrrrrrrrrr<fs2     r<c@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)rFNotImplementedErrorr.r= legalPbItemr2rrrtest_dialectNotSetlszDialectTests.test_dialectNotSetcCs.t|jd|j|j||j|jdS)zS If the PB dialect has been selected, a PB VOCAB item is accepted. pbN)r"r.r=rrr4vocabr2rrrtest_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}. rN)r"r.rFKeyErrorr= illegalPbItemr2rrrtest_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. rN)r"r.r%rrrr,rr2rrr test_sendPbs zDialectTests.test_sendPbN)rrrr:rrSrr;outgoingVocabularyVOCABrrrrrrrrrrrcs rc@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. sBsrQN)rFrrGdecoder)r undecodable decodablerrrtest_statelessDecodesz%GlobalCoderTests.test_statelessDecodeN)rrrr:rrrrrrs r)rw functoolsrr,rtwisted.internetrrtwisted.pythonrtwisted.python.compatrtwisted.spreadrtwisted.test.proto_helpersr twisted.trial.unittestr _maxintr r"r*r+r<rrrrrrs$       ~*