$TXg7bdZddlmZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZmZmZmZddlZddlZd d lmZd d lmZd d lmZd dlm Z m!Z!m"Z"d dl#m$Z$d dl%m&Z&d dl'm(Z(d dl)m*Z*d dl+m,Z,d dl-m.Z.d dl/m0Z0d dl1m2Z2d dl3m4Z4d dl5m6Z6er"ddl7m8Z8m9Z9ddl:m;Z;ddlmZ>m?Z?m@Z@ddlAmBZBe eCZDejddZFejd !d"ZGejdd#ZHejd !d$ZIejd !d%ZJejd&ZKejejjd'd(jd)* dGd,ZOej dHd-ZPej dId.ZQed/ed0ed1ZR dJd2ZSejd3!dKd4ZTe Gd5d6ZUejdLd7ZVejd3!dMd8ZWe Gd9d:ZXejdNd;ZYe Gd<d=ZZej dOd>Z[ejd3! dPd?Z\e Gd@dAZ]ej dQdBZ^ejd+CdRdDZ_ej dSdEZ`ej dSdFZay)Tz2Collection of pytest fixtures used in conda tests.) annotationsN)contextmanager nullcontext) dataclass) getLogger)Path)copyfile) TYPE_CHECKINGLiteralTypeVaroverload) EntityEncoder)dals)PACKAGE_CACHE_MAGIC_FILE)conda_tests_ctxt_mgmt_def_polcontext reset_context) main_subshell)YamlRawParameter)env_vars)yaml_round_trip_load) path_to_url)PackageCacheData) SubdirData) CondaExitZero)TemporaryDirectory) PackageRecord)IterableIterator) MultiCapture)CaptureFixture ExceptionInfoFixtureRequest MonkeyPatchTempPathFactory) MockerFixtureT)autousec:tjdty)z Suppress `Unclosed Socket Warning` It seems urllib3 keeps a socket open to avoid costly recreation costs. xref: https://github.com/kennethreitz/requests/issues/1882 ignore)categoryN)warningsfilterwarningsResourceWarning6lib/python3.12/site-packages/conda/testing/fixtures.pysuppress_resource_warningr25s H?r0function)scopectt|}|j|jtj j |jS)N)dir)rstr addfinalizercleanuppypathlocalname)tmpdirrequests r1r>r>As9 CK 0F ( 77== %%r0c,tjyN)rclear_cached_local_channel_datar/r0r1clear_subdir_cacherCHs..0r0c#Ktd}tddtjdt |i}t j |dtdyw)z Fixture that will set "context.number_channel_notices" to 0 and then set it back to its original value. This is also a good example of how to override values in the context object. z+ number_channel_notices: 0 r/testdataN)rrrmake_raw_parametersrr _set_raw_data)yaml_strrds r1disable_channel_noticesrJMs_ H "$88 ,X6  B  " "sAAc#(Kdtyw)z:Resets the context object after each test function is run.N)rr/r0r1reset_conda_contextrLgs Oc#K|jd}tdt|it5|dddy#1swYyxYww)zB Used to isolate package or index cache from other tests. pkgsCONDA_PKGS_DIRS)stack_callbackN)mktemprr7r)tmp_path_factorypkgs_dirs r1temp_package_cacherUosE  &&v.H  CM*;X    s.A > A AA CONDA_TEST_SOLVERSzlibmamba,classic,)params monkeypatchc#NKt|||jEd{y7w)at A parameterized fixture that sets the solver backend to (1) libmamba and (2) classic for each test. It's using autouse=True, so only import it in modules that actually need it. Note that skips and xfails need to be done _inside_ the test body. Decorators can't be used because they are evaluated before the fixture has done its work! So, instead of: @pytest.mark.skipif(context.solver == "libmamba", reason="...") def test_foo(): ... Do: def test_foo(): if context.solver == "libmamba": pytest.skip("...") ... N)_solver_helperparamr?rYs r1parametrized_solver_fixturer^{s<g{GMMBBBs %#%c#:Kt||dEd{y7w)Nclassicr[r]s r1solver_classicrbs g{I>>> c#:Kt||dEd{y7w)Nlibmambarar]s r1solver_libmambarfs g{J???rcSolverrer`c#,Ktjjj|j tjjj|j d|t tj|k(sJ|yw)N CONDA_SOLVER)rplugin_managerget_cached_solver_backend cache_clearr8setenvrsolver)r?rYrns r1r[r[sl  44@@B //IIUUV~v.O >>V ## # LsBBsessionc#K|jjjd}|j5|jdddy#1swYyxYww)Ncapturemanager)config pluginmanager getpluginglobal_and_fixture_disabled_global_capturing)r? capmanagers r1session_capsysrxsF--778HIJ  / / 1*** 2 1 1s6AA AAAc`eZdZUded<e ddZed dZdd d dZy) CondaCLIFixturezCaptureFixture | MultiCapturecapsyscyrAr/)selfraisesargvs r1__call__zCondaCLIFixture.__call__s *-r0cyrAr/)r}rs r1rzCondaCLIFixture.__call__sQTr0N)r~cJ|jjttt|}d}|rt j |n t5}t|}ddd|jj\}}t|||rfS|fS#1swY:xYw)aTest conda CLI. Mimic what is done in `conda.cli.main.main`. `conda ...` == `conda_cli(...)` :param argv: Arguments to parse. :param raises: Expected exception to intercept. If provided, the raised exception will be returned instead of exit code (see pytest.raises and pytest.ExceptionInfo). :return: Command results (stdout, stderr, exit code or pytest.ExceptionInfo). N) r{ readouterrtuplemapr7pytestr~rrr)r}r~rcode exceptionouterrs r1rzCondaCLIFixture.__call__s  Sd^$&,V]]6 "+- ?9 $'D@;;))+S Cf66$66@ ?s  BB")rstr | os.PathLike | Pathr~z-type[Exception] | tuple[type[Exception], ...]returnztuple[str, str, ExceptionInfo])rrrztuple[str, str, int])rrr~z4type[Exception] | tuple[type[Exception], ...] | Nonerz$tuple[str, str, int | ExceptionInfo])__name__ __module__ __qualname____annotations__r rr/r0r1rzrzso )) -'->- ( -- TT HL7'7E7 . 7r0rzc#&Kt|yw)zA function scoped fixture returning CondaCLIFixture instance. Use this for any commands that are local to the current test (e.g., creating a conda environment only used in the test). Nrz)r{s r1 conda_clirs & !!c#&Kt|yw)zA session scoped fixture returning CondaCLIFixture instance. Use this for any commands that are global to the test session (e.g., creating a conda environment shared across tests, `conda info`, etc.). Nr)rxs r1session_conda_clirs . ))rc6eZdZUded< d ddZy)PathFactoryFixturertmp_pathNc|xsd}|xstjj}|xsd}|j||z|zz S)aUnique, non-existent path factory. Extends pytest's `tmp_path` fixture with a new unique, non-existent path for usage in cases where we need a temporary path that doesn't exist yet. :param name: Path name to append to `tmp_path` :param prefix: Prefix to prepend to unique name generated :param suffix: Suffix to append to unique name generated :return: A new unique path )uuiduuid4hexr)r}r=prefixsuffixs r1rzPathFactoryFixture.__call__sD 2'tzz|''2}}  677r0)NNN)r= str | Nonerrrrrr)rrrrrr/r0r1rr s=N !! 888 8  8r0rc#&Kt|yw)zA function scoped fixture returning PathFactoryFixture instance. Use this to generate any number of temporary paths for the test that are unique and do not exist yet. N)r)rs r1 path_factoryr$s X &&rcLeZdZUded<ded<d dZedd d dZy) TmpEnvFixturez$PathFactoryFixture | TempPathFactoryrrzrct|jtr|jS|jjdS)Nztmp_env-) isinstancerrrR)r}s r1get_pathzTmpEnvFixture.get_path3s; d''); <$$& &$$++J7 7r0N)rc'Kt|xs|j}|jdd|g|dd|yw)aGenerate a conda environment with the provided packages. :param packages: The packages to install into environment :param prefix: The prefix at which to install the conda environment :return: The conda environment's prefix createz--prefix--yes--quietN)rrr)r}rpackagess r1rzTmpEnvFixture.__call__;sCf/ 0xVShSSS s<>)rr)rr7rzstr | os.PathLike | NonerIterator[Path])rrrrrrrr/r0r1rr.sH668,0)  r0rc#(Kt||yw)zA function scoped fixture returning TmpEnvFixture instance. Use this when creating a conda environment that is local to the current test. Nrrrs r1tmp_envrOs  i 00rMc#(Kt||yw)zA session scoped fixture returning TmpEnvFixture instance. Use this when creating a conda environment that is shared across tests. Nr)rSrs r1session_tmp_envr[s (*; <>) T "8# T "B/G&,,W5 E)6E>:.;/-5MMO,A,A,C,CLE5 (FFu ,C/$U+ 6  / !--djj}.UV / !--djj.OPG'11':; ;; {3w<000sDGG !B G/"GN)rr7rzIterator[tuple[Path, str]])rrrrrrr/r0r1rrgs $$*1*1r0rc#(Kt||yw)z?A function scoped fixture returning TmpChannelFixture instance.N)rrs r1 tmp_channelrs L) 44rM)r=c#6K||jDcgc]+\}}}|tjur|jdr|-c}}}x}rCtj ddj ||jtgyycc}}}ww)z:A monkeypatch fixture that resets context after each test.CONDA_z+monkeypatch cleanup: undo & reset context: z, N) _setitemosenviron startswithlogdebugjoinundor)rYobjr=_ conda_varss r1context_aware_monkeypatchrs  (000LCq "** !: 0z ? *@U?VWXbsB0BABc#:K|dz }|jd|tz j|jd|jt |x}ft j|fk(sJ|tjj|dyw)zHA function scoped fixture returning a temporary package cache directory.rOTrz$conda.base.context.Context.pkgs_dirs new_callable return_valueN) rrtouchpatch PropertyMockr7r pkgs_dirsr_cache_pop)rmockerrT pkgs_dir_strs r1 tmp_pkgs_dirrs ~&H NN4N  ((//1 LL.((&)(m3l5    // / N  t4sBBc#K|dz }|jd|jd|jt|x}ftj |fk(sJ|yw)zFA function scoped fixture returning a temporary environment directory.envsTrz$conda.base.context.Context.envs_dirsrN)rrrr7r envs_dirs)rrenvs_dir envs_dir_strs r1 tmp_envs_dirrsk ~&H NN4N  LL.((&)(m3l5    // / NsA"A$)r?r$rYr%rz(Iterable[Literal['libmamba', 'classic']])r?r$rYr%rzIterable[Literal['classic']])r?r$rYr%rzIterable[Literal['libmamba']])r?r$rYr%rnrgrzIterable[Solver])rzIterator[MultiCapture])r{r"rIterator[CondaCLIFixture])rxr!rr)rrrzIterator[PathFactoryFixture])rrrrzrIterator[TmpEnvFixture])rSr&rrzrr)rrrrzrzIterator[TmpChannelFixture])rYr%rr%)rrrr'rr)b__doc__ __future__rrrrr, contextlibrr dataclassesrloggingrpathlibrshutilr typingr r r r r:r auxlib.entityr auxlib.ishrbase.constantsr base.contextrrrcli.mainrcommon.configurationr common.iorcommon.serializer common.urlrcore.package_cache_datarcore.subdir_datar exceptionsrgateways.disk.creatermodels.recordsrcollections.abcrr _pytest.capturer!r"r#r$r%r& pytest_mockr'rrfixturer2r>rCrJrLrUrgetsplitr^rbrfrgr[rxrzrrrrrrrrrrrrr/r0r1rs99" 2!<< )5PP$3 3$6)&5*2,*@@j!&"& 11j!"2j!" ::>>.0B C I I# NC CC.C C:? ??"??@ @@#@@ 7:. 0B C        i +!+ +7+7 +7\""i *!* 88 82''  @1$1111i =%=&==!= /1/1 /1d5$55!55]# $  5$5.;555($.;r0