B |?XcU(@sddlZddlZddlZddlZddlZeedo:ejdkZeej Z ej ddZ ddZ ej dd Zej d d Zej d d ddZej d dddZedddZedddZeZddZejddddZejddddZdS) N version_tuple)rc s*tjd}|r8yt|Stk r6tdYnXy ddl}Wntk rXYn*X|j j dk}|j |dpx| }|r|Syddlm fdd}WnZtk rtjd d krd Sydd lm }Wn tk rdd l m }YnXYnXy |}Wntk rd SX|r&|Sd S)NZPYTEST_XDIST_AUTO_NUM_WORKERSzHPYTEST_XDIST_AUTO_NUM_WORKERS is not a number: {env_var!r}. Ignoring it.rlogical)r)sched_getaffinitycs tdS)Nr)len)rr+lib/python3.7/site-packages/xdist/plugin.py cpu_count%sz0pytest_xdist_auto_num_workers..cpu_countZTRAVIStrue)r )osenvirongetint ValueErrorwarningswarnpsutil ImportErroroption numprocessesr rZmultiprocessingNotImplementedError)configZenv_varrZ use_logicalcountr nr)rrpytest_xdist_auto_num_workerss<      rcCs |dkr |S|dk rt|SdS)N)autor)r)srrrparse_numprocesses7src Cs"|dd}|jdddddtdd|jd d d dtd d|jd ddd dd|jdddddddddgdddd|jdddgddd |jd!d"d#d$d%d&|jd'dgd(d)d*|jd+dgd,d-d*|jd.dd/d0|jd1d2trd3nd4d5|jd6d7trd3nd4d5|jd8trd3nd4d9d:dS);Nxdistz"distributed and subprocess testingz-nz--numprocessesrstorezShortcut for '--dist=load --tx=NUM*popen'. With 'auto', attempt to detect physical CPU count. With 'logical', detect logical CPU count. If physical CPU count cannot be found, falls back to logical count. This will be 0 when used with --pdb.)destmetavaractiontypehelpz--maxprocesses maxprocesseszWlimit the maximum number of workers to process the tests when using --numprocesses=autoz--max-worker-restartZmaxworkerrestartzbmaximum number of workers that can be restarted when crashed (set to zero to disable this feature))r$defaultr"r&z--distZdistmodeZeachloadZ loadscopeZloadfileZ loadgroupnodistaset mode for distributing tests to exec environments. each: send each test to all available environments. load: load balance by sending any pending test to any available environment. loadscope: load balance by sending pending groups of tests in the same scope to any available environment. loadfile: load balance by sending test grouped by file to any available environment. loadgroup: like load, but sends tests marked with 'xdist_group' to the same worker. (default) no: run tests inprocess, don't distribute.)r#r$choicesr"r(r&z--txtxappendZxspeczadd a test execution environment. some examples: --tx popen//python=python2.5 --tx socket=192.168.1.102:8888 --tx ssh=user@codespeak.net//chdir=testcache)r"r$r(r#r&z-d store_truedistloadFz/load-balance tests. shortcut for '--dist=load')r$r"r(r&z --rsyncdirZDIRz.add directory for rsyncing to remote tx nodes.)r$r(r#r&z --rsyncignoreZGLOBzs   r7cCsddlm}||dS)Nr)newhooks)r r8Z add_hookspecs) pluginmanagerr8rrrpytest_addhookss r:T)ZtrylastcCs|ddkrP|dsPddlm}||}|j|d|jd}|rPd|_|d dsf|d rzt d }| |d |d ds|drt d}| |d d}| d|dS)Nr+r* collectonlyr)DSessionZdsessionZterminalreporterFZ looponfailr3zThe --looponfail command line argument and looponfailroots config variable are deprecated. The loop-on-fail feature will be removed in pytest-xdist 4.0.r Zrsyncdirr1zThe --rsyncdir command line argument and rsyncdirs config variable are deprecated. The rsync feature will be removed in pytest-xdist 4.0.zuxdist_group: specify group for tests should run in same session.in relation to one another. Provided by pytest-xdist.Zmarkers) getoptiongetvalueZxdist.dsessionr<r9registerZ getpluginZ showfspathZgetiniDeprecationWarningZissue_config_time_warningZaddinivalue_line)rr<sessiontrZwarningZ config_linerrrpytest_configures"    rC)ZtryfirstcCs|dd}|jjdkrD|r.d|j_d|j_n|jj|d}||j_|jjr|jjdkr`d|j_|jj}|jjr~t||jj}dg||j_|jj rd|j_|j }|d s|d dkr|rt d dS) NusepdbF)rrrr*)rr)popenr;r+zG--pdb is incompatible with distributing tests; try using -n0 or -nauto.) r=rrr+hookrr'minr-r0r>pytestZ UsageError)rrDZ auto_num_cpusrvalrrrpytest_cmdline_mains(    rJ)returncCs t|jdS)zReturn `True` if this is an xdist worker, `False` otherwise :param request_or_session: the `pytest` `request` or `session` object workerinput)hasattrr)request_or_sessionrrris_xdist_workersrOcCst| o|jjjdkS)zReturn `True` if this is the xdist controller, `False` otherwise Note: this method also returns `False` when distribution has not been activated at all. :param request_or_session: the `pytest` `request` or `session` object r*)rOrrr+)rNrrris_xdist_controllers rPcCs t|jdr|jjdSdSdS)a*Return the id of the current worker ('gw0', 'gw1', etc) or 'master' if running on the controller node. If not distributing tests (for example passing `-n0` or not passing `-n` at all) also return 'master'. :param request_or_session: the `pytest` `request` or `session` object rLZworkeridZmasterN)rMrrL)rNrrrget_xdist_worker_ids  rQrA)ZscopecCst|S)zkReturn the id of the current worker ('gw0', 'gw1', etc) or 'master' if running on the master node. )rQ)requestrrr worker_id)srScCs&t|jdr|jjdStjSdS)z)Return the unique id of the current test.rLZ testrunuidN)rMrrLuuidZuuid4hex)rRrrr testrun_uid2s  rV)r rTsysrrHrMrr4listpathZ _sys_pathZhookimplrrr7r:rCrJboolrOrPZis_xdist_masterrQZfixturerSrVrrrrs$ )u