FwgMddlmZmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z mZmZmZmZddlmZmZmZerddlZdZdZGdd eZGd d eZGd d eZGddej8Zej<edZdZ dZ!dZ"dZ#e$ej8ds%ej8jJej8_&ddZ'e$ej8dse ej8de'Gdde(Z)Gdde)Z*ddZ+e$ej8dse ej8de+yy))print_functionabsolute_importN)dedent) bind_methodPY26PY3PY2PY27) check_outputSTDOUTCalledProcessErrorcD|jdr|dd}t|S)z, Removes any leading and dedents.  N) startswithr)codes H/var/www/horilla/myenv/lib/python3.12/site-packages/future/tests/base.py reformat_coders% tABx $<cH|jd}t|Dcgc]\}}|jdr|}}}t|Dcgc])\}}|jds|jdr|+}}}t|Dcgc]\}}|jdr|}}}|j|k(sJdd}d}||||ksJd t |Dcgc]}|| c}} t t || } t |Dcgc]}|| c}} t t || } t |Dcgc]}|| c}} t t || }g}tt|D]a}||vr|j| |||vr|j| |5||vr|j||N|j||cdj|Scc}}wcc}}wcc}}wcc}wcc}wcc}w) a Returns the code block with any ``__future__`` import lines sorted, and then any ``future`` import lines sorted, then any ``builtins`` import lines sorted. This only sorts the lines within the expected blocks. See test_order_future_lines() for an example. rfrom __future__ import z from futurez from pastz from builtinszIinternal usage error: dedent the code before calling order_future_lines()c8t|dkDr t|SdS)Nr)lenmaxnumberss rmymaxz!order_future_lines..mymax7s"7|a/s7|6Q6rcJt|dkDr t|StdS)Nrinf)rminfloatrs rmyminz!order_future_lines..mymin:s "7|a/s7|AU5\Arz2the __future__ and future imports are out of order) split enumeraterlstripsorteddictziprangerappendjoin)rlinesilineuufuture_line_numbersfuture_line_numbersbuiltins_line_numbersrr"uulsorted_uufuture_linesflsorted_future_linesblsorted_builtins_lines new_liness rorder_future_linesr9sZ JJt E.7.>N71d"&//2K"LNN-6e,<>D $ > $ <>>/8.>B71d $ @BB ;;=D C$BC 7B & '51D+E EA @A E $9:q%(: ;C %:C!@A #67aq7 8Bs#6;< #89aq9 :B %:B!?@I 3u: ' % %   215 6 % %   03 4 ' '   215 6   U1X &' 99Y ]N>B$;8:s#H.H  H. H H HceZdZdZddZdZy)VerboseCalledProcessErrorz Like CalledProcessError, but it displays more information (message and script output) for diagnosing test failures etc. Nc<||_||_||_||_yN)msg returncodecmdoutput)selfr>r?r@rAs r__init__z"VerboseCalledProcessError.__init__`s$ rcdd|j|j|j|jfzS)Nz>Command '%s' failed with exit status %d Message: %s Output: %s)r@r?r>rA)rBs r__str__z!VerboseCalledProcessError.__str__fs-R88T__dhh DE Frr=)__name__ __module__ __qualname____doc__rCrErrr;r;[s Frr;c eZdZy) FuturizeErrorNrFrGrHrJrrrLrLjrrLc eZdZy)PasteurizeErrorNrMrJrrrPrPmrNrrPceZdZdZdZ ddZddZdZ ddZdZ dd Z dd Z dd Z de jfd Zy ) CodeHandlerzt Handy mixin for test classes for writing / reading / futurizing / running .py files in the test suite. ctd|_td|_tjg|_t jtjjz|_ tjd}|r0dtjtjz|zi|_ydtji|_y)zi The outputs from the various futurize stages should have the following headers: z from __future__ import absolute_import from __future__ import division from __future__ import print_function a4 from __future__ import absolute_import from __future__ import division from __future__ import print_function from __future__ import unicode_literals from future import standard_library standard_library.install_aliases() from builtins import * PYTHONPATHN)rheaders1headers2sys executable interpreterstempfilemkdtempospathseptempdirgetenvgetcwdpathsepenv)rBpypaths rsetUpzCodeHandler.setUpvs&'   &'   !^^,'')BGGKK7 <( $biikBJJ&>&GHDH$biik2DHrc|r t|}|j||j|||||j}|r#|jD]} |j | } |S)a Converts the code block using ``futurize`` and returns the resulting code. Passing stages=[1] or stages=[2] passes the flag ``--stage1`` or ``stage2`` to ``futurize``. Passing both stages runs ``futurize`` with both stages by default. If from3 is False, runs ``futurize``, converting from Python 2 to both 2 and 3. If from3 is True, runs ``pasteurize`` to convert from Python 3 to both 2 and 3. Optionally reformats the code block first using the reformat() function. If run is True, runs the resulting code under all Python interpreters in self.interpreters. )stages all_importsfrom3 conservative) interpreter)r_write_test_script_futurize_test_script_read_test_scriptrY_run_test_script) rBrrgrhrireformatrunrjrArk_s rconvertzCodeHandler.converts|&  &D % ""&k).\ # K'') #00 C ))k)B C rc~|r"|j|}|j|}t|tr!t|ts|jd}t|tr!t|ts|jd}|j t |j |j y)a Compares whether the code blocks are equal. If not, raises an exception so the test fails. Ignores any trailing whitespace like blank lines. If ignore_imports is True, passes the code blocks into the strip_future_imports method. If one code block is a unicode string and the other a byte-string, it assumes the byte-string is encoded as utf-8. utf-8N)strip_future_imports isinstancebytesdecode assertEqualr9rstrip)rBrAexpectedignore_importss rcomparezCodeHandler.compares ..v6F00:H fe $Z%-H]]7+F h &z&%/Hw/H +FMMO<!* ,rcg}|jdD]e}|jdr|jdr'|jdr9d|vr>d|vrC|jdrU|j|gdj|S)a Strips any of these import lines: from __future__ import from future from future. from builtins or any line containing: install_hooks() or: install_aliases() Limitation: doesn't handle imports split across multiple lines like this: from __future__ import (absolute_import, division, print_function, unicode_literals) rrz from future zfrom builtins zinstall_hooks()zinstall_aliases()z from future.)r#rr*r+)rBrrAr.s rrvz CodeHandler.strip_future_importss(JJt$ $DOO$=>~6'78(D0*d2~6 d# $yy  rc |j||||||} |rd|vr |jn |j} nd} t|} | | vrd} |j | | | z|y)a< Convenience method that calls convert() and compare(). Reformats the code blocks automatically using the reformat_code() function. If all_imports is passed, we add the appropriate import headers for the stage(s) selected to the ``expected`` code-block, so they needn't appear repeatedly in the test code. If ignore_imports is True, ignores the presence of any lines beginning: from __future__ import ... from future import ... for the purpose of the comparison. )rgrhrirqrj)r}N)rsrVrUrr~) rBbeforer|rgrhr}rirqrjrAheaders reformatteds r convert_checkzCodeHandler.convert_checksv*fV$)sO '(F{dmm GG#H- k !G VW{2$2  4rc ,|j||fi|y)ze Convenience method to ensure the code is unchanged by the futurize process. N)r)rBrkwargss r unchangedzCodeHandler.unchangeds 400rmytestscript.pyct|tr|jd}tj|j |zdd5}|j t|dddy#1swYyxYw)z Dedents the given code (a multiline string) and writes it out to a file in a temporary folder like /tmp/tmpUDCn7x/mytestscript.py. ruwtencodingN)rwrxryioopenr_writer)rBrfilenamefs rrlzCodeHandler._write_test_script!s\ dE ";;w'D WWT\\H,dW E " GGF4L ! " " "s A++A4ctj|j|zdd5}|j}ddd|S#1swYSxYw)Nrtrur)rrr_read)rBrr newsources rrnzCodeHandler._read_test_script,sD WWT\\H,dW E !I ! !s AA c g}t|}|r|jd|rd}nNd}|dgk(r|jdn!|dgk(r|jdn |ddgk(sJ|r|jd|j|z}tj|g|zd |gz} t | t |j } | S#t$r} t|5} d d j| d d|jzd|dd| jz} dddn #1swYnxYwd|vrtnt}t| dsd| _| | j | j"| jd} ~ wwxYw)Nz --all-importsz pasteurize.pyz futurize.pyrz--stage1rz--stage2z--conservativez-w)stderrrcError running the command  renv=%s Contents of file : ---- %s ----futurizerArA)listr*r_rWrXr r rcr rr+rrLrPhasattrrAr?r@)rBrrgrhrirjparamsscriptfn call_argsrAerr> ErrorClasss rrmz!CodeHandler._futurize_test_script1sgf  MM/ * $F"F!} j)A3 j)!Q''' ./\\H $^^V,v5r B  H!)FIF( '" Hb Q+ 488+(16683    ,6+?-_J1h'S!,,ahhG G% Hs1B<< E3 E.s rrozCodeHandler._run_test_script`s\\H $ W!;"3&*hhv?F& #" Wb Q+r!23 488+(16683    1h'+CquuQXXV V! Ws-1 C C>B  C B ACCN)rrFFTTF)T)rFTFTF)r)rrFFF)rFrGrHrIrersr~rvrrrlrnrmrWrXrorJrrrRrRqsm)3VEJ6;<,, !DJO<@#(!4F1 " HN7<+0-^):%(^^rrRz#this test is known to fail on Py2.6c<ts|Stj|Sr=)runittestexpectedFailurefuncs rexpectedFailurePY3r   # #D ))rc<ts|Stj|Sr=)rrrrs rexpectedFailurePY26r   # #D ))rc<ts|Stj|Sr=)r rrrs rexpectedFailurePY27rrrc<ts|Stj|Sr=)r rrrs rexpectedFailurePY2rrrassertRaisesRegexct|ttfr|sJdtj|}|j |s+|xsd}|d|j d|}|j|y)z=Fail the test unless the text matches the regular expression.z!expected_regex must not be empty.zRegex didn't matchz: z not found in N)rwstrunicoderecompilesearchpatternfailureException)rBtextexpected_regexr>s r assertRegexrsm.3.1BBB~N3   &))*-~/E/EtL##C(( 'rrc$eZdZ ddZdZdZy)_AssertRaisesBaseContextNc||_||_| |j|_nd|_t |tt frtj|}||_ d|_ y#t$rt ||_YUwxYwr=) r| test_caserFobj_nameAttributeErrorrrwrxrrrr>)rBr|r callable_objrs rrCz!_AssertRaisesBaseContext.__init__sy  "  # 2 , 5 5 !DM nucl 3ZZ7N," 2 #L 1  2sA%%BBc|jj|j|}|jj|r=)r_formatMessager>r)rB standardMsgr>s r _raiseFailurez&_AssertRaisesBaseContext._raiseFailures1nn++DHHkBnn--c22rcx||jdd|_|S|5||i|dddy#1swYyxYw)z If callable_obj is None, assertRaises/Warns is being used as a context manager, so check for a 'msg' kwarg and return self. If callable_obj is not None, call it passing args and kwargs. Nr>)popr>)rBnamerargsrs rhandlez_AssertRaisesBaseContext.handlesF  zz%.DHK  * $ )& ) * * *s 09)NN)rFrGrHrCrrrJrrrrs9= $ 3 *rrceZdZdZdZdZy)_AssertWarnsContextzBA context manager used to implement TestCase.assertWarns* methods.c*tjjD]}t|ddsi|_t j d|_|jj|_t jd|j|S)N__warningregistry__T)recordalways) rWmodulesvaluesgetattrrwarningscatch_warningswarnings_manager __enter__ simplefilterr|)rBvs rrz_AssertWarnsContext.__enter__sy##% +Aq/6(*% +!) 7 7t D--779 h 6 rc|jj||||y |jj}d}|j D]}|j}t||js&||}|j%|jjt |s[||_ |j|_ |j|_ y|>|jdj|jj t ||j"r,|jdj||j"y|jdj|y#t$rt |j}YOwxYw)Nz"{}" does not match "{}"z{} not triggered by {}z{} not triggered)r__exit__r|rFrrrmessagerwrrwarningrlinenorformatrr)rBexc_type exc_valuetbexc_namefirst_matchingmws rrz_AssertWarnsContext.__exit__sb &&xB?    *}}--H A Aa/%!"##/''..s1v6DLJJDM((DK    %   9@@((00#n2E G H ==   7>>x?C}} N O   188B C3 *4==)H *sE&&FFN)rFrGrHrIrrrJrrrrsL  DrrcDt|||}|jd|||S)aFail unless a warning of class warnClass is triggered by callable_obj when invoked with arguments args and keyword arguments kwargs. If a different type of warning is triggered, it will not be handled: depending on the other warning filtering rules in effect, it might be silenced, printed out, or raised as an exception. If called with callable_obj omitted or None, will return a context object used like this:: with self.assertWarns(SomeWarning): do_something() An optional keyword argument 'msg' can be provided when assertWarns is used as a context object. The context manager keeps a reference to the first matching warning as the 'warning' attribute; similarly, the 'filename' and 'lineno' attributes give you information about the line of Python code from which the warning was triggered. This allows you to inspect the warning after the assertion:: with self.assertWarns(SomeWarning) as cm: do_something() the_warning = cm.warning self.assertEqual(the_warning.some_attribute, 147) assertWarns)rr)rBexpected_warningrrrcontexts rrrs(8""2D,GG >>-tV DDrrr=), __future__rrr\rZrrWrrrtextwrapr future.utilsrrrr r future.moves.subprocessr r r unittest2rr9r;rLrPTestCaserRskipIfskip26rrrrrassertRaisesRegexprrobjectrrrrJrrrsF6   ::LL = @ F 2 F -  / G(##GV D E* * * *x  "56*2*;*;*N*NH')x  -0!!=+> *v *D.D2.DbE>x  -0!!=+>1r