U W[ 9@s<ddlmZmZddlZddlmZz ddlZWnek rLddlZYnXddl Z zddl m Z Wnek r~dZ YnXdZ z ddlZWnddlZYnXddlmZmZmZddlmZGdd d ejZGd d d ejZGd d d ejZGdddeZGdddeeZGdddejZGdddejZGdddejZ GdddZ!Gddde"Z#Gddde"Z$Gddde"Z%d d!Z&ee$e&Gd"d#d#Z'Gd$d%d%Z(d&d'Z)Gd(d)d)ejZ*Gd*d+d+ejZ+Gd,d-d-Z,Gd.d/d/Z-Gd0d1d1ejZ.Gd2d3d3ejZ/ee e*gZ0dS)4)divisionabsolute_importN)unittest)StringIOzNo cStringIO available.)stylesaotcrefutil)_PY3c@s,eZdZddZddZddZddZd S) VersionTestscCsTGdddtatt}Gdddtjtat|}t|jsPt ddS)Nc@seZdZddZdS) NullVersionedcSs d|_dS)Nrokselfr=/usr/lib/python3/dist-packages/twisted/test/test_persisted.py__init__(szNullVersioned.__init__N__name__ __module__ __qualname__rrrrrr 'sr c@seZdZdZddZdS)r cSs d|_dSNrr rrrrupgradeToVersion1-szNullVersioned.upgradeToVersion1N)rrrpersistenceVersionrrrrrr +szinitial upgrade not run!) objectr pickledumpsr Versionedloads doUpgrader AssertionError)rZpkclZmnvrrrtest_nullVersionUpgrade%s   z$VersionTests.test_nullVersionUpgradecCsGdddtjat}|js$|jr,tdt|}dt_t |}t |js\td|jsjtdt|}t |}t |jdkstd|jdkstddS) Nc@s6eZdZdZdgZdZdZddZddZdd Z d S) MyVersioned garbagedatarcSsd|_dd|_dS)NZxxxcSsdS)Nz cant persistr)qrrr=z&MyVersioned.__init__..)Zsomedatar%rrrrr;szMyVersioned.__init__cSs|jd7_dSr)v3rrrrupgradeToVersion3?szMyVersioned.upgradeToVersion3cSs|jd7_dSr)v4rrrrupgradeToVersion4BszMyVersioned.upgradeToVersion4N) rrrrZpersistenceForgetsr)r+rr*r,rrrrr#5sr#zhasn't been upgraded yetzdidn't do version 3 upgradezdidn't do version 4 upgraderzupgraded unnecessarily) rrr#r)r+r!rrrrr )rZmvpicklobjrrrtest_versionUpgrade3s    z VersionTests.test_versionUpgradecCsGdddtjatdd}tdd}t||f}~~dt_ddt_t|\}}t| |j d| |j d| |j | |j dS) Nc@seZdZddZddZdS)ClassWithCustomHashcSs||_||_dSN)uniquehash)rr3r4rrrrUszClassWithCustomHash.__init__cSs|jSr2)r4rrrr__hash__XszClassWithCustomHash.__hash__N)rrrrr5rrrrr1Tsr1v1rv2rcSs t|ddSNupgradedTsetattrrrrrr'ar(z3VersionTests.test_nonIdentityHash..) rrr1rrrrrr assertEqualr3 assertTruer9)rr6r7Zpklrrrtest_nonIdentityHashRs    z!VersionTests.test_nonIdentityHashcsGdddtjaGdddtjat}t}t|t|}~~dt_fdd}|t_dt_dd t_t|}t | |j j dS) Nc@s eZdZdS) ToyClassANrrrrrrrr?ksr?c@s eZdZdS) ToyClassBNr@rrrrrAmsrArcst|_tdSr2)rryrr rZpklBrrrts zWVersionTests.test_upgradeDeserializesObjectsRequiringUpgrade..upgradeToVersion1cSs t|ddSr8r:rrrrr'yr(zNVersionTests.test_upgradeDeserializesObjectsRequiringUpgrade..) rrr?rArrrrrr r=rBr9)rxrBZpklArrrCr/test_upgradeDeserializesObjectsRequiringUpgradeis   zrErrrrr $sr c@s eZdZdS)VersionedSubClassNr@rrrrrFsrFc@s eZdZdS)SecondVersionedSubClassNr@rrrrrGsrGc@s eZdZdS)VersionedSubSubClassNr@rrrrrHsrHc@s eZdZdS)VersionedDiamondSubClassNr@rrrrrIsrIc@s0eZdZdZddZddZddZdd Zd S) AybabtuTestsz L{styles._aybabtu} gets all of classes in the inheritance hierarchy of its argument that are strictly between L{Versioned} and the class itself. cCs|ttjgdS)zM L{styles._aybabtu} of L{Versioned} itself is an empty list. N)r<r_aybabturrrrrtest_aybabtuStrictEmptysz$AybabtuTests.test_aybabtuStrictEmptycCs|ttgdS)z There are no classes I{between} L{VersionedSubClass} and L{Versioned}, so L{styles._aybabtu} returns an empty list. N)r<rrKrFrrrrtest_aybabtuStrictSubclasssz'AybabtuTests.test_aybabtuStrictSubclasscCs|tttgdS)z With a sub-sub-class of L{Versioned}, L{styles._aybabtu} returns a list containing the intervening subclass. N)r<rrKrHrFrrrrtest_aybabtuSubsubclasss z$AybabtuTests.test_aybabtuSubsubclasscCs|tttttgdS)z For a diamond-shaped inheritance graph, L{styles._aybabtu} returns a list containing I{both} intermediate subclasses. N)r<rrKrIrHrFrGrrrrtest_aybabtuStrictszAybabtuTests.test_aybabtuStrictN)rrr__doc__rLrMrNrOrrrrrJs  rJc@seZdZddZdS) MyEphemeralcCs ||_dSr2rDrrDrrrrszMyEphemeral.__init__NrrrrrrQsrQc@seZdZddZdS)EphemeralTestscCs\td}||jt||jdt|}t|}||jtj| t |ddS)NrD) rQr< __class__rDrrrr EphemeralZ assertFalsehasattr)ror.rrrtest_ephemerals  zEphemeralTests.test_ephemeralN)rrrrZrrrrrTsrTc@seZdZddZddZdS) PickleablecCs ||_dSr2rRrSrrrrszPickleable.__init__cCs|jSr2rRrrrrgetXszPickleable.getXN)rrrrr\rrrrr[sr[c@seZdZdZddZdS) NotPickleablez) A class that is not pickleable. cCs tddS)z9 Raise an exception instead of pickling. zNot serializable.N) TypeErrorrrrr __reduce__szNotPickleable.__reduce__N)rrrrPr_rrrrr]sr]c@seZdZdZddZdS)CopyRegisteredzb A class that is pickleable only because it is registered with the C{copyreg} module. cCs t|_dS)zE Ensure that this object is normally not pickleable. N)r]Z notPickleablerrrrrszCopyRegistered.__init__N)rrrrPrrrrrr`sr`c@seZdZdZdS)CopyRegisteredLoadedz2 L{CopyRegistered} after unserialization. N)rrrrPrrrrrasracCstdfS)z Externally implement C{__reduce__} for L{CopyRegistered}. @param cr: The L{CopyRegistered} instance. @return: a 2-tuple of callable and argument list, in this case L{CopyRegisteredLoaded} and no arguments. r)ra)ZcrrrrreduceCopyRegistereds rbc@seZdZdZddZdS)A dummy class cCsdSr2rrrrramethodsz A.amethodN)rrrrPrerrrrrc srcc@seZdZdZddZdS)BrdcCsdSr2rrrrrbmethodsz B.bmethodN)rrrrPrgrrrrrfsrfcCsdSr2rrrrrfunktionsrhc@sJeZdZdZddZddZejdkr,de_dd Z d d Z e rFe e _d S) PicklingTestsz$Test pickling of extra object types.cCs$tt}t|}||tdSr2)rrrrr<rr.rYrrr test_module!s  zPicklingTests.test_modulecCs(ttj}t|}||tjdS)zt After importing L{twisted.persisted.styles}, it is possible to pickle classmethod objects. N)rrr[r\rr<rjrrrtest_classMethod's  zPicklingTests.test_classMethod)rUr-z_As of Python 3.4 it is no longer possible to globally change the behavior of function pickling.cCsFtd}t|j}t|}||d|t|t|jdS)Nr-)r[rrr\rr<type)rr/r.rYrrrtest_instanceMethod7s   z!PicklingTests.test_instanceMethodcCsLt}|dt|}t|}|t|t|||ddS)Nabc)_oldStyleCStringIOwriterrrr<rmgetvalue)rfr.rYrrr test_stringIO>s    zPicklingTests.test_stringION) rrrrPrkrlsys version_infoskiprnrt skipStringIOrrrrris ric@s eZdZdZesdZddZdS)StringIOTransitionTestsz When pickling a cStringIO in Python 2, it should unpickle as a BytesIO or a StringIO in Python 3, depending on the type of its contents. z4In Python 2 we can still unpickle cStringIO as such.cCs0d}t|}||tj||ddS)zg A cStringIO pickled with bytes in it will yield an L{io.BytesIO} on python 3. sEctwisted.persisted.styles unpickleStringI p0 (S'test' p1 I0 tp2 Rp3 .ZtestN)rrassertIsInstanceiorr<rr)rZpickledStringIWithTextZloadedrrrtest_unpickleBytesIOUs  z,StringIOTransitionTests.test_unpickleBytesION)rrrrPr rwr|rrrrryKsryc@seZdZddZdS) EvilSourcerorcCs||_||j_||jj_dSr2)abcrSrrrreszEvilSourceror.__init__Nrrrrrr}dsr}c@seZdZddZddZdS) NonDictStatecCs|jSr2staterrrr __getstate__kszNonDictState.__getstate__cCs ||_dSr2r)rrrrr __setstate__mszNonDictState.__setstate__N)rrrrrrrrrrjsrc@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS)AOTTestsc CsFddddtdddddtjddtf }tt|}|||dS) Nrg@y@Tr$rUhelloworld)slicerumaxsizeEllipsisrunjellyFromSource jellyToSourcer<)rr/ZrtObjrrrtest_simpleTypesqszAOTTests.test_simpleTypescCsLt}t}|j|_||_tt|jj}|t|j t |dSr2) rcrfrgr~rrrr<Z _selfOfMethodrVZ_classOfMethod)rr~rZim_rrrtest_methodSelfIdentityxsz AOTTests.test_methodSelfIdentitycCsLt}t}|j|_||_tj}t`z|ttjt|W5|t_XdS)z If a class change after an instance has been created, L{aot.unjellyFromSource} shoud raise a C{TypeError} when trying to unjelly the instance. N) rcrfrgr~ assertRaisesr^rrr)rr~rZ savedbmethodrrrtest_methodNotSelfIdentitys z#AOTTests.test_methodNotSelfIdentitycCs&Gdddt}|ttj|dS)z L{aot.jellyToSource} should raise a C{TypeError} when trying to jelly an unknown type without a C{__dict__} property or C{__getstate__} method. c@seZdZeddZdS)z2AOTTests.test_unsupportedType..UnknownTypecSs tdSr2)AttributeErrorrrrr__dict__sz;AOTTests.test_unsupportedType..UnknownType.__dict__N)rrrpropertyrrrrr UnknownTypesrN)rrr^rr)rrrrrtest_unsupportedTypeszAOTTests.test_unsupportedTypec Cstj}d|d}dddddddd d ttj|tg }t|}||||||tt||g}|d |dkst |dd d |d d kst dS) Nr)rmethodrr$rUzhe llo "x world!ugoodbye တ world!g?dr) rZ AOTJellierZ jellyToAOrrhtupleappendrrr!)rZajdltujrrrtest_basicIdentitys(     zAOTTests.test_basicIdentitycCs,t}d|_tt|j|jks(tdS)Nz meringue!)rrrrrr!)rr~rrrtest_nonDictStateszAOTTests.test_nonDictStatecCs"ttt}||tdS)z L{aot.jellyToSource} and L{aot.unjellyFromSource} honor functions registered in the pickle copy registry. N)rrrr`rzra)rrrrr test_copyRegszAOTTests.test_copyRegcCsVttg}t|}t|}|j|ks.t|jj|jks@t|j|jjk sRtdSr2)r}r jellyToAOTunjellyFromAOTr~r!rr)rrYj1ojrrrtest_funkyReferencess    zAOTTests.test_funkyReferencescCsfg}|df}||t|}t|}||dt||dd|||ddddS)zS L{aot.jellyToAOT} can persist circular references through tuples. irrN)rrrrrzrZassertIsr<)rrrrrrrrtest_circularTuples   zAOTTests.test_circularTupleN) rrrrrrrrrrrrrrrrrps    rc@s8eZdZdZddZddZddZdd Zd d Zd S) CrefUtilTestsz Tests for L{crefutil}. cCs ti}|t|jdddS)zO L{crefutil._DictKeyAndValue} only support keys C{0} and C{1}. r$rUN)rZ_DictKeyAndValuer RuntimeError __setitem__rrrrrtest_dictUnknownKeys z!CrefUtilTests.test_dictUnknownKeycCs&t}d|d<|t|jdddS)zI L{crefutil._Defer} can be assigned a key only one time. rrN)rZ_Deferrrrrrrrtest_deferSetMultipleTimessz(CrefUtilTests.test_deferSetMultipleTimescCs(tdddgt}|t|jiddS)z A L{crefutil._Container} where all of its elements are known at construction time is nonsensical and will result in errors in any call to addDependant. rr$rUz ignore-meN)rZ _Containerlistrr!Z addDependant)rZ containerrrr&test_containerWhereAllElementsAreKnowns z4CrefUtilTests.test_containerWhereAllElementsAreKnowncCs|ttjtddS)z If a dictionary key contains a circular reference (which is probably a bad practice anyway) it will be resolved by a L{crefutil._DictKeyAndValue}, not by placing a L{crefutil.NotKnown} into a dictionary key. valueN)rr!dictrrNotKnownrrrr.test_dontPutCircularReferencesInDictionaryKeyss zsT     ])   -n8