<jCdZddlZddlZddlZddlZddlmZddlmZddl m Z ddl Z ddl m Z ddl mZdd l mZe rdd lmZejd ejZejd ejZd ZdZdZGddejZdZdZe jddZGddZGddZ e j!dZ"e j!dZ#dZ$dS)zCoverage plugin for pytest.N)StringIO)Path) TYPE_CHECKING)CovDisabledWarning)CovReportWarning)PytestCovWarning) CovController2unclosed database in z'StoreReport.__call__..Ps+RRQqI((RRRRRRr&)r7all_validate_markdown_dest_files)selfparserr:r option_stringr!files ` r$__call__zStoreReport.__call__Is" T,0 [) 9 9 9dl0=I  - RRRR2QRRR R R M  . .y/CV L L L L L M Mr&cv|d}|d}||kr"d|d}|dz }||dSdS)NrrzEmarkdown and markdown-append options cannot point to the same file: 'z'.zR Please redirect one of them using :DEST (e.g. --cov-report=markdown:dest_file.md))error)r>cov_report_optionsr? markdown_filemarkdown_append_file error_messages r$r=z)StoreReport._validate_markdown_dest_filesSs`*:6 12CD 0 0 0udquuuM q qM LL ' ' ' ' ' 1 0r&r6)__name__ __module__ __qualname__rBr=r&r$r3r3Hs;MMMM(((((r&r3c |dd}|ddgddddd |d d gdd |dtidtd|ddddd|dddd|dddd|d dd!td"#|d$ddd%|d&dd'd(|d)t d'd*+|d,dd-t d.#d'S)/z Add options to control coverage.covz3coverage reporting with distributed testing support--covappendSOURCE?T cov_sourcezPath or package name to measure during execution (multi-allowed). Use --cov= to not do any source filtering and record everything.)actiondefaultmetavarnargsconstdesthelpz --cov-reset store_constz1Reset cov sources accumulated in options so far. )rTrXrYrZz --cov-reportTYPEacType of report to generate: term, term-missing, annotate, html, xml, json, markdown, markdown-append, lcov (multi-allowed). term, term-missing may be followed by ":skip-covered". annotate, html, xml, json, markdown, markdown-append and lcov may be followed by ":DEST" where DEST specifies the output location. Use --cov-report= to not generate any output.)rTrUrVtyperZz --cov-configstorez .coveragercPATHz.Config file for coverage. Default: .coveragerc)rTrUrVrZz--no-cov-on-fail store_trueFz8Do not report coverage if test run fails. Default: False)rTrUrZ--no-covzIDisable coverage report completely (useful for debuggers). Default: Falsez--cov-fail-underMINz,Fail if the total coverage is less than MIN.)rTrVr]rZz --cov-appendzz'_prepare_cov_source..s#+\+\+\T4W[K[K[DK[K[K[r&rL)rSs r$_prepare_cov_sourcerls':%%44+\+\Z+\+\+\\r&T)tryfirstc|j}dx}|_|D]:}t|}|dkrd}|dr |r d|_n;|jjr2t ||j}|j|ddSdS)NFraTrO_cov)known_args_namespaceno_cov_should_warnstr startswithrS CovPlugin pluginmanagerregister) early_configr?argsoptionsno_covrplugins r$pytest_load_initial_conftestsr|s/G*//FW '#hh *  FF ^^G $ $  )-G & E(3<7L$>??"++FF;;;;;<ryrurrqis_distrs r$__init__zCovPlugin.__init__s""**   #'>599KWWjZ_=`=`Kdklsu{~BeCeCGKeK 7He , , !DN F|& )'-hDL # # ( ) )Q . .29P3P3P&(DL #"5dl6M"N"N    'u ' JJv( ) ) ) ) )  ' JJv~ & & & & & ' 'r&Ncontroller_clsr c|Gfdd}|}|j||_jd_t j_jjj}jj !t|dr|j j_ jj t|ddj_ dSdS)NceZdZjZdS)CovPlugin.start..ConfigN)rIrJrKryoption)r>sr$Configrsr&rT fail_under precisionr)ryrrrrcwdrrNconfigcov_fail_underhasattrr cov_precisionr)r>rrnodeidr cov_configs` r$rzCovPlugin.starts > & & & & & & & & & &VXXF,nT\66JJ !!### 8::(,3 < & .7:|3T3T .*4*?DL ' < % -)0[!)L)LDL & & & . -r&c2t|jddduS)N workerinput)rr)r>sessions r$ _is_workerzCovPlugin._is_workersw~}d;;4GGr&c|jjr d|_dSddlm}t j|_||rG|j j d|j }| |j|j |n!|js| |j|jjdkr4|j jt)|jddSdS)zAAt session start determine our implementation and delegate to it.TNrrworkeridr0 _cov_contexts)ryrzrrrosgetpidrrrrgetrr DistWorkerrr cov_contextrurvTestContextPluginr)r>rrrs r$pytest_sessionstartzCovPlugin.pytest_sessionstarts <  !DN F 9;; ??7 # # '^/33JOOF JJv('.& A A A A ' JJv~ & & & < #v - - N ( 1 12CDDW2X2XZi j j j j j . -r&) optionalhookcL|js|j|dSdSzlDelegate to our implementation. Mark this hook as optional in case xdist is not installed. N)rrconfigure_node)r>nodes r$pytest_configure_nodezCovPlugin.pytest_configure_node,s5 ~ 5   . .t 4 4 4 4 4 5 5r&cN|js|j||dSdSr)rr testnodedown)r>rrDs r$pytest_testnodedownzCovPlugin.pytest_testnodedown5s7 ~ :   , ,T5 9 9 9 9 9 : :r&c^|jjp |jj}|o|jo |jj Sr6)ryr7rrno_cov_on_fail)r>neededs r$_should_reportzCovPlugin._should_report>s0(GDL,GKt{Jt|/JKKr&)wrapperc#BK|jrdVStjD]#\}}}}}|tur|tt fvrn$tjddttjD]\}}}}}|turntjdtddl m }tjD]\}}}}}||urntjd|dV}t|j |_ |j|j||s|rnddlm}ddlm}ddlm}  |j|j|_n{#|$rs} d | d }|jjd d |d d d tjt?|dd|_Yd} ~ nd} ~ wwxYw|j Jd|j j!} |j j"} | |j j#rdS| |j| | rpd$||j| | | }|jjd d|d d d |xj dz c_ |S)NrUr oncer)CoverageWarning)CoverageException)display_covered)should_fail_underzFailed to generate report:  terminalreporterz WARNING: Tredboldr stacklevelz$Test coverage should never be `None`zMCoverage failure: total of {total} is less than fail-under={fail_under:.{p}f})totalrpz ERROR: )%rwarningsfiltersResourceWarningCOVERAGE_SQLITE_WARNING_RECOVERAGE_SQLITE_WARNING_RE2filterwarningsr simplefiltercoverage.exceptionsrbool testsfailedrrfinishrr coverage.miscrcoverage.resultsrrsummaryr7rrru getpluginwritewarnrryrr collectonlyformat) r>r_messagecategoryrresultrrrexcrrs r$pytest_runtestloopzCovPlugin.pytest_runtestloopDsY > EEN+3*: v v &Aw!Q?**w;UWr:s/s/s  #I/cet u u u$,$4 < < Aq(Aq+++,  !&*: ; ; ;777777$,$4 ; ; Aq(Aq?**+  !&/ : : :7.//   *   & & ( ( (w'' )D,?,?,A,A ) 8 7 7 7 7 7 8 8 8 8 8 8 : : : : : : #!%!4!--/U---!\8N L6M%)A%  OO )ipp)/$.-HH-#q ,667IJJPPQh]dQhQhQhnry}P~~~##q(## s$E))G!.A)GG!cR|js|ddd|_dSdS)N=ztests coverageT)r write_sep)r>rs r$ write_headingzCovPlugin.write_headings<" '  & &s,< = = ="&D    ' 'r&c|jrd|jjrV||d}|d|dddt jt|ddS|jdS|j dS|j }|r*|||||jj |jj dkrw|||j |jj k}|rdddnd di}d |jj |j |rd nd |rd nd}|j|fi|dSdSdS)Nz&Coverage disabled via --no-cov switch!z WARNING: rTrrrrgreenzU{fail}Required test coverage of {required}% {reached}. Total coverage: {actual:.2f}% zFAIL rz not reachedreached)requiredactualfailr)rryrqrrrrrrrr7getvaluerr)r>rrreportrmarkups r$pytest_terminal_summaryz!CovPlugin.pytest_terminal_summarys > |. I""#3444B &&'>7'>'>'>Dt&TTT 099aHHHH F   & F > ! F))++  +   / 0 0 0  " "6 * * * < & 2t|7RUV7V7V   / 0 0 0^dl&AAF4:OT4000$Fnuu4~ &.WWB)/> Y vG #  "7 5 5f 5 5 5 5 5 3 27V7Vr&) hookwrapperc#K|dsdt|ddvr8|jdV|jdSdVdS)Nno_cover fixturenamesrL)get_closest_markerrrpauseresumer>items r$pytest_runtest_callzCovPlugin.pytest_runtest_callsy  " ": . . *n^`@a@a2a2a   % % ' ' ' EEE   & & ( ( ( ( ( EEEEEr&)TF)NN)rIrJrK__doc__r Namespacerr]rrrpytesthookimplrrrrrrrrLr&r$rtrtso&'&' 2&'&'&'&'TMMD$9MMMM$HHHkkk,V_$'''55('5V_$'''::(':LLL V_T"""::#":x''' 666BV_&&&'&r&rtc8eZdZUded<dZdZdZdZdZdS) rr rc||_dSr6)r)r>rs r$rzTestContextPlugin.__init__s,r&c2||ddS)Nsetupswitch_contextrs r$pytest_runtest_setupz&TestContextPlugin.pytest_runtest_setups D'*****r&c2||ddS)Nteardownrrs r$pytest_runtest_teardownz)TestContextPlugin.pytest_runtest_teardowns D*-----r&c2||ddS)Nrunrrs r$rz%TestContextPlugin.pytest_runtest_calls D%(((((r&ct|jjr+|jj|jd|dSdS)N|)rstartedrNrr)r>rwhens r$rz TestContextPlugin.switch_contextsM   & L   # 2 2dk3J3JD3J3J K K K K K L Lr&N) rIrJrK__annotations__rrr rrrLr&r$rrsr####---+++...)))LLLLLr&rcdS)z%A pytest fixture to disable coverage.NrLrLr&r$rrsr&c|jjdr2|jjd}|jr |jjSdS)zEA pytest fixture to provide access to the underlying coverage object.roN)rru haspluginrrrN)requestr{s r$rNrNsS ~#--f55--77??   -(, , 4r&c2|dddS)Nmarkersz)no_cover: disable coverage for this test.)addinivalue_line)rs r$pytest_configurers I'RSSSSSr&)%rrrreriorpathlibrtypingrrrrrr rr compileIrrr%r.r1Actionr3rgrlrr|rtrfixturerrNrrLr&r$r#sT!!  &%%%%%%(RZ(\^`^bcc(bj)ikmkopp2    ((((((/((((UUUp]]]$ < <  < ________DLLLLLLLL(000TTTTTr&