U 42io,@s dZddlmZddlZddlZddlmZddlZddlm Z ddl Z ddl m Z ddl m Z ddl mZdd l mZdd l mZd d lmZd d lmZd dlmZd dlmZd dlmZddlmZddlmZddlmZddlmZddlmZddlmZddl m!Z!ddl m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*e*e e+e,fZ-edZ.eej/GdddZ0d d!d"d#Z1d$d%d&d'd(Z2d)d%d*d+d,Z3e!d-d.d/dd0d1d2Z4d/dd3d4d5d6Z5e!d/dd7d8d9d:Z6d;d<d=d>Z7ed?d?d@dAdBdCdDdEZ8dS)Fz>Support for providing temporary directories to test functions.) annotationsN)Path)rmtree)Any)Dict)final) Generator)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)StashKeyallfailednonec @seZdZUdZded<ded<ded<ded<d ed <d#d d ddd ddddddZed d ddddddZdddddZd$ddddddZ dd d!d"Z d S)%TempPathFactoryzFactory for temporary directories under the common base temp directory, as discussed at :ref:`temporary directory location and retention`. z Path | None_given_basetempr_trace _basetempint_retention_count RetentionType_retention_policyNF _ispytestboolNone)given_basetempretention_countretention_policybasetempr)returncCsJt||dkrd|_nttjt||_||_||_||_ ||_ dSN) rr!rospathabspathstrr"r%r'r#)selfr,r-r.tracer/r)r82/tmp/pip-unpacked-wheel-7n2p7kht/_pytest/tmpdir.py__init__7s zTempPathFactory.__init__r)configr)r0cCsnt|t|d}|dkr.td|d|d}|dkrPtd|d||jj|jd||d 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: ZtmpdirT)r,r7r-r.r))rr$Zgetini ValueErroroptionr/r7get)clsr;r)countpolicyr8r8r9 from_configNs$     zTempPathFactory.from_configr5)basenamer0cCs8tj|}||j|kr4t|d|S)Nz& is not a normalized and relative path)r2r3normpath getbasetempresolveparentr?)r6rFr8r8r9_ensure_relative_to_basetempns z,TempPathFactory._ensure_relative_to_basetempTr)rFnumberedr0cCsL||}|s*||}|jddnt||dd}|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)rootprefixrOmktemp)rKrHjoinpathmkdirr r")r6rFrLpr8r8r9rRts  zTempPathFactory.mktempr0c Csd|jdk r|jS|jdk rF|j}|r0t||jdd|}ntjd}t |p^t }t pnd}| d|}z|jdddWn,tk r| d }|jdddYnXt}|dk r|}|j|krtd |d |jd @d krt||jd@|j}|jdkr*d }td||tdd}|dk sNt|||_|d||S)zReturn the base temporary directory, creating it if needed. :returns: The base temporary directory. NrMrNZPYTEST_DEBUG_TEMPROOTunknownz pytest-of-T)rOexist_okzpytest-of-unknownzThe temporary directory z: is not owned by the current user. Fix this and try again.?rirzpytest-)rQrPkeepZ lock_timeoutrOz new basetemp)r#r!existsrrTrIr2environrArtempfile gettempdirget_userrSOSErrorrstatst_uidst_modechmodr%r'rr AssertionErrorr") r6r/Zfrom_envZtemprootuserrootdiruidZ rootdir_statrZr8r8r9rHsP            zTempPathFactory.getbasetemp)N)T) __name__ __module__ __qualname____doc____annotations__r: classmethodrErKrRrHr8r8r8r9r )s r z str | NonerVc Cs4zddl}|WStttfk r.YdSXdS)zlReturn the current user name, or None if getuser() does not work in the current environment (see #1010).rN)getpassgetuser ImportErrorr`KeyError)ror8r8r9r_s  r_rr+)r;r0cCs6t}||jtj|dd}|j|d|dddS)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)ZraisingN)rZ add_cleanupZundor rEsetattr)r;mprsr8r8r9pytest_configures rvr)parserr0cCs$|jdddd|jdddddS)Nr<zfHow many sessions should we keep the `tmp_path` directories, according to `tmp_path_retention_policy`.)helpdefaultr>zvControls which directories created by the `tmp_path` fixture are kept around, based on test outcome. (all/failed/none)r)Zaddini)rwr8r8r9pytest_addoptionsr{session)Zscoper)requestr0cCs|jjS)zGReturn a :class:`pytest.TempPathFactory` instance for the test session.)r;rs)r}r8r8r9tmp_path_factorysr~r)r}factoryr0cCs4|jj}tdd|}d}|d|}|j|ddS)Nz[\W]_T)rL)nodenameresubrR)r}rrZMAXVALr8r8r9_mk_tmps  rzGenerator[Path])r}r~r0ccsZt||}|V|jjj}|j}|jjt}|dkrL|ddrLt |dd|jjt=dS)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|r;rsr'rstashtmppath_result_keyrAr)r}r~r3rDZ result_dictr8r8r9tmp_paths    rzint | ExitCode) exitstatuscCsb|jj}|j}|dkrdS|j}|dkrN|dkrN|jdkrN|rNt|dd|r^t|dS)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)r;rsr#r'r!is_dirrr )r|rr~r/rDr8r8r9pytest_sessionfinishs rT)wrapperZtryfirstrz'Generator[None, TestReport, TestReport])itemr0ccs2dV}|jdk sti}|j|jt||j<|Sr1)whenreZpassedr setdefaultr)rrrepemptyr8r8r9pytest_runtest_makereport2s r)9rl __future__rZ dataclassesr2pathlibrrshutilrr]typingrrrrr r r r rrZ_pytest.compatrZ_pytest.configrrrZ_pytest.config.argparsingrZ_pytest.deprecatedrZ_pytest.fixturesrrZ_pytest.monkeypatchrZ _pytest.nodesrZ_pytest.reportsrZ _pytest.stashrr5r*rr&Z dataclassr r_rvr{r~rrrrr8r8r8r9s\