U  W[ve@s:dZddlmZmZddlmZddlmZmZm Z ddl m Z ddl m Z mZddlmZddlmZdd lmZmZmZGd d d ejZGd d d eZGdddeZeeGdddejZeeGdddejZGdddejZGdddeZGdddeZ GdddeZ!GdddeZ"GdddeZ#Gd d!d!eZ$ee!e#e$Gd"d#d#ejZ%ee"Gd$d%d%e&Z'Gd&d'd'e&Z(Gd(d)d)ej)e(Z*Gd*d+d+ej)Z+Gd,d-d-eZ,ee,Gd.d/d/ejZ-ee,Gd0d1d1ejZ.Gd2d3d3e&Z/Gd4d5d5ejZ0ee,Gd6d7d7ejZ1Gd8d9d9eZ2Gd:d;d;eZ3ee2Gdd?d?e&Z5Gd@dAdAe(ej)Z6GdBdCdCe(ej)Z7GdDdEdEeZ8GdFdGdGe8Z9ee8GdHdIdIe&Z:ee9GdJdKdKe&Z;GdLdMdMeZdRS)Sz0 Test cases for Twisted component architecture. )divisionabsolute_import)wraps) Interface implementer Attribute)AdapterRegistry) comparablecmp)unittest) components)_addHook _removeHookproxyForInterfacec@seZdZdZddZdS)ComporcCs|jd|_|jSNnumselfrE/usr/lib/python3/dist-packages/twisted/python/test/test_components.pyincs z Compo.incN)__name__ __module__ __qualname__rrrrrrrsrc@seZdZddZdS)IAdeptcCs tdSN)NotImplementedErrorrrrr adaptorFuncszIAdept.adaptorFuncN)rrrr rrrrrsrc@seZdZddZdS)IElapsedcCsdS)z 1! Nrrrrr elapsedFunc!szIElapsed.elapsedFuncNrrrr"rrrrr! sr!c@seZdZddZddZdS)AdeptcCs||_d|_dSNroriginalrrZorigrrr__init__(szAdept.__init__cCs|jd|_|j|jfSr)rr'rrrrrr +s zAdept.adaptorFuncN)rrrr)r rrrrr$&sr$c@seZdZddZdS)ElapsedcCsdSrrrrrrr"1szElapsed.elapsedFuncNr#rrrrr*/sr*c@s eZdZdS)ACompNrrrrrrrr+4sr+c@s eZdZdS)BCompNr,rrrrr-6sr-c@s eZdZdS)CCompNr,rrrrr.8sr.c@s eZdZdS)ITestNr,rrrrr/;sr/c@s eZdZdS)ITest2Nr,rrrrr0?sr0c@s eZdZdS)ITest3Nr,rrrrr1Csr1c@s eZdZdS)ITest4Nr,rrrrr2Gsr2c@seZdZddZdS)TestcCsdSrrr(rrrr)Msz Test.__init__Nrrrr)rrrrr3Ksr3c@seZdZdZddZdS)Test2rcCsdSrrr(rrrr)TszTest2.__init__N)rrrZtemporaryAdapterr)rrrrr5Qsr5c@seZdZdZddZdS)RegistryUsingMixinzH Mixin for test cases which modify the global registry somehow. cCs,t}|td|t|}|t|dS)z Configure L{twisted.python.components.registerAdapter} to mutate an alternate registry to improve test isolation. ZglobalRegistryN)rZpatchr r Z addCleanupr)rZscratchRegistryhookrrrsetUp]szRegistryUsingMixin.setUpN)rrr__doc__r8rrrrr6Ysr6c@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS)ComponentizedTestsz8 Simple test case for caching in Componentized. cCs8t|tttttttttttt dSr) r6r8r registerAdapterr3r+r/r1r5r0rrrrr8rs zComponentizedTests.setUpcCsjttttttttt}|tdks8t |tdksNt tt| dksft dS)N)rr)r<r) r r;r$rrr*r! getComponentr AssertionErrorr"rcrrrtestComponentizedzs z$ComponentizedTests.testComponentizedcCst}|t}|t}|t}|t}||ks:t||k sFt|||t}|t}||kspt||k s|tdSr)r.r=r/r0r>ZremoveComponent)rr@co1Zco2co3co4Zco5Zco6rrrtestInheritanceAdaptations          z,ComponentizedTests.testInheritanceAdaptationcCs>t}|t}|t}|t}|||||dSr)r.r=r/r1r2 assertIsNoneassertIs)rr@rBrCrDrrrtestMultiAdapters     z#ComponentizedTests.testMultiAdaptercCsNt}t}||t||||jt|d|||tddS)z Test that a default value specified to Componentized.getComponent if there is no component for the requested interface. )defaultN)r ComponentizedobjectrGr=r/)r componentizedrIrrrtest_getComponentDefaultss  z,ComponentizedTests.test_getComponentDefaultscCs<t}|tt|t}||j|||tdS)z C{Componentized.setAdapter} sets a component for an interface by wrapping the instance with the given adapter class. N) r rJZ setAdapterrr$r= assertEqualr'assertIsInstancerrLZ componentrrrtest_setAdapters   z"ComponentizedTests.test_setAdaptercCs>t}|jtdd|t}||j|||tdS)z C{Componentized.setAdapter} adapts the instance by wrapping it with given adapter class, then stores it using C{addComponent}. T)Z ignoreClassN) r rJZ addAdapterr$r=rrNr'rOrPrrrtest_addAdapters  z"ComponentizedTests.test_addAdaptercCs0t}t}|t|||t|dS)zx C{Componentized.setComponent} stores the given component using the given interface as the key. N)r rJrK setComponentr/rGr=rrLobjrrrtest_setComponents z$ComponentizedTests.test_setComponentcCs8t}t}|t||t||tdS)zm C{Componentized.setComponent} removes the cached component for the given interface. N)r rJrKrSr/ZunsetComponentrFr=rTrrrtest_unsetComponents   z&ComponentizedTests.test_unsetComponentcCs8t}|tdt|}|d||d|dS)zR C{ReprableComponentized} has a C{__repr__} that lists its cache. hellor/N)r ZReprableComponentizedrSr/reprZassertIn)rZrcresultrrrtest_reprableComponentizeds   z-ComponentizedTests.test_reprableComponentizedN)rrrr9r8rArErHrMrQrRrVrWr[rrrrr:ns      r:c@seZdZdZddZdS) AdapterTestszTest adapters.cCs2t}t|}|tjt||t|ddSr)rKr$ assertRaisesr Z CannotAdaptr/rF)roarrrtestAdapterGetComponentsz$AdapterTests.testAdapterGetComponentN)rrrr9r`rrrrr\sr\c@s eZdZdS)IMetaNr,rrrrrasrac@seZdZddZdS) MetaAddercCs |jj|Srr&rrrrraddsz MetaAdder.addNrrrrdrrrrrbsrbc@seZdZddZdS)BackwardsAddercCs |jj|Srr&rcrrrrdszBackwardsAdder.addNrerrrrrfsrfc@seZdZdZddZdS) MetaNumberz9 Integer wrapper for Interface adaptation tests. cCs ||_dSrrrcrrrr) szMetaNumber.__init__N)rrrr9r)rrrrrgsrgc@seZdZddZdS)ComponentNumbercCsd|_tj|dSr%)rr rJr)rrrrr)szComponentNumber.__init__Nr4rrrrrhsrhc@s eZdZdZddZddZdS)ComponentAdderz0 Adder for componentized adapter tests. cCstj|||jj|_dSr)r Adapterr)r'rrr'rrrr)szComponentAdder.__init__cCs|j|7_|jSrrrcrrrrdszComponentAdder.addN)rrrr9r)rdrrrrrisric@seZdZdZddZdS)IAttrXz= Base interface for test of adapter with C{__cmp__}. cCsdS)z! Return a value. Nrrrrrx)szIAttrX.xNrrrr9rmrrrrrl%srlc@seZdZdZddZdS)IAttrXXz@ Adapted interface for test of adapter with C{__cmp__}. cCsdS)z+ Return a tuple of values. Nrrrrrxx4sz IAttrXX.xxN)rrrr9rprrrrro0sroc@seZdZdZddZdS)XcellentzG L{IAttrX} implementation for test of adapter with C{__cmp__}. cCsdS)z; Return a value. @return: a value x!rrrrrrm@sz Xcellent.xNrnrrrrrq;srqc@s,eZdZdZdZddZddZddZd S) DoubleXAdapterz Adapter with __cmp__. *cCs ||_dSr)r'rkrrrr)PszDoubleXAdapter.__init__cCs|j|jfSr)r'rmrrrrrpTszDoubleXAdapter.xxcCst|j|jSr)r r)rotherrrr__cmp__XszDoubleXAdapter.__cmp__N)rrrr9rr)rprvrrrrrsJs rsc@s$eZdZddZddZddZdS)MetaInterfaceTestscCs0tttttd}|t|dddS)zS Registered adapters can be used to adapt classes to an interface. rr<N)r r;rbrgrarNrd)rnrrr test_basic^szMetaInterfaceTests.test_basiccCsJttttt}t|dt|d|t|dddS)Nr)r r;rirhrardrNr?rrrtestComponentizedInteractionfs z/MetaInterfaceTests.testComponentizedInteractioncCs,tttttt}|d|dS)N)rrrr)r r;rsrlrorqrNrp)rrprrrtestAdapterWithCmpms z%MetaInterfaceTests.testAdapterWithCmpN)rrrryr{r|rrrrrw]srwc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd S)!RegistrationTestsz) Tests for adapter registration. cCs0dd}t||t|t|td|dS)z Register an adapter with L{components.registerAdapter} for the given class or interface and verify that the adapter can be looked up with L{components.getAdapterFactory}. cSsdSrrr^rrr~zGRegistrationTests._registerAdapterForClassOrInterface..Nr r;r/rGgetAdapterFactoryrr'Zadapterrrr#_registerAdapterForClassOrInterfacexs  z5RegistrationTests._registerAdapterForClassOrInterfacecCsGdddt}||S)za Test that an adapter from a class can be registered and then looked up. c@s eZdZdS)zCRegistrationTests.test_registerAdapterForClass..TheOriginalNr,rrrr TheOriginalsr)rKrrrrrrtest_registerAdapterForClasssz.RegistrationTests.test_registerAdapterForClasscCs |tS)zf Test that an adapter from an interface can be registered and then looked up. )rr0rrrr test_registerAdapterForInterfacesz2RegistrationTests.test_registerAdapterForInterfacecCsLdd}dd}t||t|ttj||t|t|td|dS)z Verify that L{components.registerAdapter} raises L{ValueError} if the from-type/interface and to-interface pair is not unique. cSsdSNFrr~rrrrrzHRegistrationTests._duplicateAdapterForClassOrInterface..cSsdSNTrr~rrrrrN)r r;r/r] ValueErrorrGr)rr' firstAdapter secondAdapterrrr$_duplicateAdapterForClassOrInterfaces z6RegistrationTests._duplicateAdapterForClassOrInterfacecCsGdddt}||S)zz Test that attempting to register a second adapter from a class raises the appropriate exception. c@s eZdZdS)zDRegistrationTests.test_duplicateAdapterForClass..TheOriginalNr,rrrrrsr)rKrrrrrtest_duplicateAdapterForClasssz/RegistrationTests.test_duplicateAdapterForClasscCs |tS)z Test that attempting to register a second adapter from an interface raises the appropriate exception. )rr0rrrr!test_duplicateAdapterForInterfacesz3RegistrationTests.test_duplicateAdapterForInterfacecCsdd}dd}Gdddt}t|||dt_z(t||||t||d|W5dt_X|ttj||||t||d|dS) z Verify that when C{components.ALLOW_DUPLICATES} is set to C{True}, new adapter registrations for a particular from-type/interface and to-interface pair replace older registrations. cSsdSrrr~rrrrrzORegistrationTests._duplicateAdapterForClassOrInterfaceAllowed..cSsdSrrr~rrrrrc@s eZdZdS)zSRegistrationTests._duplicateAdapterForClassOrInterfaceAllowed..TheInterfaceNr,rrrr TheInterfacesrTFN)rr r;ZALLOW_DUPLICATESrGrr]r)rr'rrrrrr+_duplicateAdapterForClassOrInterfaceAlloweds.  z=RegistrationTests._duplicateAdapterForClassOrInterfaceAllowedcCsGdddt}||S)z Test that when L{components.ALLOW_DUPLICATES} is set to a true value, duplicate registrations from classes are allowed to override the original registration. c@s eZdZdS)zKRegistrationTests.test_duplicateAdapterForClassAllowed..TheOriginalNr,rrrrrsr)rKrrrrr$test_duplicateAdapterForClassAllowedsz6RegistrationTests.test_duplicateAdapterForClassAllowedcCsGdddt}||S)z Test that when L{components.ALLOW_DUPLICATES} is set to a true value, duplicate registrations from interfaces are allowed to override the original registration. c@s eZdZdS)zORegistrationTests.test_duplicateAdapterForInterfaceAllowed..TheOriginalNr,rrrrrsr)rrrrrr(test_duplicateAdapterForInterfaceAllowedsz:RegistrationTests.test_duplicateAdapterForInterfaceAllowedcCsHdd}t||tt|t|td||t|td|dS)zh Verify that an adapter can be registered for multiple to-interfaces at a time. cSsdSrrr~rrrrrzJRegistrationTests._multipleInterfacesForClassOrInterface..N)r r;r/r0rGrrrrr&_multipleInterfacesForClassOrInterfaces  z8RegistrationTests._multipleInterfacesForClassOrInterfacecCsGdddt}||S)zi Test the registration of an adapter from a class to several interfaces at once. c@s eZdZdS)zFRegistrationTests.test_multipleInterfacesForClass..TheOriginalNr,rrrrrsr)rKrrrrrtest_multipleInterfacesForClasssz1RegistrationTests.test_multipleInterfacesForClasscCs |tS)zn Test the registration of an adapter from an interface to several interfaces at once. )rr1rrrr#test_multipleInterfacesForInterfacesz5RegistrationTests.test_multipleInterfacesForInterfacecCsldd}dd}Gddd|}t||tt||t|t|td||t|td|dS)a  Verify that a new adapter can be registered for a particular to-interface from a subclass of a type or interface which already has an adapter registered to that interface and that the subclass adapter takes precedence over the base class adapter. cSsdSrrr~rrrrrzSRegistrationTests._subclassAdapterRegistrationForClassOrInterface..cSsdSrrr~rrrrrc@s eZdZdS)zVRegistrationTests._subclassAdapterRegistrationForClassOrInterface..TheSubclassNr,rrrr TheSubclasssrNr)rr'rrrrrr/_subclassAdapterRegistrationForClassOrInterfaces  zARegistrationTests._subclassAdapterRegistrationForClassOrInterfacecCsGdddt}||S)z~ Test that an adapter to a particular interface can be registered from both a class and its subclass. c@s eZdZdS)zORegistrationTests.test_subclassAdapterRegistrationForClass..TheOriginalNr,rrrrr)sr)rKrrrrr(test_subclassAdapterRegistrationForClass$sz:RegistrationTests.test_subclassAdapterRegistrationForClasscCs |tS)z Test that an adapter to a particular interface can be registered from both an interface and its subclass. )rr0rrrr,test_subclassAdapterRegistrationForInterface.sz>RegistrationTests.test_subclassAdapterRegistrationForInterfaceN)rrrr9rrrrrrrrrrrrrrrrrrrr}ts         r}c@s eZdZdZedZddZdS)IProxiedInterfacez= An interface class for use by L{proxyForInterface}. z@ An example declared attribute, which should be proxied.cOsdS)z: A sample method which should be proxied. Nr)r_kwrrryay?szIProxiedInterface.yayN)rrrr9rifaceAttributerrrrrr7src@seZdZdZddZdS)IProxiedSubInterfacezS An interface that derives from another for use with L{proxyForInterface}. cCsdS)zD A different sample method which should be proxied. NrrrrrbooIszIProxiedSubInterface.booNrrrr9rrrrrrDsrc@s eZdZdZddZddZdS)Yayablez A provider of L{IProxiedInterface} which increments a counter for every call to C{yay}. @ivar yays: The number of times C{yay} has been called. cCsd|_g|_dSr%)yaysyayArgsrrrrr)XszYayable.__init__cOs$|jd7_|j||f|jS)z) Increment C{self.yays}. r)rrappend)rr_rrrrr\sz Yayable.yayN)rrrr9r)rrrrrrOsrc@s(eZdZdZdZdZddZddZdS)Booablez3 An implementation of IProxiedSubInterface FcCs d|_dS)z; Mark the fact that 'yay' has been called. TN)yayedrrrrrmsz Booable.yaycCs d|_dS)z< Mark the fact that 'boo' has been called.1 TN)booedrrrrrtsz Booable.booN)rrrr9rrrrrrrrrfs rc@s eZdZdZddZddZdS)IMultipleMethodsz- An interface with multiple methods. cCsdS)z4 The first method. Should return 1. Nrrrrr methodOneszIMultipleMethods.methodOnecCsdS)z5 The second method. Should return 2. Nrrrrr methodTwoszIMultipleMethods.methodTwoNrrrr9rrrrrrr|src@s eZdZdZddZddZdS)MultipleMethodImplementorz: A precise implementation of L{IMultipleMethods}. cCsdS)z @return: 1 rrrrrrrsz#MultipleMethodImplementor.methodOnecCsdS)z @return: 2 r<rrrrrrsz#MultipleMethodImplementor.methodTwoNrrrrrrsrc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZdS)ProxyForInterfaceTestsz) Tests for L{proxyForInterface}. cCs$t}tt|}||j|dS)z Proxy objects should have an C{original} attribute which refers to the original object passed to the constructor. N)rKrrrGr')rr'proxyrrr test_originals z$ProxyForInterfaceTests.test_originalcCs@tt}t}||}|||d||jddS)z The class created from L{proxyForInterface} passes methods on an interface to the object which is passed to its constructor. r<NrrrrrNr)rklassyayablerrrrtest_proxyMethods z'ProxyForInterfaceTests.test_proxyMethodcsDttGfddd}t}||}||d|jdS)z Methods of the class created from L{proxyForInterface} can be used with the decorator-helper L{functools.wraps}. cs"eZdZejfddZdS)z?ProxyForInterfaceTests.test_decoratedProxyMethod..klasscs|jjd7_|Sr)r'rrrbaserrrszCProxyForInterfaceTests.test_decoratedProxyMethod..klass.yayN)rrrrrrrrrrsrr<Nr)rrr'rrrrtest_decoratedProxyMethods z0ProxyForInterfaceTests.test_decoratedProxyMethodcsBt}t|_tt||j|j|tfdddS)zc Proxy objects should proxy declared attributes, but not other attributes. csjSr)rrrrrrrz.N)rrKrrrrGr]AttributeError)rrrrrtest_proxyAttributes  z*ProxyForInterfaceTests.test_proxyAttributecCs0t}tt|}t}||_||j|dS)zv The attributes that proxy objects proxy should be assignable and affect the original object. N)rrrrKrrGrrrZthingyrrrtest_proxySetAttributes  z-ProxyForInterfaceTests.test_proxySetAttributecCs0t}d|_tt|}|`|t|ddS)zu The attributes that proxy objects proxy should be deletable and affect the original object. Nr)rrrr assertFalsehasattr)rrrrrrtest_proxyDeleteAttributes  z0ProxyForInterfaceTests.test_proxyDeleteAttributecCs6t}tt|}||d||ddS)z [Regression test] The proxy should send its method calls to the correct method, not the incorrect one. rr<N)rrrrNrr)rZmultirrrrtest_multipleMethodss z+ProxyForInterfaceTests.test_multipleMethodscCsJGdddtt}t}||}||jd|||jddS)zP It is possible to subclass the result of L{proxyForInterface}. c@seZdZdZddZdS)zAProxyForInterfaceTests.test_subclassing..SpecializedProxyzU A specialized proxy which can decrement the number of yays. cSs|jjd8_dS)z? Decrement the number of yays. rN)r'rrrrrrszEProxyForInterfaceTests.test_subclassing..SpecializedProxy.booNrrrrrSpecializedProxysrrN)rrrrNrr)rrrZspecialrrrtest_subclassings  z'ProxyForInterfaceTests.test_subclassingcCstt}||jddS)zQ The name of a proxy class indicates which interface it proxies. zA(Proxy for twisted.python.test.test_components.IProxiedInterface)N)rrrNrrrrrrtest_proxyNames z%ProxyForInterfaceTests.test_proxyNamecCstt}|t|dS)zO The resulting proxy implements the interface that it proxies. N)rr assertTrueZ implementedByrrrrtest_implementssz&ProxyForInterfaceTests.test_implementscCsdGdddt}|}|}||_tdd}||||j|d}||||jdS)a\ _ProxyDescriptor's __get__ method should return the appropriate attribute of its argument's 'original' attribute if it is invoked with an object. If it is invoked with None, it should return a false class-method emulator instead. For some reason, Python's documentation recommends to define descriptors' __get__ methods with the 'type' parameter as optional, despite the fact that Python itself never actually calls the descriptor that way. This is probably do to support 'foo.__get__(bar)' as an idiom. Let's make sure that the behavior is correct. Since we don't actually use the 'type' argument at all, this test calls it the idiomatic way to ensure that signature works; test_proxyInheritance verifies the how-Python-actually-calls-it signature. c@seZdZdZddZdS)z>ProxyForInterfaceTests.test_proxyDescriptorGet..SampleFcSs d|_dSr)calledrrrrrX9szDProxyForInterfaceTests.test_proxyDescriptorGet..Sample.helloN)rrrrrXrrrrSample7srrXr'N) rKr'r Z_ProxyDescriptorrN__get__rXrr)rrZ fakeProxyZ testObjectZpdZfakeClassMethodrrrtest_proxyDescriptorGet's  z.ProxyForInterfaceTests.test_proxyDescriptorGetcspGdddttGfddd}t}||}||jddddd d ||jd tddd fgd S) z Subclasses of the class returned from L{proxyForInterface} should be able to upcall methods by reference to their superclass, as any normal Python class can. c@seZdZdZdS)zDProxyForInterfaceTests.test_proxyInheritance..YayableWrapperzI This class does not override any functionality. N)rrrr9rrrrYayableWrapperKsrcs eZdZdZdZfddZdS)zEProxyForInterfaceTests.test_proxyInheritance..EnhancedWrapperz@ This class overrides the 'yay' method. rcs$|jd7_j|f||dS)Nr) wrappedYaysr)rr_krrrrUszIProxyForInterfaceTests.test_proxyInheritance..EnhancedWrapper.yayN)rrrr9rrrrrrEnhancedWrapperPsrrz)rmy)rzrN)rrrrNrrdict)rrrwrapperrrrtest_proxyInheritanceEs z,ProxyForInterfaceTests.test_proxyInheritancecCsBtt}t}||}||||j||jdS)z Proxies of subinterfaces generated with proxyForInterface should allow access to attributes of both the child and the base interfaces. N)rrrrrrrr)rZ proxyClassZbooablerrrrtest_interfaceInheritance`s z0ProxyForInterfaceTests.test_interfaceInheritancecCs~t}t|_ttdd|}||j|||d||j|jt}||_||j||`| t |ddS)a The original attribute name can be customized via the C{originalAttribute} argument of L{proxyForInterface}: the attribute should change, but the methods of the original object should still be callable, and the attributes still accessible. foo)ZoriginalAttributerrN) rrKrrrrGrrNrrrrrrrtest_attributeCustomizationns z2ProxyForInterfaceTests.test_attributeCustomizationN)rrrr9rrrrrrrrrrrrrrrrrrrs       rN)?r9Z __future__rr functoolsrZzope.interfacerrrZzope.interface.adapterrZtwisted.python.compatr r Z twisted.trialr Ztwisted.pythonr Ztwisted.python.componentsr rrrJrrr!rjr$r*r+r-r.r/r0r1r2r3rKr5r6ZSynchronousTestCaser:r\rarbrfrgrhrirlrorqrsrwr}rrrrrrrrrrrsp     y    D