j@vdZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl Z ddlmZddlmZddlmZdd lmZdd lmZGd d eZGd dZejdZdZdZGddZGddeZGddeZ GddeZ!dS)z8Coverage controllers for use by pytest-cov and nose-cov.N)StringIO)Path) CoverageDatafilename_suffix)CentralCovContextWarning) DistCovError)cleanupceZdZdS)BrokenCovConfigErrorN)__name__ __module__ __qualname__[/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/pytest_cov/engine.pyr r sDrr c$eZdZedZdS) _NullFilecdSNr)vs rwritez_NullFile.writes rN)rrr staticmethodrrrrrrs-  \   rrc#Kt||} t||tj|dVt|||dS#t|||wxYwr)getattrsetattrcopy)objattrbackups r_backupr"!so S$  F#T49V,,--- T6"""""T6""""s 'AA!cFtjfd}|S)Nc tj}n#t$rd}YnwxYwtj|j |g|Ri||tj|SS#|tj|wwxYwr)rcwdOSErroroschdirtopdir)selfargskwargs original_cwdmeths rensure_topdir_wrapperz-_ensure_topdir..ensure_topdir_wrapper,s 8::LL    LLL   '4.t...v..'&&&&(|'&&&&(s && A&&A?) functoolswraps)r.r/s` r_ensure_topdirr2+s9_T ' ' ' ' ' ! rc*tdd|S)NT.r)names r _data_suffixr6>sd## , ,d , ,,rceZdZdZd dZejdZedZ edZ edZ e dZ e d Ze d Zed ZdS) CovControllerz0Base class for different plugin implementations.Nc(||_||_||_||_||_||_||_d|_d|_d|_ t|_ g|_ tjtj|_d|_dS)z8Get some common config used by multiple derived classes.N) cov_source cov_report cov_config cov_append cov_branchconfignodeidcov combining_cov data_fileset node_descsfailed_workersr'fspathrr%r) is_collocated)r*r:r;r<r=r>r?r@s r__init__zCovController.__init__Es$$$$$  !%% i ++ !rc#Ktj}tj|jdVtj|dSr)rr%r'r(r))r*r-s r ensure_topdirzCovController.ensure_topdirWsBxzz    rc`|j|dSr)rAstop unset_envr*s rpausezCovController.pause^s'   rc`|j|dSr)rAstartset_envrOs rresumezCovController.resumecs%  rc:|jtjtjd<n1tj|jtjd<t |j}|r4tj| tjd<ntjtjd<tj |j j jtjd<|jrdtjd<dSdS)zPPut info about coverage into the env so that subprocesses can activate coverage.NCOV_CORE_SOURCECOV_CORE_CONFIGCOV_CORE_DATAFILEenabledCOV_CORE_BRANCH)r:r'pathsepenvironjoinrr<existsrGresolvepathabspathrAr?rCr>)r* config_files rrSzCovController.set_envhs ? ",.JBJ( ) ),.JOODO,L,LBJ( )4?++      7,.Ik6I6I6K6K,L,LBJ( ) ),.JBJ( )*,'//$(/:S*T*T &' ? 6,5BJ( ) ) ) 6 6rcFtjddtjddtjddtjddtjdddS)zRemove coverage info from env.rVNrWrXrZCOV_CORE_CONTEXT)r'r\poprrrrNzCovController.unset_envys} ($/// ($/// *D111 ($/// )400000rcNd|dj|ddS)z"Return a description of this node.zplatform {}, python {}z{}.{}.{}-{}-{}N)format)platform version_infos r get_node_desczCovController.get_node_descs3(..x9P9I9PR^_a`a_aRb9cdddrct|dr|||dStdt|z d}|dz}|dz}||zd|d|||zzd}||dS)NsepD  )hasattrrmmaxlenr)streamstxt sep_totalsep_len sep_extraouts rrmzCovController.seps 65 ! !  JJq#     Vc#hh.33I1nG!A I[FF3FFg .A)BFFFC LL     rcZ d}jsOtjd5jddtcdddS#1swxYwYt jdkr3|dddjnF|dd tjD]}|d |j rX|dd | d j D]$}| |j j d %tfddDrdjvpdd|d}tjt odjv}|d|pditjd5jjd!i|}dddn #1swxYwYdjvrՉjd}tjd5jd|dddn #1swxYwYtjd5jdt}dddn #1swxYwY|r| d|d n| ddjvrjd}tjd5jd|}dddn #1swxYwY| d|jjjn|d djvrjd}tjd5jd|}dddn #1swxYwY| d|jjjn|d djvrjd}tjd5jd|}dddn #1swxYwY| d|jjjn|zdjvrjd}tjd5jd|jdt}dddn #1swxYwY| d |jjjn|d |S)"zProduce coverage reports.Nr?T) show_missing ignore_errorsfiler-z coverage: coveragerpzcoverage: failed workerszlThe following workers failed to return coverage data, ensure that pytest-cov is installed on these workers. rqc3*K|] }|jvVdSr)r;).0xr*s r z(CovController.summary..s*FFqDO#FFFFFFr)term term-missingrz skip-covered skip_coveredannotate)r~ directory)r~rz)Coverage annotated source written to dir z1Coverage annotated source written next to source htmlzCoverage HTML written to dir xml)r~outfilezCoverage XML written to file jsonz!Coverage JSON written to file %s lcovzCoverage LCOV written to file r)r;r"rAreportrrtrErmr]sortedrFrgatewayidany isinstancedictvaluesupdater html_reportr?html_dir xml_report xml_output json_report json_output lcov_report lcov_output) r*rutotal node_descnodeoptionsr annotate_diroutputs ` rsummaryzCovController.summarys ^8,, ^ ^xDS\]] ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ t  1 $ $ HHVS"Irwwt/G/G"I"I J J J J HHVS* - - -#DO44 6 6 n5555   5 HHVS"< = = = LLI J J J+ 5 5  3334444 FFFF-EFFF F F 3!/4?!B Kt!%G &dot<<kSWSbSiSiSkSkAkL NNNL,@DA B B B8,, 3 3'22'22 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3  ( (?:6L8,, N N!! !MMM N N N N N N N N N N N N N N N 8,, L LdKK L L L L L L L L L L L L L L L S YYYYZZZZ QRRR T_ $ $_V,F8,, S S,,46,RR S S S S S S S S S S S S S S S LLsU[Uc9Q9Qiosss t t t DO # #_U+F8,, P P++$+OO P P P P P P P P P P P P P P P LLuW]We9S9Skquuu v v v T_ $ $_V,F8,, Q Q,,4,PP Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q LL=`f`nA\A\tz{ | | | T_ $ $_V,F8,, L L$$4$HHHdKK  L L L L L L L L L L L L L L L LLwY_Yg$(/:U:Umswww x x x s"AAAG66G:=G:,III5"J##J'*J' L33L7:L7N>>OO Q  Q Q *>S44S8;S8)NN)rrr__doc__rI contextlibcontextmanagerrKr2rPrTrSrrNrkrmrrrrr8r8Bs::""""$ ^^66^6 11\1ee\e \TT^TTTrr8c>eZdZdZedZedZdS)Centralz)Implementation for centralised operation.cnttj|j|jt d|j|_|jjj dkr%d}tj t|dtj|j|jt dtj|jjj|j|_|js|j|j|dS) Ncsourcebranch data_suffixrb test_functionzDetected dynamic_context=test_function in coverage configuration. This is unnecessary as this plugin provides the more complete --cov-context option.r) stacklevelccrrrrCrb)r rCoverager:r>r6r<rAr?dynamic_contextwarningswarnr r'r`rarCrBr=eraserRrS)r*messages rrRz Central.starts  $??$S))     8? *o = =f  M27;; J J J J%.??$T**goodho&?@@      HNN     rc||j|j|j|_|j|j|j|tj tj }|j |dS)zSStop coverage, save data to file and set the list of coverage objects to report on.N) rNrArMsaverBloadcombinerksysrirjrEadd)r*rs rfinishzCentral.finish s     %     &&s|S5EFF  I&&&&&rN)rrrrr2rRrrrrrrsN33^< ' '^ ' ' 'rrcJeZdZdZedZdZdZedZdS) DistMasterz&Implementation for distributed master.cttj|j|jt d|j|_|jjj dkrtdd|j_ d|j_ d|j_ tj|j|jt dtj|jjj|j|_|js|j|j|jg|jjjd<dS) NmrrzDetected dynamic_context=test_function in coverage configuration. This is known to cause issues when using xdist, see: https://github.com/pytest-dev/pytest-cov/issues/604 It is recommended to use --cov-context instead.Fmcrr)r rrr:r>r6r<rAr?rr _warn_no_data_warn_unimported_source_warn_preimported_sourcer'r`rarCrBr=rrRr)pathsrOs rrRzDistMaster.start!s! $??$S))     8? *o = =B  "'+0(,1)%.??$T**goodho&?@@      HNN    +/;-h'''rc|jtj|jd|jjDddS)zFWorkers need to know if they are collocated and what files have moved.c,g|]}t|Sr)str)rroots r z-DistMaster.configure_node..Gs*X*X*X3t99*X*X*Xr)cov_master_hostcov_master_topdircov_master_rsync_rootsN) workerinputrsocket gethostnamer) nodemanagerroots)r*rs rconfigure_nodezDistMaster.configure_node@s[ #)#5#7#7%)[*X*XAQAW*X*X*X       rc2t|di}d|vr|j|dSd|vrdtjt jtjdd|dfz}tj |j |j ||j }|tjd krQt!}|t%|d|j|nSt!d d }||d|||||d }|jjjd||j}||j|j} |j !| dS)z#Collect data file name from worker. workeroutputcov_worker_node_idNcov_worker_dataz %s.%s.%06d.%sri?BrrgrTzshould-not-exist)no_disksuffixcov_worker_pathr)"rrFappendrrr'getpidrandomrandintrrr:r>r<rRrjr read_fileobjrdatarloadsget_datarMrrAr?rr_rinforkrirEr) r*rerrorrrrArr`rinfors r testnodedownzDistMaster.testnodedownKs ~r22 v - -   & &t , , , F  & &)"$$ q&))+, -K#4?4?`ky}zIJJJC IIKKK$v--#~~!!(62C+D"E"EFFF%%%%#D9KLLL 6"34555 %%d+++ HHJJJ HHJJJ+,D HO !( + 2 24 8 8 8 ##%%&&u~u7IJJ  I&&&&&rc|j|j|j|_|j|j|jdS)zJCombines coverage data and sets the list of coverage objects to report on.N)rArMrrBrrrOs rrzDistMaster.finishssb    %     rN) rrrrr2rRrrrrrrrrsm0088^8<    &'&'&'P  ^   rrcDeZdZdZedZedZdZdS) DistWorkerz'Implementation for distributed workers.cPttj|jjdko|j|jjdk|_|jsY|jjd|j|jfd|jD|_|j |_tj |j|j td|j|j|_|j|dS)Nrrc<g|]}|Sr)replace)rr master_topdir worker_topdirs rrz$DistWorker.start..s("n"n"nTZ6>>-#O#O"n"n"nrwr)r rrr?rr)rHr:r<rrrr>r6r@rArRrS)r*rrs @@rrRzDistWorker.starts    DK$;B[DK $%9 : : : H      HMMOOO $v--zz ++D111}}x((**0022 K $ + +'+{*.+'+     rcdS)z&Only the master reports so do nothing.Nr)r*rus rrzDistWorker.summarysrN)rrrrr2rRrrrrrrrs^11^6##^#J55555rr)"rrrr0r'rrrriorpathlibrr coverage.datarcoverage.sqldatarrr r embedr Exceptionr rrr"r2r6r8rrrrrrrs<>>  &&&&&&,,,,,,&&&&&&     9            ###!!!&---ffffffffR0'0'0'0'0'm0'0'0'f________DF5F5F5F5F5F5F5F5F5F5r