U n5{ié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)ŚStringTransport)Śfailure)Ś iterbytes)Śbanana)ŚTestCasel’’’’c@seZdZdd„ZdS)Ś MathTestscCspttddƒƒttddƒƒttddƒƒdg}|D]6}tƒ}t ||j”| ”}t |”}| ||”q4dS)NrédičiLi@Bi¤Bl) ŚlistŚrangerr Zint2b128ŚwriteŚgetvalueZb1282intŚ assertEqual)ŚselfZ funkylistŚiŚxŚvŚy©rśC/tmp/pip-unpacked-wheel-mnhg7r0q/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)rŚdialectrrrŚ 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ŚobjŚ transportr rrrŚ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,t‚dS)Nshello)r)r!Ś dataReceivedr'rr-ŚAssertionErrorr+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)r6Ś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)r6r rr+rrrŚtest_unsupportedUserType€sz$BananaTests.test_unsupportedUserTypecCs0| tj|jj|”}| d|›dt|ƒ”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Śexcrrrr6ˆ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!r3r'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)rJr;r r<r)r!)rrIrrrŚ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. rCrN) rJr)ŚsetPrefixLimitrEr!r'rŚtruncater;r r<r3)rrIŚencodedrrrŚtest_decodeTooLargeLong·s   z#BananaTests.test_decodeTooLargeLongcCs | ” Sr.)rJr+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)rPr;r r<r)r!)rrHrrrŚ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. rCrN) rPr)rLrEr!r'rrMr;r r<r3)rrHrNrrrŚtest_decodeTooSmallLongŠs   z#BananaTests.test_decodeTooSmallLongcCs0|j d”|j |j ””| |jd”dS)Ni÷©r)r!r3r'rrr-r+rrrŚ test_integerŽs zBananaTests.test_integercCs0|j d”|j |j ””| |jd”dS)Ni ü’’rSr+rrrŚ test_negativećs zBananaTests.test_negativecCs0|j d”|j |j ””| |jd”dS)Ngø@rSr+rrrŚ test_floatčs zBananaTests.test_floatc CsXddddgddgddd d d ggd ggg}|j |”|j |j ””| |j|”dS) NrDrCééē€>@ēš™™™™D@éósixósevenóeighté é rS©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. rDrCrWrXrYrZr[r\r]r^r_r`éž’’’N) ŚsysŚmaxsizer)r!Śfeedr'rrr-rarrrŚ 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)r3)rŚdataŚbyterrrrfs zBananaTests.feedcCsd}| tj|j|”dS)Ns€©r;r r<rf©rrhrrrŚtest_oversizedListszBananaTests.test_oversizedListcCsd}| tj|j|”dS)Ns‚rjrkrrrŚtest_oversizedString!sz BananaTests.test_oversizedStringcCs0d}z|j |”Wntjk r*YnXdS)Ns€)r)r3r r<)rZ crashStringrrrŚtest_crashString&s  zBananaTests.test_crashStringcCs0|j d”|j |j ””| |jd”dS)Né€rSr+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’’’rosrCó}rDó~rós…óórWósƒsƒs†óN)rr%)rZ baseIntInZ baseNegInZ baseIntOutZbaseLongIntOutZ baseNegOutZbaseLongNegOutrrrŚtest_sizedIntegerTypesEs$z"BananaTests.test_sizedIntegerTypesN)rrrr0r5r7r9r:r6rArJrKrOrPrQrRrTrUrVrbrgrfrlrmrnrprxrrrrr2fs0     r2c@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)r;ŚNotImplementedErrorr)r3Ś 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)r3r|rr-Ś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}. r~N)rr)r;ŚKeyErrorr3Ś 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. r~N)rr)r!rrr|r'rr+rrrŚ test_sendPbƒs zDialectTests.test_sendPbN)rrrr0rrFr r1ZoutgoingVocabularyZVOCABr|r‚r}r€rƒr„rrrrrycsryc@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…srDN)r;r r<Śdecoder)rZ undecodableZ decodablerrrŚtest_statelessDecode’sz%GlobalCoderTests.test_statelessDecodeN)rrrr0r‡rrrrr…sr…)rdŚ functoolsrr'rZtwisted.internetrrZtwisted.internet.testingrZtwisted.pythonrZtwisted.python.compatrZtwisted.spreadr Ztwisted.trial.unittestr Z_maxintr rr%r&r2ryr…rrrrŚs"       ~*