U c6{i @sdZddlZddlZddlmZmZddlmZGdddZGdddZ dd d d Z d d Z Gdddej Z Gdddej ZdS)z( Tests for L{twisted.persisted.styles}. N)_UniversalPicklingErrorunpickleMethod)unittestc@s&eZdZdZddddZddZdS)Foo Helper class. NreturncCs d|_dS)N test-valueinstance_memberselfrF/tmp/pip-unpacked-wheel-hfbvu0bz/twisted/persisted/test/test_styles.py__init__sz Foo.__init__cCs|jS)z Helper method. r r rrrmethodsz Foo.method)__name__ __module__ __qualname____doc__rrrrrrrsrc@seZdZdZdS)BarrN)rrrrrrrrrsrrcCsdS)z) A sample function for pickling. NrrrrrsampleFunction$srcCs|S)Nr)xrrr*rc@s&eZdZdZddZddddZdS)UniversalPicklingErrorTestsz9 Tests the L{_UniversalPicklingError} exception. cCstdS)z2 Raise L{UniversalPicklingError}. N)rr rrrraise_UniversalPicklingError2sz8UniversalPicklingErrorTests.raise_UniversalPicklingErrorNrcCs|tj|jdS)z^ Handling L{pickle.PicklingError} handles L{_UniversalPicklingError}. N) assertRaisespickle PicklingErrorrr rrrtest_handledByPickleModule8sz6UniversalPicklingErrorTests.test_handledByPickleModule)rrrrrr rrrrr-src@sdeZdZdZddddZddddZdddd Zddd d Zddd d ZddddZ dS)UnpickleMethodTestsz0 Tests for the unpickleMethod function. NrcCs2t}td|t}|||j|||jdS)zj L{unpickleMethod} returns an instance method bound to the instance passed to it. rN)rr assertEqualr assertIsNotr Zfoomrrr 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. r z new-valueN)rcopyrr"r#r r$rrrtest_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$rrr#test_instanceBuildingNameNotPresent\s z7UnpickleMethodTests.test_instanceBuildingNameNotPresentcCstt}||tdS)zi Copying a function returns the same reference, without creating an actual copy. N)r'rr")r frrrtest_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)nrrrexpectedus z9UnpickleMethodTests.test_primeDirective..expectedr)protocol0N)r"rdumpsr)r r4rrrtest_primeDirectiveosz'UnpickleMethodTests.test_primeDirectivecCs|tjtjtdS)zY Pickling a C{lambda} function ought to raise a L{pickle.PicklingError}. N)rrrr7 lambdaExampler rrrtest_lambdaRaisesPicklingError|sz2UnpickleMethodTests.test_lambdaRaisesPicklingError) rrrrr&r(r)r+r8r:rrrrr!@s    r!)rr'rZtwisted.persisted.stylesrrZ twisted.trialrrrrr9ZTestCaserr!rrrrs