B '(bZi@s`UdZddlZddlZddlZddlZddlZddlZddlmZddl Z ddl Z ddl m Z m Z mZmZmZmZmZmZmZmZddlmZddlmZmZmZejeZddgZee d <Gd d d eZ!Gd d d eZ"ee!e"fZ#dddddZ$Gddde j%Z&e'e'dddZ(GdddZ)e'e'e e)dddZ*e e'e e'dddZ+e e'e e'dddZ,e'e'd d!d"Z-e e'e e'e'dd#d$d%Z.e'e'd&d'd(Z/e'e'd&d)d*Z0edd+d,d-Z1ee'e2d.d/d0d1Z3d2d3e'edd4d5d6Z4Gd7d8d8e j5Z6Gd9d2d2e j7Z8e'e'e'd:d;d<Z9e&e:d=d>d?Z;e&e:d=d@dAZ.,zrechecked[0-9]*$Z recheckedcSsh|] }|qSr)r,)r-r.rrrr/gsztargets[0-9]*$targetscSsg|] }|qSr)r,)r-r.rrr lsz#parse_test_case..deletez(.*)\.([0-9]+)$zInvalid delete section: zCan't delete during step z out[0-9]*$zskip-path-normalizationFversion >==>=zB{}, line {}: Only >= and == version checks are currently supportedcss|]}t|VqdS)N)int)r-xrrr sz"parse_test_case...z/{}, line {}: "{}" is not a valid python versionz>=z==zW{}, line {}: Only minor or patch version checks are currently supported with "==": "{}"cSsg|] }t|qSr)expand_variables)r-linerrrr2s\cSsg|] }t|qSr) fix_win_path)r-r@rrrr2soutZout1 triggeredzInvalid section header z in z at line z, line z#: Required output section not foundzIStale modules after pass {} must be a subset of rechecked modules ({}:{})main)7parse_test_datadatanamesuite base_path native_seposrjoin posixpathrequired_out_sectionidargAssertionErrorr?appendrecompilerstripSdirnamer%openreadmatchr:lensetsplitgroup setdefaultadd startswith ValueErrorformatr@tuplesys version_infosepkeysissubset expand_errorsinputoutputoutput2 last_linefiles output_filesexpected_stale_modulesexpected_rechecked_modules deleted_pathsrDnormalize_outputexpected_fine_grained_targets)&r Z test_itemsrJrMZout_section_missingrurprqrmrnrtZ stale_modulesZrechecked_modulesrDr1itemZ first_itemcontentsZ file_entryZmpathfnamfZsrc_pathZpassnummodulesZ reprocessedmnumfullargsZ version_checkrQZ compare_op version_strr5Z tmp_outputrl file_pathrrrparse_test_case*s      &  $ $&( &( &&            "      rc seZdZUdZded<eeed<eeed<eeeefed<dZ dZ ee eefed <eee efed <eee efed <eeeefed <d Z eed<ee eeeeeffed<eee efed<eeed<ddeeeeeeeeeedd fdd ZddddZddddZddddZe eeefdddZd%eeeed d!d"Zeeedd#d$ZZS)&rzNHolds parsed data-driven test cases, and handles directory setup and teardown.DataSuiteCollectorparentrlrmrnrrprrrsrvTrorqrtrD DataSuiteN) rrIr%rH writescache only_whenplatformskipxfailrGr@r!c svt||||_||_||_||_|dkr8tjdksJ|dkrNtjdkrNd}||_| |_ | |_ | |_ d|_ d|_ dS)NZwindowswin32posixT)super__init__rIr%rrrfrrrrGr@old_cwdtmpdir) selfrrIr%rHrrrrrrGr@) __class__rrrs zDataDrivenTestCase.__init__)r!c Cs|jrtn|jr$|tjj|t}|dk s>td|}| y| |Wnt k r|j dd}|r|jdk sttj|tj|jj}td|jd|tj|s|jsttj|j|}t|jj|YnXdS)NzShould not happenz--save-failures-tozCopying data from test z to )rpytestrZ add_markermarkZ getparentrrRobjsetuprun_case ExceptionconfigZ getoptionrrLrrMbasenamerHprintisabsrshutilcopytree)rrrIsave_dir target_dirrrrruntest s*    zDataDrivenTestCase.runtestc szt|dt|_tjdd|_t|jjt t ix|j D]\}}t d|}|rt|d}|dksxtt dd|}t|}t|||}|g|qFtj|}tj|d d t|d d d } | |WdQRXqFWxT|jD]F\}} |dkstx.| D]&}t|}|gt||qWqWrTtnd} fddtd| dD|_ dS)N)r z mypy-test-)prefixz .*\.([0-9]+)$r"r4z \.[0-9]+$rT)exist_okwr()r)csg|]}|gqSr)get)r-r})stepsrrr2Dsz,DataDrivenTestCase.setup..)!rrLgetcwdrtempfileTemporaryDirectoryrchdirrHmkdirrrprTr[r:r_rRsubmodule_from_pathrr`rSrrXmakedirsrYwritertitemsrmaxranger) rrrr|r}rrZ operationdirrzpathsZmax_stepr)rrr"s4        "zDataDrivenTestCase.setupcCs\|jdk r|jdk stdt|jy|jWntk rJYnXd|_d|_dS)Nztest was not properly set up)rrrRrLrcleanupOSError)rrrrteardownFs zDataDrivenTestCase.teardowncCs|j|j|jfS)N)r%r@rH)rrrr reportinfoQszDataDrivenTestCase.reportinfo)excinfostyler!cCsF|tr|}n|j||jdd}d|jd|jd|S)Nshort)rzdata: :z: )Z errisinstance SystemExitZexconlyrZ_prunetracebackZgetreprr%r@)rrrZexcreprrrr repr_failureTs     zDataDrivenTestCase.repr_failurecCs|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. )r)rrrr find_stepsas zDataDrivenTestCase.find_steps)N)rrr__doc__rrrr r:r%r@rrrur r boolrrrrrrrr FileOperationr __classcell__rr)rrrsB     $  )rr!cCs<tdd|}td|}|d=d|}tdd|}|S)Nz\.pyi?$rz[/\\]rr=z \.__init__$)rTrr^rM)rpartsrrrrrms   rc@sXeZdZUdZdZdZeeed<e eed<dZ dZ eeee ee dddd Z dS) TestItemzVParsed test caseitem. An item is of the form [id arg] .. data .. rrQrGrN)rPrQrGr@r!cCs||_||_||_||_dS)N)rPrQrGr@)rrPrQrGr@rrrrszTestItem.__init__)rrrrrPrQrrrrr%r@r:rrrrrrws  r)raw_datarHr!c Csdd|dg|d}g}g}d}d}d}d}x&|t|krZ||} ||dr| dr|rt|}t|}|t||t||d|}| dd }d}d |kr|| d dd}|d| d }g}nn||d r |||ddnF||d s*|||n&||d rP|||dd|d7}q6W|rt|}t|}|t||||d|S)z>Parse a list of lines that represent a sequence of test items.rz[case ]r&Nr[r" z\[z--z----r4) r^r\r,rbendswithcollapse_line_continuation strip_listrSrindex) rrHlinesretrGrPrQiZi0srrrrFsB   rF)lr!cCsPg}x |D]}|tdd|q Wx$t|dkrJ|ddkrJ|q(W|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+$rrr)rSrTrr\pop)rrrrrrrs   rc Cs\g}d}xN|D]F}tdd|}|r@|dtdd|7<n |||d}qW|S)NFz\\$rrz^ +rA)rTrrSr)rrZcontrssrrrrs  r)rr!cCs |dtS)Nz)replaceroot_dir)rrrrr?sr?)rlrmryr!c Csxtt|D]}x||dddD]}td|}|r*|ddkrVd}n&|ddkrjd}n|dd kr|d }|d }|d }|d d}|dkr||d|dd|d|q*||d|dd|d|d| q*WqWdS)zTransform comments such as '# E: message' or '# E:3: message' in input. The result is lines like 'fnam:line: error: message'. z # r"Nz*^([ENW]):((?P\d+):)? (?P.*)$EerrorNZnoteWwarningcolmessagez\##rz: ) rr\r^rTsearchr,r_rrS) rlrmryrZpossible_err_commentr|Zseverityrrrrrrks&    "rk)r@r!cCsT|ttdd}td|}|s(|S|\}}}d|dd|pJd|SdS)z`Changes Windows paths to Linux paths in error messages. E.g. foo\bar.py -> foo/bar.py. rA/z^([\S/]+):(\d+:)?(\s+.*)z{}:{}{}rN)rrrTr[groupsrd)r@r|filenamelinenorrrrrBs rBcCsLtd|}|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&.Fz mypy.ini.T)anyrlrprLrrrb)r rrxrrrhas_stable_flagss rc@sPeZdZUeeed<eZeZ dZ dZ dZ ddddZ eeddd d ZdS) rrpFrN)r!cCsdS)zSetup fixtures (ad-hoc)Nr)rrrrrszDataSuite.setup)r r!cCstdS)N)NotImplementedError)rr rrrrszDataSuite.run_case)rrrrrrrrJrrrOrKrrrrrrrrrrs  )>ros.pathrLrrNrTrabcrrfrr*rrrrrrr r r r Ztyping_extensionsr Zmypy.test.configrrrrnormpathrrrrrrrZItemrrrrrFrrr?rkrBrrobjectrrZClassrZ Collectorrrrr rrrrrrsR 0   ' - /