a 4j @sdZddlmZddlmZddlZddlmZddlmZddl Z ddl Z ertddl m Z ddl mZdd lmZe d d d d ddZe dd dddZe jeddddd dddZe jeddddd dddZe jdddd d!d"Ze d#d d$d%d&Ze d#d d$d'd(Ze d#d)d d*d+d,Ze d#d-d d.d/d0Ze jddd d1d2d3d4d5Ze jddd d6d7d8d9Ze jddd:d;dd?ZdS)@a xdist hooks. Additionally, pytest-xdist will also decorate a few other hooks with the worker instance that executed the hook originally: ``pytest_runtest_logreport``: ``rep`` parameter has a ``node`` attribute. You can use this hooks just as you would use normal pytest hooks, but some care must be taken in plugins in case ``xdist`` is not installed. Please see: http://pytest.org/en/latest/writing_plugins.html#optionally-using-hooks-from-3rd-party-plugins ) annotations)SequenceN)Any) TYPE_CHECKING)Producer) Scheduling)WorkerControllerz pytest.ConfigzSequence[execnet.XSpec]None)configspecsreturncCsdS)z(Called before any remote node is set up.N)r r r r 4E:\soccer\ml_env\Lib\site-packages\xdist/newhooks.pypytest_xdist_setupnodes srzexecnet.Gateway)gatewayr cCsdS)z#Called on new raw gateway creation.Nr )rr r rpytest_xdist_newgateway'srzCrsync feature is deprecated and will be removed in pytest-xdist 4.0)Z warn_on_implzstr | os.PathLike[str]zSequence[execnet.Gateway])sourcegatewaysr cCsdS)zBCalled before rsyncing a directory to remote gateways takes place.Nr rrr r rpytest_xdist_rsyncstart,s rcCsdS)zACalled after rsyncing a directory to remote gateways takes place.Nr rr r rpytest_xdist_rsyncfinish8s rT)Z firstresultr)r cCsdS)z!Called when creating remote node.Nr r r r rpytest_xdist_getremotemoduleDsrr)noder cCsdS)z7Configure node information before it gets instantiated.Nr rr r rpytest_configure_nodeIsrcCsdS)zTest Node is ready to operate.Nr rr r rpytest_testnodereadyNsrz object | None)rerrorr cCsdS)zTest Node is down.Nr )rrr r rpytest_testnodedownSsrz Sequence[str])ridsr cCsdS)zECalled by the controller node when a worker node finishes collecting.Nr )rrr r r%pytest_xdist_node_collection_finishedXsrrzScheduling | None)r logr cCsdS)z'Return a node scheduler implementation.Nr )r r r r rpytest_xdist_make_scheduler_sr!int)r r cCs tdS)z Return the number of workers to spawn when ``--numprocesses=auto`` is given in the command-line. .. versionadded:: 2.1 N)NotImplementedError)r r r rpytest_xdist_auto_num_workersfsr$strzpytest.TestReportr) crashitemreportschedr cCsdS)a Handle a crashitem, modifying the report if necessary. The scheduler is provided as a parameter to reschedule the test if desired with `sched.mark_test_pending`. def pytest_handlecrashitem(crashitem, report, sched): if should_rerun(crashitem): sched.mark_test_pending(crashitem) report.outcome = "rerun" .. versionadded:: 2.2.1 Nr )r&r'r(r r rpytest_handlecrashitemqsr))__doc__ __future__rcollections.abcrostypingrrZexecnetZpytestZ xdist.remoterZxdist.scheduler.protocolrZxdist.workermanagerZhookspecrrDeprecationWarningrrrrrrrr!r$r)r r r rsV