6 fuUdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZddlmZmZmZmZmZmZmZddlmZddlZdd lmZdd lmZm Z m!Z!ej"#eZ$d d gZ%d e&d<GddeZ'GddeZ(ee'e(fZ)de&d<dRdZ*dSdZ+Gddej,Z-dTdZ.e Gd d!Z/dUd%Z0dVd(Z1dVd)Z2dWd+Z3dXd/Z4dYd1Z5dYd2Z6dZd5Z7d[d8Z8d\d=Z9ej:d>Z;d]dEZ<GdFdGej=Z>GdHdIeZ?GdJd@ej@ZAd^dLZBd_dOZCd_dPZDGdQdBZEdS)`zGUtilities for processing .test files containing test case descriptions.) annotationsN)abstractmethod) dataclass)Path)AnyFinalIterator NamedTupleNoReturnPatternUnion) TypeAlias)defaults)PREFIXtest_data_prefix test_temp_dirgdblldbrSUPPORTED_DEBUGGERSc.eZdZUded<ded<ded<dS) UpdateFilestrmodulecontent target_pathN__name__ __module__ __qualname____annotations__D/var/www/equiseq/venv/lib/python3.11/site-packages/mypy/test/data.pyrr s0KKKLLLr"rc$eZdZUded<ded<dS) DeleteFilerrpathNrr!r"r#r%r%'s"KKK IIIIIr"r% _TypeAlias FileOperationfilenamerreturnctj|\}}|d}|ddkr|d|S)N/__init__.)osr&splitextsplitpopjoin)r)_partss r#_file_arg_to_moduler7/sX'""8,,KHa NN3  E RyJ 88E??r"caseDataDrivenTestCaseNonecV+tjj}jj}jjrt jj}n tj}jj }g}g}g}i}i} i} i} g} i} g}dIfd }|dx+}| d|dd D]+dI+fd }+j d vr +j Jtd +j}||+j }+j d kr'| t+j +j dvr| ||f+j dkrG| |t!j|t jf+j dkr| ||f+j dkr+j J|t jj+j }t-|d5}| ||d|fd d d n #1swxYwYÉ+j dkr+j J|t jj+j }t-|d5}| ||d|fd d d n #1swxYwYh+j dkr+j J|t jj+j }t-|d5}| ||d|fd d d n #1swxYwY t!jd+j r~+j dkrdn(t3+j t5dd }|dksJ+j t7n#d+j dD}|| |<t!jd+j r~+j dkrdn(t3+j t5dd }|dksJ+j t7n#d+j dD}|| |<=t!jd +j rr+j d!krdn(t3+j t5d!d }|dksJ+j gn#d"+j dD}|| |<ɉ+j d#krĉ+j Jt!jd$+j }||d%+j t3|d&}|d&kr|d'||||d}| |t7|t!jd(+j r+j g}n+j d}d)}|D]H}| d*r/|d+d,} | d-vr |d.|d,d }! tCd/|!d0D}"n#tD$r||!d1YnwxYw| d2kr;|"tFj$kr||d3tFj$tJj&|"k}| d4kry|"tFj$kr||d5tFj$dt5|"cxkrd6ksn|d7|!tJj&d t5|"|"k}J|rd8+jD}#t jj'd9krj(r d:|#D}#+j d;ks +j dkr+j +j} Ή+j +j r d?+j nd@z}$|dA|$dBj|r|dCj| )D]I}|| vr | || |<|| vr4|| vr0| |*| |s|dD|dEJt5|}%|j}&tW|&|dF|D])\}'}tW|d ||'*t7}(|D]-\})}*|)|(vr|dG|)dH|(|).|&_,|_-|%_.|_/j0+j0zt5+jzd&z _1|_2|_3| _4| _5| _6| pg_7| _8|_9d S)JzParse and prepare a single case from suite with test case descriptions. This method is part of the setup phase, just before the test case is run. msgrr*r cXtjjdjd|ddS)N:: Fpytrace)pytestfailfileline)r<r8s r# _case_failz#parse_test_case.._case_failPs8 ty55495555uEEEEEEr"r__main__Ncrjjzdz }tjjd|d|ddS)Nr>r?Fr@)rErBrCrD)r< item_abs_liner8items r# _item_failz#parse_test_case.._item_failYsI I 1A5M K49==}====u M M M M M Mr">rDfixtureoutfile outfile-re rN>rDrNrPrObuiltinsutf8encodingz builtins.pyitypingz typing.pyi _typeshedz _typeshed.pyiz stale[0-9]*$stalec6h|]}|Sr!strip.0ts r# z"parse_test_case.. 5]5]5]Aaggii5]5]5]r",zrechecked[0-9]*$ recheckedc6h|]}|Sr!rZr\s r#r_z"parse_test_case..r`r"ztargets[0-9]*$targetsc6g|]}|Sr!rZr\s r# z#parse_test_case..s 6^6^6^Qqwwyy6^6^6^r"deletez(.*)\.([0-9]+)$zInvalid delete section rJzCan't delete during step z out[0-9]*$Tversion >==>=z5Only >= and == version checks are currently supportedc34K|]}t|VdSN)int)r]xs r# z"parse_test_case..s('O'O1A'O'O'O'O'O'Or"r/z is not a valid python versionrlz. always true since minimum runtime version is rkz/ always false since minimum runtime version is zFOnly minor or patch version checks are currently supported with "==": c,g|]}t|Sr!)expand_variablesr]rEs r#rfz#parse_test_case..s!KKK.t44KKKr"\c,g|]}t|Sr!) fix_win_pathrus r#rfz#parse_test_case..s !L!L!L,t"4"4!L!L!Lr"outout1F triggered zInvalid section header [z ] in case z*Required output section not found in case zStale modules after pass z& must be a subset of rechecked modulesmainzDuplicated filename z$. Did you include it multiple times?)r<rr*r ):parse_test_datadatanamesuite base_path native_sepr0r&r4 posixpathrequired_out_sectionappendidargrtr7recompilerstripSdirnamerDopenreadmatchrolensetr2group setdefaultadd startswithtuple ValueErrorrPYTHON3_VERSIONsys version_infosepnormalize_outputkeysissubset expand_errorsinputoutputoutput_inline_startoutput2rE last_linefiles output_filesexpected_stale_modulesexpected_rechecked_modules deleted_pathsr{expected_fine_grained_targets test_modules),r8 test_itemsrr4out_section_missingrrrrr stale_modulesrechecked_modulesr{rdrrF first_itemrMcontentsr&mpathfsrc_pathpassnummodules reprocessedmnumfullargs version_checkr compare_op version_strrh tmp_output section_strrr file_path seen_filesrDr5rLs,` @r#parse_test_caser7s2 !DI66J $I zw|~*9#%E9;LF$&G)+M)+M-/I$&G LFFFFFF #1 %D: ###122oXoX N N N N N N N 7B B B8'''' $)(<(<==H4 48,,Dw)####$7$A$ABBBw--- dH-....L((##T2:hoo6G6G+N+N$OPPPPI%%##T8$4555 W " "8'''D33TX>>Eef--- J dd9n==qvvxxHIII J J J J J J J J J J J J J J J W 8'''tBGOODI66AAHh000 HA dd9l;;QVVXXFGGG H H H H H H H H H H H H H H H W # #8'''tBGOODI66AAHh000 KA dd9o>>IJJJ K K K K K K K K K K K K K K K Xotw / /J X7g--aa3tws7||~~7N3O3OGQ;;;;#x/ceee5]5]X[I\I\5]5]5]G%,M' " " X)47 3 3E X7k11aas473{CSCSCUCU;V7W7WGQ;;;;#x/ceee5]5]X[I\I\5]5]5]G)0 g & & X' 1 1@ X7i//aaSYAQAQ9R5S5SGQ;;;; $ 0""6^6^$(..Y\J]J]6^6^6^K*GG   W 8'''+TX66Ay ATXAABBBaggajj//CQww >),,T!$QqSJ!55" #Z[[["%abb'KU"''O'O 8I8I#8N8N'O'O'O"O"O%UUU" k#S#S#STTTTTU!T))"h&>>>&J#& p pV^Vn p p),(8G(C #t++"X%===&J#& q qW_Wo q q !3w<<3333!3333&J xit x x),(83w<<(HG(S  ,KKKKK 7;$&&4+@&!L!L!L!L!LJ7e##tw&'8'8'FF!$'#e**,,"788G"Q;;;;'1GG$&+# W # #(8 II'tx%G^^^^RHK JV+VVVV W W W WO M MMNNN %%'' d d + + +*7w)? g & } $ $,,,!'*334Eg4NOO- Jb7bbb c c cf++ OE%((($?? 8hnnT**FI>>>>Ja :   JXdXXX Y Y YtDJDK2DDLY*S^^;a?DNDJ$D"/D&7D#&D_"DN)0D&$DsH4II I =4K==L L "4N""N& )N& :,[''\\ceZdZUdZded<ded<ded<ded<d ed <d Zd Zd ed<ded<ded<ded<d ed<ded<ded<ded<ded<ded<d8fd* Zd9d+Zd9d,Z d9d-Z d:d/Z d;dr9zNHolds parsed data-driven test cases, and handles directory setup and teardown.DataSuiteCollectorparent list[str]rrrorzdict[int, list[str]]rr}rzlist[tuple[str, str]]rrzdict[int, set[str]]rrrboolrrz$list[tuple[str, str | Pattern[str]]]rrr{r DataSuiterDrr writescache only_whenplatform str | NoneskipxfailrrEr*r:c Bt||||_||_||_||_||_|dkrtjdks|dkrtjdkrd} | |_ | |_ | |_ | |_ d|_ d|_dS)Nwindowswin32posixT)superr.rrDrrrrrrrrrEold_cwdtmpdir)selfrrrDrrrrrrrrrE __class__s r#r.zDataDrivenTestCase.__init__s v&&&  &" 0  ! !clg&=&=   CLG$;$;D    #' ?C r"c^|jrtjn+|jr$|tjj|t }| Jd|}| | |dS#t$r|j dd}|r|j Jtj|tj|j j}t'd|jd|tj|s.|jsJtj|j|}t-j|j j|wxYw)NzShould not happen--save-failures-tozCopying data from test z to )rrBr add_markermark getparentrobjsetuprun_case Exceptionconfig getoptionrr0r&r4basenamerprintisabsrshutilcopytree)rrrsave_dir target_dirs r#runtestzDataDrivenTestCase.runtest5s 9 / KMMMM Z / OOFK- . . . 233!!#6!!!    NN4    #';#8#89Mt#T#TH >{...W\\(BG4D4DT[EU4V4VWW K KKzKKLLLw}}Z00H<''.ps'KKKSeiiR((KKKr") rr0getcwdrtempfileTemporaryDirectoryrchdirrmkdirrrrrrorsubmodule_from_pathrrrr&rmakedirsrwriteritemsr%maxranger) rr&rrrrr operationdirrpathsmax_steprs @r#rzDataDrivenTestCase.setupNs|T""""y{{ 1FFF  !""" 13!Z % %MD')400A %!''!**ooaxxxx f\2t<< )+66&vw DD   b))00;;;;good++ C$////$f555%GGG$$$%%%%%%%%%%%%%%%,2244 K KJC!8888 K K)$//  b))00FD1I1IJJJJ K"'-3u:::AKKKKE!X\4J4JKKK s2FF F c|jtj|j|j+ |jn#t $rYnwxYwd|_d|_dSrn)rr0rrcleanupOSErrorrs r#teardownzDataDrivenTestCase.teardownrsr < # HT\ " " " ; "  ##%%%%      sA AAtuple[str, int, str]c*|j|j|jfSrn)rDrErrs r# reportinfozDataDrivenTestCase.reportinfo}sy$)TY..r"Nexcinfo#pytest.ExceptionInfo[BaseException]style Any | Nonec~t|jtr|}nzt|jtjjr!|jjs|}n5|j ||_ | d}d|j d|j d|S)Nshort)rzdata: r>z: ) isinstancevalue SystemExitexconlyrBrCrrAr_traceback_filter tracebackgetreprrDrE)rr rexcreprs r# repr_failurezDataDrivenTestCase.repr_failures gmZ 0 0 5 oo''GG  v{'< = = 5gmF[ 5oo''GG $ = =g F FG ooGo44G; ;;DI;;';;;r"list[list[FileOperation]]c|jS)axReturn a list of descriptions of file operations for each incremental step. The first list item corresponds to the first incremental step, the second for the second step, etc. Each operation can either be a file modification/creation (UpdateFile) or deletion (DeleteFile). Defaults to having two steps if there aern't any operations. )rrs r# find_stepszDataDrivenTestCase.find_stepss zr")rrrrrDrrrrrrrrrrrrrrrrrrEror*r:r*r:)r*r rn)r rrrr*r)r*r)rrr__doc__r rDrEr.rrr r rr __classcell__rs@r#r9r9sXX!!!! D D!   ////33337777NNN6666&&&&DDDDDDB2"L"L"L"LH    ////QU<<<<<$        r"r&ctjdd|}tjd|}|d=d|}tjdd|}|S)Nz\.pyi?$r}z[/\\]rr/z \.__init__$)rrr2r4)r&r6rs r#rrsT 6*b$ ' 'D HXt $ $E a XXe__F VNB / /F Mr"c^eZdZUdZded<ded<ded<ded <ded <edd Zd S)TestItemzVParsed test caseitem. An item is of the form [id arg] .. data .. rrrrrrrorEend_liner*cJ|j|jz t|jz Srn)r&rErrrs r#trimmed_newlineszTestItem.trimmed_newliness}ty(3ty>>99r"N)r*ro)rrrr r propertyr(r!r"r#r%r%sm GGGOOOOOO IIIMMM :::X:::r"r%raw_datarlist[TestItem]c dd|zdzg|dz}g}g}d}d}d}d}|t|kr||} ||dr| dr|rGt |}t |}|t||||dz||}| dd }d}d |vr=|| d dzd}|d| d }g}n||d r$|||ddnu||d s|||n>||d r#|||dd|dz }|t|k|rJt |}t |}|t||||dz|dz |S)z>Parse a list of lines that represent a sequence of test items.r}z[case ]rQNr[rHr-r|z\[z--z----rJ) r2rr[rendswithcollapse_line_continuation strip_listrr%index) r*rlinesretrrrii0ss r#rrsLD3& '(..*>*> >ECDBC A B c%jj.. !HNN   8  s # # & 3 & ?1$77!$'' 8BT261==>>>B1R4BCbyy#*,,-"((3--(DD 1X  ' ' & KKa % % % %q$$T** & KKa ! ! ! ! 1X  ( ( & KKa % % % Q- c%jj..2 ;)$//$ 8BT261q599::: Jr"lrcg}|D]+}|tjdd|,|r.|ddkr"||r |ddk"|S)zReturn a stripped copy of l. Strip whitespace at the end of all lines, and strip all empty lines from the end of the array. z\s+$r}r-)rrrr3)r8rr7s r#r1r1syA )) Q''(((( "   "  Hr"c g}d}|D]i}tjdd|}|r%|dxxtjdd|z cc<n|||d}j|S)NFz\\$r}r-z^ +rv)rrrr/)r8r:contr7sss r#r0r0sA D    VFB " "   bEEERVE2r** *EEEE HHRLLLzz$ Hr"r7c8|dtS)Nz)replaceroot_dir)r7s r#rtrt s 99Xx ( ((r"rrfnamc tt|D]3}||dddD] }tjd|}|r|ddkrd}n7|ddkrd}n|dd krd }|d }|d }|d d}|$||d|dzd|d|||d|dzd|d|d|  5dS)zTransform comments such as '# E: message' or '# E:3: message' in input. The result is lines like 'fnam:line: error: message'. z # rHNz*^([ENW]):((?P\d+):)? (?P.*)$EerrorNnoteWwarningcolmessagez\##r>r?) rrr2rsearchr[rr?r) rrrAr5possible_err_commentrseverityrIrJs r#rrs3u::  RR$)!HNN5$9$9!""$= R R  =?S?Y?Y?[?[A R771::$$&HHWWQZZ3&&%HHWWQZZ3&&(Hggenn''),,!//%55;MMT"J"JAE"J"JX"J"J"J"JKKKKMMT"P"PAE"P"PC"P"P8"P"Pw"P"PQQQ# RRRr"rEc&|ttdd}tjd|}|s|S|\}}}d|dd|pd|S)z`Changes Windows paths to Linux paths in error messages. E.g. foo\bar.py -> foo/bar.py. rvr,z^([\S/]+):(\d+:)?(\s+.*)z{}:{}{}r})r?r@rrgroupsformat)rErr)linenorJs r#rxrx,s <<("2"24"="= > >D ,d33A T $%HHJJ!&' 0 0s ; ;V\r7SSSr"ctjd|}|s|Sd|d|d|ddd||ddS)zChanges filename paths to Linux paths in Cobertura output files. E.g. filename="pkg\subpkg\a.py" -> filename="pkg/subpkg/a.py". z&NXY-=-Z-Z]^-^-^ R     -^-^r" collectorrobjectrct|tr:t|tr%|turt||SdS)zCalled by pytest on each object in modules configured in conftest.py files. collector is pytest.Collector, returns Optional[pytest.Class] rrN)rtype issubclassrr from_parent)rjrrs r#pytest_pycollect_makeitemrqxs] #t c9 % % #Y*>*>&11 t2  4r"z(?P[a-zA-Z_0-9]+)(?P-writescache)?(?P-only_when_cache|-only_when_nocache)?(?P-skip_path_normalization)?(-(?Pposix|windows))?(?P-skip)?(?P-xfail)?rDataFileCollectorrrrDIterator[DataDrivenTestCase]c#|Kt|d5}|}dddn #1swxYwYtjd|tjtjz}t |}t|ddz}t}|D]} t|}t | } | std| | d } | |vr)td | |j|t |||t%| |jt)| d | d | d t)| dt)| d| d || V||ddzz }|| hdS)zIterate over raw test cases in file, at collection time, ignoring sub items. The collection phase is slow, so any heavy processing should be deferred to after uninteresting tests are filtered (when using -k PATTERN switch). zutf-8rTNz^\[case ([^]+)]+)\][ \t]*$\n)flagsrQrHzInvalid testcase id rz1Found a duplicate test name "{}" in {} on line {}rrrrrskip_path_normalization) rrrDrrrrrrrrrE)rrrr2DOTALL MULTILINEiternextr_r_case_name_pattern fullmatch RuntimeErrorrrQrr9rpadd_test_name_suffixtest_name_suffixrupdate) rrrDrrcases cases_iterline_no test_namescase_idrrs r#split_test_casesrsc dW % % %vvxx H4d")blBZ [ [ [EeJ:$$T**Q.GJ""J  ( ( 1 1 CAgAABB Bwwv :  CJJ&+w  !,,%dE,BCCQWW]3344ggk**WWZ((aggfoo&&qwww''((!")B!C!CC-    4::d##a'' 4&!!!!=""s 599ceZdZddZdS)rr*Iterator[DataFileCollector]c#K|j}tj|jsJd|jd|jD] }t ||V!dS)zNCalled by pytest on each of the object returned from pytest_pycollect_makeitemzTest data prefix (z) not set correctlyrmN)rr0r&isdir data_prefixrrrrp)rr data_files r#collectzDataSuiteCollector.collects 8w}}     G G F 1 F F F G G  M MI#//t)/LL L L L L M Mr"N)r*r)rrrrr!r"r#rrs. M M M M M Mr"rc.eZdZUded<ded<ded<dS) DataFileFixrorR end_linenorr3Nrr!r"r#rrs0KKKOOOr"rczeZdZUdZded<ded<edfd Zdd Zdfd Zdfd Z ddZ ddZ xZ S)rrzvRepresents a single `.test` data driven test file. More context: https://github.com/python/mypy/issues/11662 rrzlist[DataFileFix]_fixesrrr*c|t||}t|tsJ|S)N)r)rrprrr)clsrrrjrs r#rpzDataFileCollector.from_parents>GG''T':: )%677777r"rsc#Kt||jjtj|jjj|jEd{VdS)N)rrrD)rrrr0r&r4rrrs r#rzDataFileCollector.collectsk#+/dko949EE            r"r:cVtg|_dSrn)rrrrrs r#rzDataFileCollector.setups   r"cpt|dSrn)rr  _apply_fixesrs r#r zDataFileCollector.teardowns1  r"fixrc:|j|dSrn)rr)rrs r# enqueue_fixzDataFileCollector.enqueue_fixs 3r"cl|jsdSt|jjj|jz }|d}t|jdD]}|j ||j dz |j dz <| d |dS)NrQT)reverserH)rrrrrr read_textr2sortedr3rRr write_textr4)r data_pathr3rs r#rzDataFileCollector._apply_fixess{  F455 A ##%%++D11$+t444 C CC9<E#*q.3>A#55 6 6TYYu--.....r")rrrrr*rr)r*rsr)rrr*r:) rrrr r classmethodrprrr rrr!r"s@r#rrrrs [        ////////r"suffixctjd|}|r3|d}|dt| |z|zS||zS)Nz-[-A-Za-z0-9]+$r)rrLrr)rrr magic_suffixs r#r~r~s] $d++Awwqzz (s<(((()F2\AAf}r"testcasercJd|jvpd|jvS)N incremental)rlowerrD)rs r#is_incrementalrs' HM//11 1 S]hm5SSr"ctd|jDrdS|jD]:\}}tj|drdS;dS)Nc3@K|]}tjd|VdS)z # flags[2-9]:N)rrrus r#rqz#has_stable_flags..s/ G G28$d + + G G G G G Gr"Fz mypy.ini.T)anyrrr0r&rr)rr)rs r#has_stable_flagsrsu G G G G GGGu&n( 7  H % % 0 0 = = 55  4r"cNeZdZUded<eZeZdZdZ dZ d dZ e d d Z d S)rrrFr}r*r:cdS)zSetup fixtures (ad-hoc)Nr!rs r#rzDataSuite.setup2sr"rr9ctrn)NotImplementedError)rrs r#rzDataSuite.run_case5s!!r"Nr)rr9r*r:)rrrr rrrrrrrrrrr!r"r#rr!suI#K J&&&&"""^"""r")r)rr*r)r8r9r*r:)r&rr*r)r*rrrr*r+)r8rr*r)r7rr*r)rrrrrArr*r:)rErr*r)rWrr*r:)rrfr*r:)rjrrrrrkr*r)rrrrrrDrr*rs)rrrrr*r)rr9r*r)Fr  __future__rr0os.pathrrrrrabcr dataclassesrpathlibrrVrrr r r r r typing_extensionsrr'rBrYrmypy.test.configrrrr&normpathr@rr rr%r(r7rItemr9rr%rr1r0rtrrxrVrerirqrr{rClassrr Collectorrrr~rrrr!r"r#rs`MMM""""""  !!!!!!MMMMMMMMMMMMMMMMMM555555 DDDDDDDDDD 7  F # # $V_,,,, "*j"89 9999y%y%y%y%xhhhhhhhhV ::::::: :,****Z    $     ))))RRRR: T T T T    ,>    " RZ,",",","^ M M M M M M M M* ,/,/,/,/,/(,/,/,/^    TTTT""""""""""r"