U  W[C @sdZddlmZmZddlZddlZddlmZmZddl m Z ddl m Z m Z ddlmZddlmZdZe rvd Zn e sd Zd d d ddddddddg ZGdddZGdddeZGdddejZGdddeZddZeedS) z2 Tests for L{twisted.python._oldstyle._oldStyle}. )absolute_importdivisionN)namedAnyfullyQualifiedName) getModule)_PY3_shouldEnableNewStyle)unittest) _oldstylezNot relevant on Python 3.z#Not running with TWISTED_NEWSTYLE=1ztwisted._threadsztwisted.applicationztwisted.internetztwisted.loggerztwisted.pluginsztwisted.positioningztwisted.protocols.haproxyztwisted.pythonztwisted.scriptz twisted.tapz twisted.trialc@seZdZdZdZddZdS)SomeOldStyleClassz I am a docstring! bazcCsdS)zX A function on an old style class. @return: "hi", for testing. hi)selfrr>/usr/lib/python3/dist-packages/twisted/test/test_nooldstyle.pyfunc1szSomeOldStyleClass.funcN)__name__ __module__ __qualname____doc__barrrrrrr +sr c@seZdZdZdS)SomeNewStyleClassz Some new style class! N)rrrrrrrrr;src@sReZdZdZddZddZddZdd Zer6d e_ e rHe e_ e e_ nd e_ d S) OldStyleDecoratorTestsz+ Tests for L{_oldstyle._oldStyle}. cCstGdddt}|t|tjt|}|t|t||jttf|| d||j ddS)z L{_oldstyle._oldStyle} wraps an old-style class and returns a new-style class that has the same functions, attributes, etc. c@s eZdZdS)zLOldStyleDecoratorTests.test_makesNewStyle..SomeClassThatUsesOldStyleN)rrrrrrrSomeClassThatUsesOldStyleLsrr r N) r assertEqualtypetypes ClassTyper _oldStyle __bases__objectrr)rr updatedClassrrrtest_makesNewStyleGs z)OldStyleDecoratorTests.test_makesNewStylecCs>tt}||jtj||jtj||jtjdS)z The class returned by L{_oldstyle._oldStyle} has the same C{__name__}, C{__module__}, and docstring (C{__doc__}) attributes as the original. N)r rr rrrrrr!rrrtest_carriesAttributesWs z-OldStyleDecoratorTests.test_carriesAttributesc Cs8|t}ttW5QRX||jjdddS)zp Using L{_oldstyle._oldStyle} on a new-style class on Python 2 will raise an exception. rztwisted.python._oldstyle._oldStyle is being used to decorate a new-style class (twisted.test.test_nooldstyle.SomeNewStyleClass). This should only be used to decorate old-style classes.N)Z assertRaises ValueErrorr rrrZ exceptionargs)rerrrtest_onlyOldStyleMayBeDecoratedcs   z6OldStyleDecoratorTests.test_onlyOldStyleMayBeDecoratedcCs.tt}|t|tt||tdS)zw On Python 3 or on Py2 when C{TWISTED_NEWSTYLE} is not set, L{_oldStyle._oldStyle} is a no-op. N)r rr rrZassertIsr#rrrtest_noOpByDefaultrs z)OldStyleDecoratorTests.test_noOpByDefaultzOnly relevant on Py2.z7Only relevant when not running under TWISTED_NEWSTYLE=1N) rrrrr"r$r(r)rskip_skiprrrrrBs  rc@seZdZdZeZddZdS) NewStyleOnlyz A base testclass that takes a module and tests if the classes defined in it are old-style. CAVEATS: This is maybe slightly dumb. It doesn't look inside functions, for classes defined there, or nested classes. c Cszt|j}Wn2tk r@}ztd|W5d}~XYnXg}t|D]8\}}t|drP|j |jkrPt |t j rP| t|qP|rd|_tD]}|j|rt|dqtdj|jd|ddS) zJ Test that C{self.module} has no old-style classes in it. zNot importable: {}Nrz2Not all classes are made new-style yet. See #8243.todoz$Old-style classes in {module}: {val}z, )moduleval)rr. ImportErrorr ZSkipTestformatinspectZ getmembershasattrr isinstancerrappendrr-forbiddenModules startswithdelattrZFailTestjoin)rr.r'ZoldStyleClassesnamer/xrrrtest_newStyleClassesOnlys,"     z%NewStyleOnly.test_newStyleClassesOnlyN)rrrrr+r*r<rrrrr,sr,cstdD]tdddg}fdd|D}d|kr6q Gfdd d ttj}jd d }||_t|d rr||_ | ||iq d S)z Build the test classes that use L{NewStyleOnly}, one class per module. @param _locals: The global C{locals()} dict. Ztwistedztwisted.test.reflect_helperztwisted.internet.test.process_ztwisted.test.process_csg|]}j|qSr)r:r7).0Zignoredr;rr sz%_buildTestClasses..TcseZdZdZjZdS)z_buildTestClasses..Testz/ @see: L{NewStyleOnly} N)rrrrr:r.rr>rrTestsr@._rN) rZ walkModulesr,r TestCaser:replacerr3rupdate)Z_localsZignoredModulesZ isIgnoredr@ZacceptableNamerr>r_buildTestClassess rF)rZ __future__rrrr2Ztwisted.python.reflectrrZtwisted.python.modulesrZtwisted.python.compatrrZ twisted.trialr Ztwisted.pythonr r+r6r r rrCrr,rFlocalsrrrrs>   E*"