B >¶†\_eã@s–dZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZmZmZmZddlmZmZejdkr˜dd lmZndd lmZeZyejd ZWnek rÎd ZYnXy ed ZWne k rôd ZYnXejd kpeZ!Z"d d„Z#Gdd„de$ƒZ%Gdd„de$ƒZ&Gdd„de$ƒZ'Gdd„deƒZ(Gdd„deƒZ)Gdd„deƒZ*Gdd„deƒZ+Gdd„de+ƒZ,dS)z) Tests for L{pyflakes.scripts.pyflakes}. éN)ÚPY2)Ú UnusedImport)ÚReporter)ÚmainÚ checkPathÚcheckRecursiveÚiterSourceCode)ÚTestCaseÚskipIf)é)ÚStringIOTF)r écOs&tj|}t_z |||ŽS|t_XdS)z? Call C{f} with C{sys.stderr} redirected to C{stderr}. N)ÚsysÚstderr)rÚfÚargsÚkwargsZouter©rú5lib/python3.7/site-packages/pyflakes/test/test_api.pyÚ withStderrTo,s rc@seZdZdZddd„ZdS)ÚNodez Mock an AST node. rcCs||_||_dS)N)ÚlinenoÚ col_offset)ÚselfrrrrrÚ__init__;sz Node.__init__N)r)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrrr7src@s2eZdZdZd dd„Zdd„Zdd„Zd d „ZdS) ÚSysStreamCapturinga/ Context manager capturing sys.stdin, sys.stdout and sys.stderr. The file handles are replaced with a StringIO object. On environments that support it, the StringIO object uses newlines set to os.linesep. Otherwise newlines are converted from \n to os.linesep during __exit__. NcCsDyt|tjdStk r>d|_|dkr2tƒSt|ƒSYnXdS)N)ÚnewlineT)r ÚosÚlinesepÚ TypeErrorÚ_newline)rÚbufferrrrÚ_create_StringIOKsz#SysStreamCapturing._create_StringIOcCsd|_| |pd¡|_dS)NFÚ)r$r&Ú_stdin)rÚstdinrrrrWszSysStreamCapturing.__init__cCsDtj|_tj|_tj|_|jt_| ¡t_|_ | ¡t_|_ |S)N) rr)Ú _orig_stdinÚstdoutÚ _orig_stdoutrÚ _orig_stderrr(r&Ú_stdout_stringioÚ_stderr_stringio)rrrrÚ __enter__[szSysStreamCapturing.__enter__cGsh|j ¡|_|j ¡|_|jrLtjdkrL|j dtj¡|_|j dtj¡|_|j t _ |j t _ |jt _dS)NÚ )r.ÚgetvalueÚoutputr/Úerrorr$r!r"Úreplacer*rr)r,r+r-r)rrrrrÚ__exit__fs  zSysStreamCapturing.__exit__)N)rrrrr&rr0r6rrrrr@s    rc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) ÚLoggingReporterzK Implementation of Reporter that just appends any error to a list. cCs ||_dS)zh Construct a C{LoggingReporter}. @param log: A list to append log messages to. N)Úlog)rr8rrrrxszLoggingReporter.__init__cCs|j dt|ƒf¡dS)NÚflake)r8ÚappendÚstr)rÚmessagerrrr9€szLoggingReporter.flakecCs|j d||f¡dS)NÚunexpectedError)r8r:)rÚfilenamer<rrrr=ƒszLoggingReporter.unexpectedErrorcCs|j d|||||f¡dS)NÚ syntaxError)r8r:)rr>ÚmsgrÚoffsetÚlinerrrr?†szLoggingReporter.syntaxErrorN)rrrrrr9r=r?rrrrr7ss r7c@s`eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dS)ÚTestIterSourceCodez& Tests for L{iterSourceCode}. cCst ¡|_dS)N)ÚtempfileÚmkdtempÚtempdir)rrrrÚsetUpszTestIterSourceCode.setUpcCst |j¡dS)N)ÚshutilÚrmtreerF)rrrrÚtearDown’szTestIterSourceCode.tearDowncGs.|st‚tjj|jf|žŽ}t|dƒ ¡|S)NÚa)ÚAssertionErrorr!ÚpathÚjoinrFÚopenÚclose)rÚpartsZfpathrrrÚ makeEmptyFile•sz TestIterSourceCode.makeEmptyFilecCs| tt|jgƒƒg¡dS)zB There are no Python files in an empty directory. N)Ú assertEqualÚlistrrF)rrrrÚtest_emptyDirectory›sz&TestIterSourceCode.test_emptyDirectorycCs(| d¡}| tt|jgƒƒ|g¡dS)zd If the directory contains one Python file, C{iterSourceCode} will find it. zfoo.pyN)rRrSrTrrF)rZ childpathrrrÚtest_singleFile¡s z"TestIterSourceCode.test_singleFilecCs&| d¡| tt|jgƒƒg¡dS)zJ Files that are not Python source files are not included. zfoo.pycN)rRrSrTrrF)rrrrÚtest_onlyPythonSource©s z(TestIterSourceCode.test_onlyPythonSourcecCs€t tj |jd¡¡| dd¡}| dd¡t tj |jd¡¡| dd¡}| d¡}| tt|jgƒƒt|||gƒ¡dS)zk If the Python files are hidden deep down in child directories, we will find them. Úfooza.pyza.py~Úbarzb.pyzc.pyN) r!ÚmkdirrMrNrFrRrSÚsortedr)rÚapathÚbpathZcpathrrrÚ test_recurses°s    z TestIterSourceCode.test_recursesc Cstj |jd¡}t|dƒ}| d¡WdQRX| d¡ttj |jd¡dƒ}| d¡WdQRXtj |jd¡}t|dƒ}| d ¡WdQRXtj |jd ¡}t|dƒ}| d ¡WdQRXtj |jd ¡}t|dƒ}| d ¡WdQRX| tt |jgƒƒt||||gƒ¡dS)zd Find Python files that don't end with `.py`, but contain a Python shebang. rKÚwz#!/usr/bin/env python NÚbÚcz hello world Údz#!/usr/bin/env python2 Úez#!/usr/bin/env python3 rz#!/usr/bin/env pythonw ) r!rMrNrFrOÚwriterRrSr[r)rZpythonÚfdZpython2Zpython3ZpythonwrrrÚ test_shebang¿s$     zTestIterSourceCode.test_shebangcCsptj |jd¡}tj |jd¡}t |¡| dd¡}t |¡| dd¡}| tt||gƒƒt||gƒ¡dS)zr L{iterSourceCode} can be given multiple directories. It will recurse into each of them. rXrYza.pyzb.pyN) r!rMrNrFrZrRrSr[r)rZfoopathZbarpathr\r]rrrÚtest_multipleDirectoriesÝs    z+TestIterSourceCode.test_multipleDirectoriescCs&| d¡}| tt|gƒƒ|g¡dS)z‹ If one of the paths given to L{iterSourceCode} is not a directory but a file, it will include that in its output. ze.pyN)rRrSrTr)rZepathrrrÚtest_explicitFilesìs z%TestIterSourceCode.test_explicitFilesN)rrrrrGrJrRrUrVrWr^rfrgrhrrrrrCŠsrCc@s8eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S) Ú TestReporterz Tests for L{Reporter}. cCsDtƒ}td|ƒ}| dddtjdkr(dndd¡| d | ¡¡dS) a  C{syntaxError} reports that there was a syntax error in the source file. It reports to the error stream and includes the filename, line number, error message, actual line of source and a caret pointing to where the error is. Nzfoo.pyz a problemr )r érjézbad line of sourcez2foo.py:3:8: a problem bad line of source ^ )r rr?rÚ version_inforSr2)rÚerrÚreporterrrrÚtest_syntaxErrorûs  zTestReporter.test_syntaxErrorcCs6tƒ}td|ƒ}| ddddd¡| d| ¡¡dS)zy C{syntaxError} doesn't include a caret pointing to the error if C{offset} is passed as C{None}. Nzfoo.pyz a problemr zbad line of sourcez'foo.py:3: a problem bad line of source )r rr?rSr2)rrmrnrrrÚtest_syntaxErrorNoOffset s  z%TestReporter.test_syntaxErrorNoOffsetc Cs‚tƒ}ddg}td|ƒ}| dddt|dƒdd  |¡¡tjd krJd nd}| d ||d d d|dd| ¡¡dS)z¸ If there's a multi-line syntax error, then we only report the last line. The offset is adjusted so that it is relative to the start of the last line. zbad line of sourcezmore bad lines of sourceNzfoo.pyz a problemr rrkr1)r rjézfoo.py:3:%d: a problem éÿÿÿÿú éz^ ) r rr?ÚlenrNrrlrSr2)rrmÚlinesrnÚcolumnrrrÚtest_multiLineSyntaxErrors  "z&TestReporter.test_multiLineSyntaxErrorcCs0tƒ}td|ƒ}| dd¡| d| ¡¡dS)zO C{unexpectedError} reports an error processing a source file. Nz source.pyz error messagezsource.py: error message )r rr=rSr2)rrmrnrrrÚtest_unexpectedError0s  z!TestReporter.test_unexpectedErrorcCsDtƒ}t|dƒ}tdtdƒdƒ}| |¡| | ¡d|f¡dS)z„ C{flake} reports a code warning from Pyflakes. It is exactly the str() of a L{pyflakes.messages.Message}. Nzfoo.pyé*rYz%s )r rrrr9rSr2)rÚoutrnr<rrrÚ test_flake9s   zTestReporter.test_flakeN) rrrrrorprxryr|rrrrriös  ric@s¾eZdZdZejdd„ƒZdd„Zdd„Zdd „Z d d „Z d d „Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zeejdkdƒdd„ƒZdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*S)+Ú CheckTestszL Tests for L{check} and L{checkPath} which check a file for flakes. c cs^t ¡\}}z@t |d¡$}t|dƒs0| d¡}| |¡WdQRX|VWdt |¡XdS)zV Make a temporary file containing C{content} and return a path to it. ÚwbÚdecodeÚasciiN)rDZmkstempr!ÚfdopenÚhasattrÚencoderdÚremove)rZcontentreÚnamerrrrÚ makeTempFileJs    zCheckTests.makeTempFilecCs8tƒ}t|t|ƒ}| || ¡ft|ƒd |¡f¡dS)z² Assert that C{path} causes errors. @param path: A path to a file to check. @param errorList: A list of errors expected to be printed to stderr. r'N)r rrrSr2rurN)rrMZ errorListrmÚcountrrrÚassertHasErrorsYs zCheckTests.assertHasErrorscCsg}t|ƒ}t||ƒ}||fS)a‘ Get any warnings or errors reported by pyflakes for the file at C{path}. @param path: The path to a Python file on disk that pyflakes will check. @return: C{(count, log)}, where C{count} is the number of warnings or errors generated, and log is a list of those warnings, presented as structured data. See L{LoggingReporter} for more details. )r7r)rrMr8rnr‡rrrÚ getErrorses  zCheckTests.getErrorscCsddlm}| |jt¡dS)Nr)Úpyflakes)Zpyflakes.scriptsrŠZassertIsr)rZscript_pyflakesrrrÚtest_legacyScriptss zCheckTests.test_legacyScriptc Cs&| d¡}| |g¡WdQRXdS)z§ Source which doesn't end with a newline shouldn't cause any exception to be raised nor an error indicator to be returned by L{check}. zdef foo(): pass N)r†rˆ)rZfNamerrrÚtest_missingTrailingNewlinews z&CheckTests.test_missingTrailingNewlinecCs,| d¡\}}| |d¡| |dg¡dS)z: L{checkPath} handles non-existing files. Úextremort)r=rzNo such file or directoryN)r‰rS)rr‡ÚerrorsrrrÚtest_checkPathNonExisting€s  z$CheckTests.test_checkPathNonExistingc Cs¸d}dd„}y ||ƒWn:tk rRt ¡d}tsN| |j d¡dk¡Yn X| ¡| |¡H}trrd}nd}tj dkr„d nd }|  |d |||d |dfg¡Wd QRXd S)zÏ Source which includes a syntax error which results in the raised L{SyntaxError.text} containing multiple lines of source are reported with only the last line of that source. zCdef foo(): ''' def bar(): pass def baz(): '''quux''' cSs t|ƒdS)N)Úexec)ÚsourcerrrÚevaluatesz6CheckTests.test_multilineSyntaxError..evaluatertr1z/EOF while scanning triple-quoted string literalzinvalid syntax)r rjrjé z%s:8:%d: %s '''quux''' %s^ rsN) Ú SyntaxErrorrÚexc_infoÚPYPYZ assertTrueÚtextr‡Zfailr†rlrˆ)rr‘r’rcÚ sourcePathr<rwrrrÚtest_multilineSyntaxErrorŠs"   z$CheckTests.test_multilineSyntaxErrorc CsB| d¡.}trd|f}n d|f}| ||g¡WdQRXdS)z“ The error reported for source files which end prematurely causing a syntax error reflects the cause for the syntax error. zdef foo(z5%s:1:7: parenthesis is never closed def foo( ^ z8%s:1:9: unexpected EOF while parsing def foo( ^ N)r†r–rˆ)rr˜ÚresultrrrÚtest_eofSyntaxError·s   zCheckTests.test_eofSyntaxErrorc CsJ| d¡6}trdnd}tr dnd}| |d|||fg¡WdQRXdS)z“ The error reported for source files which end prematurely causing a syntax error reflects the cause for the syntax error. zif True: foo =érkz ^z ^z"%s:2:%s: invalid syntax foo = %s N)r†r–rˆ)rr˜rwÚ last_linerrrÚtest_eofSyntaxErrorWithTabÎs    z%CheckTests.test_eofSyntaxErrorWithTabc Csnd}| |¡V}tr@tjdkr"dnd}d|dd}d|}nd }}| |d |||fg¡Wd QRXd S) zÎ Source which has a non-default argument following a default argument should include the line number of the syntax error. However these exceptions do not include an offset. z def foo(bar=baz, bax): pass )r rjé rjrsrtz^ z%d:r'zO%s:1:%s non-default argument follows default argument def foo(bar=baz, bax): %sN)r†ÚERROR_HAS_LAST_LINErrlrˆ)rr‘r˜rwrÚ columnstrrrrÚ(test_nonDefaultFollowsDefaultSyntaxErrorßs   z3CheckTests.test_nonDefaultFollowsDefaultSyntaxErrorc Cs„d}| |¡l}tr@tjdkr"dnd}d|dd}d|}nd }}tjd krXd }nd }| |d ||||fg¡WdQRXdS)zÊ Source which has a non-keyword argument after a keyword argument should include the line number of the syntax error. However these exceptions do not include an offset. zfoo(bar=baz, bax) )r rjéé rsrtz^ z%d:r')r rœz,positional argument follows keyword argumentz!non-keyword arg after keyword argz%s:1:%s %s foo(bar=baz, bax) %sN)r†r rrlrˆ)rr‘r˜rwrr¡r<rrrÚ&test_nonKeywordAfterKeywordSyntaxError÷s   z1CheckTests.test_nonKeywordAfterKeywordSyntaxErrorc Csštj}| d¡€}|dkr&d|f}nXd}tr4d}n d}|dksP|dd …d krTd }trjd d |d}nd}d||||f}| ||g¡WdQRXdS)zI The invalid escape syntax raises ValueError in Python 2 z foo = '\xyz')r z%s: problem decoding source rtérk)r r éNr )r r rr z%s^ rsr'zy%s:1:%d: (unicode error) 'unicodeescape' codec can't decode bytes in position 0-%d: truncated \xXX escape foo = '\xyz' %s)rrlr†r–r rˆ)rZverr˜Zdecoding_errorZ position_endrwrrrrÚtest_invalidEscapes   zCheckTests.test_invalidEscapeZwin32zunsupported on Windowsc Csjt ¡dkr| d¡| d¡@}t |d¡| |¡\}}| |d¡| |d|dfg¡WdQRXdS)za If the source file is not readable, this is reported on standard error. rz8root user can access all files regardless of permissionsr'rtr=zPermission deniedN)r!ÚgetuidZskipTestr†Úchmodr‰rS)rr˜r‡rŽrrrÚtest_permissionDenied5s     z CheckTests.test_permissionDeniedc CsV| d¡B}| |¡\}}| |d¡| |dtt|tdƒdƒƒfg¡WdQRXdS)zc If the source file has a pyflakes warning, this is reported as a 'flake'. z import foortr9rXN)r†r‰rSr;rr)rr˜r‡rŽrrrÚtest_pyflakesWarningFs   zCheckTests.test_pyflakesWarningc Cs<tdƒ}d| d¡}| |¡}| |g¡WdQRXdS)zU If source file declares the correct encoding, no error is reported. i&z# coding: utf-8 x = "%s" zutf-8N)Úunichrrƒr†rˆ)rÚSNOWMANr‘r˜rrrÚtest_encodedFileUTF8Qs zCheckTests.test_encodedFileUTF8c Cs&| d¡}| |g¡WdQRXdS)zW Source files with Windows CR LF line endings are parsed successfully. zx = 42 N)r†rˆ)rr˜rrrÚtest_CRLFLineEndings]s zCheckTests.test_CRLFLineEndingsc Csltdƒ}d| d¡}| |¡B}trBtjdkrBd}d||f}nd}d|f}| ||g¡Wd QRXd S) zn If a source file contains bytes which cannot be decoded, this is reported on stderr. i&z# coding: ascii x = "%s" zutf-8)r zN'ascii' codec can't decode byte 0xe2 in position 21: ordinal not in range(128)u"%s:0:0: %s x = "☃" ^ zproblem decoding sourcez%s: problem decoding source N)r­rƒr†r–rrlrˆ)rr®r‘r˜r<ršrrrÚtest_misencodedFileUTF8ds  z"CheckTests.test_misencodedFileUTF8c CsDtdƒ}d| d¡}| |¡}| |d|fg¡WdQRXdS)zn If a source file contains bytes which cannot be decoded, this is reported on stderr. i&z# coding: ascii x = "%s" zutf-16z%s: problem decoding source N)r­rƒr†rˆ)rr®r‘r˜rrrÚtest_misencodedFileUTF16~s  z#CheckTests.test_misencodedFileUTF16c Cst ¡}zæt tj |d¡¡tj |dd¡}t|dƒ}| d d¡¡WdQRXtj |d¡}t|dƒ}| d d¡¡WdQRXg}t |ƒ}t |g|ƒ}|  |d ¡|  t |ƒt d t t|td ƒd ƒƒfd t t|td ƒd ƒƒfgƒ¡Wdt |¡XdS)zv L{checkRecursive} descends into each directory, finding Python files and reporting problems. rXzbar.pyr~z import baz r€Nzbaz.pyzimport contrabandr r9rtZbazÚ contraband)rDrEr!rZrMrNrOrdrƒr7rrSr[r;rrrHrI)rrFZfile1reZfile2r8rnÚwarningsrrrÚtest_checkRecursiveŒs&     zCheckTests.test_checkRecursiveN)rrrrÚ contextlibÚcontextmanagerr†rˆr‰r‹rŒrr™r›ržr¢r¥r¨r rÚplatformr«r¬r¯r°r±r²rµrrrrr}Es(   -!  r}c@sZeZdZdZdd„Zdd„Zdd„Zdd d „Zd d „Zd d„Z dd„Z dd„Z dd„Z dS)ÚIntegrationTestszF Tests of the pyflakes script that actually spawn the script. cCs t ¡|_tj |jd¡|_dS)NZtemp)rDrErFr!rMrNÚ tempfilepath)rrrrrG¬s zIntegrationTests.setUpcCst |j¡dS)N)rHrIrF)rrrrrJ°szIntegrationTests.tearDowncCs(ddl}tj |j¡}tj |ddd¡S)z9 Return the path to the pyflakes binary. rNz..ÚbinrŠ)rŠr!rMÚdirnameÚ__file__rN)rrŠZ package_dirrrrÚgetPyflakesBinary³sz"IntegrationTests.getPyflakesBinaryNc CsÒttjƒ}tj tj¡|d<tj| ¡g}|  |¡|rht j ||t j t j t j d}|  | d¡¡\}}n"t j ||t j t j d}|  ¡\}}| ¡}tjdkr°| d¡}| d¡}trÈtrÈtrÈ| dd¡}|||fS) a  Launch a subprocess running C{pyflakes}. @param paths: Command-line arguments to pass to pyflakes. @param stdin: Text to use as stdin. @return: C{(returncode, stdout, stderr)} of the completed pyflakes process. Z PYTHONPATH)Úenvr)r+rr€)r¿r+r)r zutf-8z z )Údictr!ÚenvironÚpathseprNrrMÚ executabler¾ÚextendÚ subprocessÚPopenÚPIPEZ communicaterƒÚwaitrlrr–rÚWINr5) rÚpathsr)r¿ZcommandÚpr+rÚrvrrrÚ runPyflakes»s$         zIntegrationTests.runPyflakescCs.t|jdƒ ¡| |jg¡}| |d¡dS)z When a Python source file is all good, the return code is zero and no messages are printed to either stdout or stderr. rK)r'r'rN)rOrºrPrÍrS)rrbrrrÚ test_goodFileÙszIntegrationTests.test_goodFilec Csht|jdƒ}| d d¡¡WdQRX| |jg¡}t|jtdƒdƒ}| |d|tj fddf¡dS) z… When a Python source file has warnings, the return code is non-zero and the warnings are printed to stdout. r~zimport contraband r€Nrtr³z%s%sr') rOrºrdrƒrÍrrrSr!r")rrerbÚexpectedrrrÚtest_fileWithFlakesâs z$IntegrationTests.test_fileWithFlakescCs4| |jg¡}d|jtjf}| |d|df¡dS)zº When pyflakes finds errors with the files it's given, (if they don't exist, say), then the return code is non-zero and the errors are printed to stderr. z%s: No such file or directory%sr'rtN)rÍrºr!r"rS)rrbÚ error_msgrrrÚtest_errors_ioís zIntegrationTests.test_errors_ioc Cspt|jdƒ}| d d¡¡WdQRX| |jg¡}d |jtjtrJdndtrTdnd ¡}|  |d|d f¡dS) zº When pyflakes finds errors with the files it's given, (if they don't exist, say), then the return code is non-zero and the errors are printed to stderr. r~Úimportr€Nz0{0}:1:{2}: invalid syntax{1}import{1} {3}^{1}rœrkr'z rt) rOrºrdrƒrÍÚformatr!r"r–rS)rrerbrÑrrrÚtest_errors_syntaxøs  z#IntegrationTests.test_errors_syntaxcCs>|jgdd}tdtdƒdƒ}| |d|tjfddf¡dS) zU If no arguments are passed to C{pyflakes} then it reads from stdin. zimport contraband)r)zrtr³z%s%sr'N)rÍrrrSr!r")rrbrÏrrrÚtest_readFromStdinsz#IntegrationTests.test_readFromStdin)N) rrrrrGrJr¾rÍrÎrÐrÒrÕrÖrrrrr¹§s     r¹c@seZdZdZddd„ZdS)ÚTestMainz. Tests of the pyflakes main function. Nc Csty"t|ƒ}t|dWdQRXWnDtk rf}z&| |jt¡t|jƒ}|j|j|fSd}~XYn Xt dƒ‚dS)N)rzSystemExit not raised) rrÚ SystemExitZassertIsInstanceÚcodeÚboolÚintr3r4Ú RuntimeError)rrÊr)ZcapturercrÌrrrrÍs  zTestMain.runPyflakes)N)rrrrrÍrrrrr×sr×)-rr¶r!rrHrÅrDZpyflakes.checkerrZpyflakes.messagesrZpyflakes.reporterrZ pyflakes.apirrrrZpyflakes.test.harnessr r rlZ cStringIOr ÚioÚchrr­Zpypy_version_infor–ÚAttributeErrorZ WindowsErrorrÉÚ NameErrorZERROR_HAS_COL_NUMr rÚobjectrrr7rCrir}r¹r×rrrrÚsJ         3lOdg