jj3XdZddlmZddlZddlmZddlmZddlZddl Z ddl m Z ddl Z ddl mZddlZddlZddlmZdd lmZdd lmZd d l mZd d l mZd dl mZd dl mZd dl mZddlmZddlmZddlmZddlmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl$m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.e.e/e0e1fZ2edZ3eejhGddZ5d,d Z6d-d!Z7d.d"Z8e%d#$d/d%Z9d0d&Z:e% d1d'Z;d2d(Z<ed)d)* d3d+Z=y)4z>Support for providing temporary directories to test functions.) annotationsN) Generator) ExitStack)Path)rmtree)Any)final)Literal)cleanup_dead_symlinks) LOCK_TIMEOUT)make_numbered_dir)make_numbered_dir_with_cleanup)rm_rf) get_user_id)Config)ExitCode)hookimpl)Parser)check_ispytest)fixture)FixtureRequest) MonkeyPatch)Item) TestReport)StashKeyallfailednoneceZdZUdZded<ded<ded<ded<d ed < dd d  ddZed d  ddZddZdddZ ddZ y )TempPathFactoryzFactory for temporary directories under the common base temp directory, as discussed at :ref:`temporary directory location and retention`. Path | None_given_basetempr_trace _basetempint_retention_count RetentionType_retention_policyNF _ispytestct||d|_n6ttjj t ||_||_||_||_ ||_ t|_ yN) rr$rospathabspathstrr%r(r*r&r _exit_stack)selfgiven_basetempretention_countretention_policytracebasetempr,s P/opt/eval-harness/bench_task/venv/lib/python3.12/site-packages/_pytest/tmpdir.py__init__zTempPathFactory.__init__9se y!  !#'D $(N8K(L#MD  /!1!%;c0t|t|jd}|dkrtd|d|jd}|dvrtd|d||jj |j jd||d S) zTCreate a factory according to pytest configuration. :meta private: tmp_path_retention_countrz6tmp_path_retention_count must be >= 0. Current input: .tmp_path_retention_policyrzKtmp_path_retention_policy must be either all, failed, none. Current input: tmpdirT)r5r8r6r7r,)rr'getini ValueErroroptionr9r8get)clsconfigr,countpolicys r: from_configzTempPathFactory.from_configSs y!FMM"<=> 19HqQ :; 2 2]^d]eefg !==11,,""8,!#   r<ctjj|}|j|z j j |jk7rt |d|S)Nz& is not a normalized and relative path)r/r0normpath getbasetempresolveparentrC)r4basenames r:_ensure_relative_to_basetempz,TempPathFactory._ensure_relative_to_basetempss\77##H-     ) 2 2 4 ; ;t?O?O?Q Qz)OPQ Qr<c|j|}|s3|jj|}|jd|St |j|d}|j d||S)aCreate a new temporary directory managed by the factory. :param basename: Directory base name, must be a relative path. :param numbered: If ``True``, ensure the directory is unique by adding a numbered suffix greater than any existing one: ``basename="foo-"`` and ``numbered=True`` means that this function will create directories named ``"foo-0"``, ``"foo-1"``, ``"foo-2"`` and so on. :returns: The path to the new directory. mode)rootprefixrUmktemp)rQrMjoinpathmkdirrr%)r4rPnumberedps r:rXzTempPathFactory.mktempysr44X>  "++H5A GGG "t'7'7'9(QVWA KK! $r<c|j |jS|jK|j}|jr t||j d|j }nt jjd}t|xstjj }txsd}|jd|} |j ddt}|t j t j"vrd nd}|j!| }t!j$|j&rtd |d |j(|k7rtd |d |j&dzdk7rDt j*t j"vrd nd} |j+|j&dz| |j,} |j.dk(rd} t1d|| t2d|j4j6}t9j:|j4j<|j4j7t8j>|j4j<|J|||_|jAd||S#t$r(|jd}|j ddYwxYw)zReturn the base temporary directory, creating it if needed. :returns: The base temporary directory. rSrTPYTEST_DEBUG_TEMPROOTunknownz pytest-of-T)rUexist_okzpytest-of-unknownF)follow_symlinkszThe temporary directory z, is a symbolic link. Fix this and try again.z: is not owned by the current user. Fix this and try again.?rir zpytest-)rWrVkeep lock_timeoutrUregisterz new basetemp)!r&r$existsrrZrNr/environrErtempfile gettempdirget_userrYOSErrorrstatsupports_follow_symlinksS_ISLNKst_modest_uidchmodr(r*rr r3callbackatexitreclose unregisterr%) r4r9from_envtemprootuserrootdiruidstat_follow_symlinks rootdir_statchmod_follow_symlinksrcs r:rMzTempPathFactory.getbasetemps >> %>> !    +++H h NNN &'')Hzz~~&=>HH=(;(;(=>FFHH:*D''*TF(; 0Tr<r"c` ddl}|jS#tttf$rYywxYw)zlReturn the current user name, or None if getuser() does not work in the current environment (see #1010).rN)getpassgetuser ImportErrorrkKeyError)rs r:rjrjs0  ( +s --ct}|j|jtj |d}|j |d|dy)aCreate a TempPathFactory and attach it to the config object. This is to comply with existing plugins which expect the handler to be available at pytest_configure time, but ideally should be moved entirely to the tmp_path_factory session fixture. Tr+_tmp_path_factoryF)raisingN)r add_cleanupundor"rJsetattr)rGmprs r:pytest_configurersI B rww'33Fd3KJJv*,=uJMr<cX|jdddd|jdddd y) Nr>zfHow many sessions should we keep the `tmp_path` directories, according to `tmp_path_retention_policy`.3string)helpdefaulttyper@zvControls which directories created by the `tmp_path` fixture are kept around, based on test outcome. (all/failed/none)r)rrr)addini)parsers r:pytest_addoptionrs@ MM" u   MM#  r<session)scopec.|jjS)zGReturn a :class:`pytest.TempPathFactory` instance for the test session.)rGr)requests r:tmp_path_factoryrs >> + ++r<c|jj}tjdd|}d}|d|}|j |dS)Nz[\W]_T)r[)nodenameresubrX)rfactoryrMAXVALs r:_mk_tmprsF <<  D 66'3 %D F =D >>$> ..r<c#Kt||}||j}|jjt}|dk(r|j ddr t |d|jjt=yw)a;Return a temporary directory (as :class:`pathlib.Path` object) which is unique to each test function invocation. The temporary directory is created as a subdirectory of the base temporary directory, with configurable retention, as discussed in :ref:`temporary directory location and retention`. rcallT ignore_errorsN)rr*rstashtmppath_result_keyrEr)rrr0rI result_dicts r:tmp_pathr"sp 7, -D J / /F,,$$%78K koofd; t4( -.sA:A<c6|jj}|j}|y|j}|dk(r.|dk(r)|j|j r t |d|j r t||jjy)zAfter each session, remove base directory if all the tests passed, the policy is "failed", and the basetemp is not specified by a user. NrrTr) rGrr&r*r$is_dirrr r3rt)r exitstatusrr9rIs r:pytest_sessionfinishr;s)0(H(H))H  / /Fa h   , , 4 ??  84 0h'  &&(r<T)wrappertryfirstc#Kd}|jJi}|j|jjt||j<|Swr.)whenpassedr setdefaultr)itemrrepemptys r:pytest_runtest_makereportrWsK C 88  EADDJJ,e4SXX> JsAA)rz str | None)rGrrr)rrrr)rrrr")rrrr"rr)rrrr"rzGenerator[Path])rzint | ExitCode)rrrz'Generator[None, TestReport, TestReport])>r __future__rrscollections.abcr contextlibr dataclassesr/pathlibrrshutilrrlrhtypingrr r r r rrr_pytest.compatr_pytest.configrrr_pytest.config.argparsingr_pytest.deprecatedr_pytest.fixturesrr_pytest.monkeypatchr _pytest.nodesr_pytest.reportsr _pytest.stashrdictr2r~rr) dataclassr"rjrrrrrrrrr<r:rs?D" %    *!&3&!##,-$++&"/Xd39o.0/0  xxxv  N$ y,, / / //>// /0)8 $& ,'r<