o b @sdZddlZddlZddlmZmZddlmZGdddZGdddZ d d Z d d Z Gd ddej Z Gdddej ZdS)z( Tests for L{twisted.persisted.styles}. N)_UniversalPicklingErrorunpickleMethod)unittestc@ eZdZdZddZddZdS)Foo Helper class. cCs d|_dS)N test-valueinstance_memberselfr D/usr/lib/python3/dist-packages/twisted/persisted/test/test_styles.py__init__s z Foo.__init__cCs|jS)z Helper method. r r r r rmethodsz Foo.methodN)__name__ __module__ __qualname____doc__rrr r r rrs rc@seZdZdZdS)BarrN)rrrrr r r rrsrcCsdS)z) A sample function for pickling. Nr r r r rsampleFunction$rcCs|S)Nr )xr r r*rrc@r)UniversalPicklingErrorTestsz9 Tests the L{_UniversalPicklingError} exception. cCst)z2 Raise L{UniversalPicklingError}. )rr r r rraise_UniversalPicklingError2sz8UniversalPicklingErrorTests.raise_UniversalPicklingErrorcCs|tj|jdS)z^ Handling L{pickle.PicklingError} handles L{_UniversalPicklingError}. N) assertRaisespickle PicklingErrorrr r r rtest_handledByPickleModule8sz6UniversalPicklingErrorTests.test_handledByPickleModuleN)rrrrrrr r r rr-s rc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)UnpickleMethodTestsz0 Tests for the unpickleMethod function. cCs2t}td|t}|||j|||jdS)zj L{unpickleMethod} returns an instance method bound to the instance passed to it. rN)rr assertEqualr assertIsNotr foomr r r test_instanceBuildingNamePresentEs z4UnpickleMethodTests.test_instanceBuildingNamePresentcCsTt}t|j}|||j|||j|d|d|_|d|dS)za Copying an instance method returns a new method with the same behavior. rz new-valueN)rcopyrr!r"r r#r r rtest_instanceCopyMethodOs z+UnpickleMethodTests.test_instanceCopyMethodcCs2t}td|t}|||j|||jdS)z If the named method is not present in the class, L{unpickleMethod} finds a method on the class of the instance and returns a bound method from there. rN)rrrr!rr"r#r r r#test_instanceBuildingNameNotPresent\s z7UnpickleMethodTests.test_instanceBuildingNameNotPresentcCstt}||tdS)zi Copying a function returns the same reference, without creating an actual copy. N)r'rr!)r fr r rtest_copyFunctiongs z%UnpickleMethodTests.test_copyFunctioncCs&dd}|tjtdd|ddS)zd We do not contaminate normal function pickling with concerns from Twisted. cSs"ddttjd|dgdS)N cp.ascii)joinrrencode)nr r rexpectedus z9UnpickleMethodTests.test_primeDirective..expectedr)protocol0N)r!rdumpsr)r r4r r rtest_primeDirectiveosz'UnpickleMethodTests.test_primeDirectivecCs|tjtjtdS)zY Pickling a C{lambda} function ought to raise a L{pickle.PicklingError}. N)rrrr7 lambdaExampler r r rtest_lambdaRaisesPicklingError|sz2UnpickleMethodTests.test_lambdaRaisesPicklingErrorN) rrrrr&r(r)r+r8r:r r r rr @s    r )rr'rtwisted.persisted.stylesrr twisted.trialrrrrr9TestCaserr r r r rs