~Mc@sdZddlZyddlZWnek r;eZnXddlmZmZddlm Z m Z ddl m Z ddl mZdeejfdYZd fd YZd Zd fd YZdfdYZdefdYZdefdYZdfdYZde jfdYZdejejfdYZdejejfdYZde jfdYZdS(s/ Test cases for L{jelly} object serialization. iN(tjellytpb(tsett frozenset(tunittest(tStringTransporttTestNodecBs eZdZdZddZRS(sD An object to test jellyfying of new style class instances. icCsH|r)|jd|_|jj|n d|_||_g|_dS(Ni(tidtchildrentappendtparent(tselfR ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt__init__s   N(t__name__t __module__t__doc__t classAttrtNoneR (((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyRstAcBseZdZdZRS(s Dummy class. cCsdS(s; Method tp be used in serialization tests. N((R ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pytamethod+s(R RRR(((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyR&scCsdS(s: A dummy function to test function serialization. N((R ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pytafunc2stBcBseZdZdZRS(s Dummy class. cCsdS(s; Method to be used in serialization tests. N((R ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pytbmethod>s(R RRR(((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyR9stCcBseZdZdZRS(s Dummy class. cCsdS(s; Method to be used in serialization tests. N((R ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pytcmethodJs(R RRR(((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyREstDcBseZdZRS(s Dummy new-style class. (R RR(((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyRQstEcBs5eZdZdZdddZdZdZRS(s+ Dummy new-style class with slots. txtycCs||_||_dS(N(RR(R RR((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyR _s cCsi|jd6|jd6S(NRR(RR(R ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt __getstate__dscCs|d|_|d|_dS(NRR(RR(R tstate((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt __setstate__hs (RRN(R RRt __slots__RR RR(((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyRXs  tSimpleJellyTestcBseZdZdZRS(cCs||_||_dS(N(RR(R RR((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyR os cCs|j|jkS(N(t__dict__(R tother((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt isTheSameAsss(R RR R$(((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyR!ns t JellyTestCasecBseZdZdZdZdZdZdZdZdZ dZ d Z d Z d d d dgd ddgd ddgd ddggZ dZdZdZed,krdZee _ee_ee_ndZdZdZdZdZdZdZdZdZd Zd!Zd"Z d#Z!d$Z"d%Z#d&Z$d'Z%d(Z&d)Z'd*Z(d+Z)RS(-s Testcases for L{jelly} module serialization. @cvar decimalData: serialized version of decimal data, to be used in tests. @type decimalData: C{list} cCsetj|}tj}|jtj|||jj||jtjtj||dS(s Helper test method to test security options for a type. @param inputList: a sample input for the type. @param inputList: C{list} @param atom: atom identifier for the type. @type atom: C{str} N(RtSecurityOptionstallowBasicTypestunjellyt allowedTypestpopt assertRaisest InsecureJelly(R t inputListtatomtcttaster((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt _testSecuritys   cCsbt}t}|j|_||_tjtj|jj}|j|j|jj dS(N( RRRtaRR(t assertEqualstim_classtim_selft __class__(R R2tbtim_((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_methodSelfIdentitys     cCsst}t}t}|j|_||_tj}t`z#|jttj tj|Wd|t_XdS(s If a class change after an instance has been created, L{jelly.unjelly} shoud raise a C{TypeError} when trying to unjelly the instance. N( RRRRRR2R+t TypeErrorRR((R R2R7R/t savecmethod((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_methodsNotSelfIdentitys      #cCsut}d|_t}||_||_tj|}tj|}|j|t|j|j|jdS(Ni(RRtn2tn3RR(tassertIsInstancetassertIdentical(R tnR=R/tm((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt test_newStyles     cCsWt}d|_tj|}tj|}|j|t|j|jddS(s A class defined with I{slots} can be jellied and unjellied with the values for its attributes preserved. iN(RRRR(R?R3(R RAR/RB((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_newStyleWithSlotss   cCs5tg}tjtj|}|j||dS(sn Test that an old style class type can be jellied and unjellied to the original type. N(RRR(R3(R tttr((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_typeOldStyles cCs5tg}tjtj|}|j||dS(sm Test that a new style class type can be jellied and unjellied to the original type. N(RRR(R3(R RERF((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_typeNewStyles cCs5tg}tjtj|}|j||dS(se Test that a builtin type can be jellied and unjellied to the original type. N(tstrRR(R3(R RERF((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_typeBuiltins cCsptjj}tjj|}||g}tj|}tj|}|j|||j||dS(N(tdatetimetnowRR(R3tassertNotIdentical(R tdtntdtdtinputR/toutput((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt test_dateTimes cCsxtjdtjdtjdtjdg}tj|}tj|}|j|||j||dS(s Jellying L{decimal.Decimal} instances and then unjellying the result should produce objects which represent the values of the original inputs. s9.95ii@s-78.901N(tdecimaltDecimalRR(R3RM(R R-R/RQ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt test_decimals   tlistRSiiii@iicCs\tjdtjdtjdtjdg}tj|j}|j||dS(s Unjellying the s-expressions produced by jelly for L{decimal.Decimal} instances should result in L{decimal.Decimal} instances with the values represented by the s-expressions. This test also verifies that C{self.decimalData} contains valid jellied data. This is important since L{test_decimalMissing} re-uses C{self.decimalData} and is expected to be unable to produce L{decimal.Decimal} instances even though the s-expression correctly represents a list of them. s9.95ii@s-78.901N(RSRTRR(t decimalDataR3(R texpectedRQ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_decimalUnjellys   cCs|jtdd tj|j}|jt|dx+tdD]}|j||tj qHW|j|dj d|j|dj d|j|dj d|j|d j d d S( s If decimal is unavailable on the unjelly side, L{jelly.unjelly} should gracefully return L{jelly.Unpersistable} objects. RSiis!Could not unpersist decimal: 9.95isCould not unpersist decimal: 0is#Could not unpersist decimal: 123456is$Could not unpersist decimal: -78.901N( tpatchRRR(RWR3tlentrangeR?t Unpersistabletreason(R RQti((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_decimalMissingscCs&tjdg}|j|ddS(s By default, C{decimal} objects should be allowed by L{jelly.SecurityOptions}. If not allowed, L{jelly.unjelly} should raise L{jelly.InsecureJelly} when trying to unjelly it. s9.95RSN(RSRTR1(R R-((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_decimalSecurity,ssdecimal not availablecCsTtdddgg}tjtj|}|j|||j||dS(s Jellying C{set} instances and then unjellying the result should produce objects which represent the values of the original inputs. iiiN(RRR(R3RM(R R-RQ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_set<scCsTtdddgg}tjtj|}|j|||j||dS(s Jellying C{frozenset} instances and then unjellying the result should produce objects which represent the values of the original inputs. iiiN(RRR(R3RM(R R-RQ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_frozensetHscCs,tdddgg}|j|ddS(s By default, C{set} objects should be allowed by L{jelly.SecurityOptions}. If not allowed, L{jelly.unjelly} should raise L{jelly.InsecureJelly} when trying to unjelly it. iiiRN(RR1(R R-((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_setSecurityTscCs,tdddgg}|j|ddS(s By default, C{frozenset} objects should be allowed by L{jelly.SecurityOptions}. If not allowed, L{jelly.unjelly} should raise L{jelly.InsecureJelly} when trying to unjelly it. iiiRN(RR1(R R-((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_frozensetSecurity^scCstjjdddgg}tj|}|j|tjtdddggtj|}|jt|dt|dttjjkr|j|dtjjn|j|dtdS(s Test jellying C{sets.Set}: it should serialize to the same thing as C{set} jelly, and be unjellied as C{set} if available. iiiiN(Rt_setstSetR3RR(RVR?(R R-t inputJellyRQ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt test_oldSetshs+$cCstjjdddgg}tj|}|j|tjtdddggtj|}|jt|dt|dttjjkr|j|dtjjn|j|dtdS(s Test jellying C{sets.ImmutableSet}: it should serialize to the same thing as C{frozenset} jelly, and be unjellied as C{frozenset} if available. iiiiN(RRft ImmutableSetR3RR(RVR?(R R-RhRQ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_oldImmutableSetsys+$cCso|jtddjtddtdd}tj|}tj|}|j|j|dS(s% Simplest test case. R2R7iiN(t failUnlessR!R$RR((R R2tcerealR7((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt test_simples cCsg}|}|j||j||j|d|d|j|dd|tj|}tj|}|j|d|d|j|dd|dS(sJ Test to make sure that objects retain identity properly. iiN(R R@RR((R RRtstz((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt test_identitys  cCsTtd}tjtj|}|j|||jt|t|dS(Ntblah(tunicodeRR(R3ttype(R RR((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt test_unicodes cCsg}i|d6|f}|j|tj|}tj|}|j|dd|d|j|ddd|dS(NRVii(R RR(R@(R trereft toplevelTupleRoRp((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_stressReferencess cCs]g}|f}|j|ftj|}tj|}|j|ddd|dS(Ni(R RR(R@(R R2RERoRp((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_moreReferencess  cCs;tj}tji}|jtjtj||dS(s@ Test for type-level security of serialization. N(RR&R+R,R((R R0tdct((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_typeSecuritys cCs2tjt}tjt}|jt|dS(N(RRR(R@(R tjtuj((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_newStyleClassessc Cst}tjtj|tjtj|jtdddgtd tdddddttidd6tjg }x0|D](}|j|tjtj|qWdS( sA Test for all types currently supported in jelly iiittestgL4@R2N(iii( RRR(RRtboolRRR3(R R2titemsR_((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_lotsaTypess (" cCsdddYat}t|}t|}t||f}i||6||6||6|d6}tjtj|d}|j|jdj|jdjdS(Nt TupleStatecBs,eZdZdZdZdZRS(cSs ||_dS(N(R#(R R#((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyR scSs |jfS(N(R#(R ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyRscSs|d|_dS(Ni(R#(R R((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyRscSs t|jS(N(thashR#(R ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt__hash__s(R RR RRR(((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyRs   tt3ii((RRRR(R@R#(R R2tt1tt2Rtdtt3prime((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyt test_setStates   "cCstj}|jttt}t}t}||_||_||_||_tj||}tj ||}|j |jtj tj|}|j tj tj |||j|j|jdtjt|}tj ||}|j|td|dS(sA Test for class-level security of serialization. sIdentity mismatchsA came back: %sN(RR&tallowInstancesOfRRRR7R/RR(R?R]R+R,R@(R R0R2R7R/tfriendlyRtmean((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_classSecuritys$        cCsdtjfdY}tj|tj|}d|_tjtj|}|j|jd|j|tjdS(sr Test that if Unjellyable is used to deserialize a jellied object, state comes out right. tJellyableTestClasscBseZRS((R R(((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyRstvalueN(Rt JellyabletsetUnjellyableForClasst Unjellyablet attributeR(R3R?(R RRPRQ((s;/usr/lib/python2.7/dist-packages/twisted/test/test_jelly.pyttest_unjellyables  c Csidg}|d}|d}tdd}tdd}tdd}||_||_||_tj|d |}tj|d |}|j|j|jj|j|d d |j|j|jd dS(NicSs2|dd|d<||d|ds,