o hg@sdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZmZmZmZmZddlmZmZdd ZGd d d ZGd d d ZGdddZGdddeZGdddeZGdddeZGdddeZGdddeZdS)z) Tests for L{pyflakes.scripts.pyflakes}. N)PYPY) UnusedImport)Reporter)maincheck checkPathcheckRecursiveiterSourceCode)TestCaseskipIfcOs.tj|}t_z ||i|W|t_S|t_w)z? Call C{f} with C{sys.stderr} redirected to C{stderr}. )sysstderr)r fargskwargsouterrL/var/www/html/ai/venv/lib/python3.10/site-packages/pyflakes/test/test_api.py withStderrTosrc@seZdZdZdddZdS)Nodez Mock an AST node. rcCs||_||_dSN)lineno col_offset)selfrrrrr__init__)s z Node.__init__N)r)__name__ __module__ __qualname____doc__rrrrrr%rc@s(eZdZdZddZddZddZdS) SysStreamCapturingzContext manager capturing sys.stdin, sys.stdout and sys.stderr. The file handles are replaced with a StringIO object. cCstj|pdtjd|_dS)Nnewline)ioStringIOoslinesep_stdin)rstdinrrrr4szSysStreamCapturing.__init__cCsPtj|_tj|_tj|_|jt_tj t j dt_|_ tj t j dt_|_ |S)Nr")r r) _orig_stdinstdout _orig_stdoutr _orig_stderrr(r$r%r&r'_stdout_stringio_stderr_stringiorrrr __enter__7szSysStreamCapturing.__enter__cGs4|j|_|j|_|jt_|jt_ |j t_ dSr) r.getvalueoutputr/errorr*r r)r,r+r-r )rrrrr__exit__Bs   zSysStreamCapturing.__exit__N)rrrrrr1r5rrrrr .s  r c@s0eZdZdZddZddZddZdd 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)rr7rrrrPs zLoggingReporter.__init__cCs|jdt|fdS)Nflake)r7appendstr)rmessagerrrr8XszLoggingReporter.flakecCs|jd||fdS)NunexpectedErrorr7r9)rfilenamer;rrrr<[szLoggingReporter.unexpectedErrorcCs|jd|||||fdS)N syntaxErrorr=)rr>msgroffsetlinerrrr?^szLoggingReporter.syntaxErrorN)rrrrrr8r<r?rrrrr6Ks  r6c@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS)TestIterSourceCodez& Tests for L{iterSourceCode}. cCst|_dSr)tempfilemkdtemptempdirr0rrrsetUpgszTestIterSourceCode.setUpcCt|jdSrshutilrmtreerFr0rrrtearDownjzTestIterSourceCode.tearDowncGs0|sJtjj|jg|R}t|d|S)Na)r&pathjoinrFopenclose)rpartsfpathrrr makeEmptyFilemsz TestIterSourceCode.makeEmptyFilecCs|tt|jggdS)zB There are no Python files in an empty directory. N) assertEquallistr rFr0rrrtest_emptyDirectoryssz&TestIterSourceCode.test_emptyDirectorycCs(|d}|tt|jg|gdS)zd If the directory contains one Python file, C{iterSourceCode} will find it. foo.pyNrUrVrWr rF)r childpathrrrtest_singleFileys z"TestIterSourceCode.test_singleFilecCs&|d|tt|jggdS)zJ Files that are not Python source files are not included. zfoo.pycNrZr0rrrtest_onlyPythonSources z(TestIterSourceCode.test_onlyPythonSourcecCsttj|jd|dd}|ddttj|jd|dd}|d}|tt|jgt|||gdS)zk If the Python files are hidden deep down in child directories, we will find them. fooa.pyza.py~barb.pyzc.pyN) r&mkdirrOrPrFrUrVsortedr )rapathbpathcpathrrr test_recursess     z TestIterSourceCode.test_recursesc CsTtj|jd}t|d }|dWdn1swY|dttj|jdd }|dWdn1sBwYtj|jd}t|d }|d Wdn1sdwYtj|jd }t|d }|d Wdn1swYtj|jd }t|d }|d Wdn1swYtj|jd}t|d }|dWdn1swYtj|jd}t|d }|dWdn1swYtj|jd}t|d }|dWdn 1swY|tt |jgt||||||gdS)zd Find Python files that don't end with `.py`, but contain a Python shebang. rNwz#!/usr/bin/env python Nbcz hello world ez#!/usr/bin/env python3 rz#!/usr/bin/env pythonw gz#!/usr/bin/python3 -u iz#!/usr/local/bin/python3d jz#! /usr/bin/env python3.8m lz#!/bin/sh #!/usr/bin/python ) r&rOrPrFrQwriterUrVrcr ) rpythonfdpython3pythonw python3argspython3d python38mnotfirstrrr test_shebangsN                 zTestIterSourceCode.test_shebangcCsptj|jd}tj|jd}t||dd}t||dd}|tt||gt||gdS)zr L{iterSourceCode} can be given multiple directories. It will recurse into each of them. r^r`r_raN) r&rOrPrFrbrUrVrcr )rfoopathbarpathrdrerrrtest_multipleDirectoriess     z+TestIterSourceCode.test_multipleDirectoriescCs&|d}|tt|g|gdS)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)rUrVrWr )repathrrrtest_explicitFiless z%TestIterSourceCode.test_explicitFilesN)rrrrrGrLrUrXr\r]rgryr|r~rrrrrCbs. rCc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) TestReporterz Tests for L{Reporter}. cCs8t}td|}|ddddd|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. NrY a problembad line of sourcez2foo.py:3:8: a problem bad line of source ^ r$r%rr?rVr2rerrreporterrrrtest_syntaxErrors zTestReporter.test_syntaxErrorcCs8t}td|}|ddddd|d|dS)zy C{syntaxError} doesn't include a caret pointing to the error if C{offset} is passed as C{None}. NrYrrrz'foo.py:3: a problem bad line of source rrrrrtest_syntaxErrorNoOffsets  z%TestReporter.test_syntaxErrorNoOffsetcCs8t}td|}|ddddd|d|dS)z C{syntaxError} doesn't include text or nonsensical offsets if C{text} is C{None}. This typically happens when reporting syntax errors from stdin. Nrrz:1:1: a problem rrrrrtest_syntaxErrorNoTexts z#TestReporter.test_syntaxErrorNoTextc Csft}ddg}td|}|dddt|ddd ||d |d d d d |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. rzmore bad lines of sourceNrYrrr zfoo.py:3:25: a problem z ^ )r$r%rr?lenrPrVr2)rrlinesrrrrtest_multiLineSyntaxError s( z&TestReporter.test_multiLineSyntaxErrorcCs2t}td|}|dd|d|dS)zO C{unexpectedError} reports an error processing a source file. Nz source.pyz error messagezsource.py: error message )r$r%rr<rVr2rrrrtest_unexpectedError s  z!TestReporter.test_unexpectedErrorcCsFt}t|d}tdtdd}|||||ddS)z C{flake} reports a code warning from Pyflakes. It is exactly the str() of a L{pyflakes.messages.Message}. NrY*r`r)r$r%rrrr8rVr2)routrr;rrr test_flake)s   zTestReporter.test_flakeN) rrrrrrrrrrrrrrrs  rc@seZdZdZejddZddZddZdd Z d d Z d d Z ddZ ddZ ddZddZddZddZeejdkdddZddZd d!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\}}z0t|d}t|ds|d}||Wdn1s(wY|VWt|dSt|w)zV Make a temporary file containing C{content} and return a path to it. wbdecodeasciiN)rDmkstempr&fdopenhasattrencoderpremove)rcontentrrnamerrrr makeTempFile:s    zCheckTests.makeTempFilecCs:t}t|t|}|||ft|d|fdS)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$r%rrrVr2rrP)rrO errorListrcountrrrassertHasErrorsIs  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. )r6r)rrOr7rrrrr getErrorsUs  zCheckTests.getErrorscCsddlm}||jtdS)Nr)pyflakes)pyflakes.scriptsrassertIsr)rscript_pyflakesrrrtest_legacyScriptcs zCheckTests.test_legacyScriptcC<|d}||gWddS1swYdS)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 Nrr)rfNamerrrtest_missingTrailingNewlinegs "z&CheckTests.test_missingTrailingNewlinecCs,|d\}}||d||dgdS)z: L{checkPath} handles non-existing files. extremo)r<rzNo such file or directoryN)rrV)rrerrorsrrrtest_checkPathNonExistingps  z$CheckTests.test_checkPathNonExistingc Csd}dd}z||Wn$ty0}zts&tjdkr&||jddkWYd}~n d}~ww|||5}tr@d}n tjdkrHd }nd }tsQtjdkrTd }nd }| |d |||d|dfgWddS1srwYdS)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|dSr)exec)sourcerrrevaluates z6CheckTests.test_multilineSyntaxError..evaluater rrNz=end of file (EOF) while scanning triple-quoted string literalz>unterminated triple-quoted string literal (detected at line 8)zinvalid syntax rz%s:8:%d: %s '''quux''' %s^  ) SyntaxErrorrr version_info assertTruetextrfailrr)rrrrk sourcePathr;columnrrrtest_multilineSyntaxErrorzs4    "z$CheckTests.test_multilineSyntaxErrorcCs|d9}tr d}n tjdkrd}nd}tstjdkrd}nd}d|d }d ||||}|||gWd d S1sAwYd S) The error reported for source files which end prematurely causing a syntax error reflects the cause for the syntax error. zdef foo(zparenthesis is never closedrz'(' was never closedzunexpected EOF while parsingr rrz{}:1:{}: {} def foo( {}^ N)rrr rformatr)rrr@rspacesexpectedrrrtest_eofSyntaxErrors   "zCheckTests.test_eofSyntaxErrorcCsD|d}|||dgWddS1swYdS)rzif True: foo =z$:2:7: invalid syntax foo = ^ Nrrrrrrtest_eofSyntaxErrorWithTabs "z%CheckTests.test_eofSyntaxErrorWithTabc Csd}||N}tjdkrd}nd}trtjdkrd}ntr!d}ntjdkr)d}n tjdkr1d }nd }d |d d }|||d|d|d|gWddS1sXwYdS)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 rrz:1:5: Generator expression must be parenthesized,max(1 for i in range(10), key=lambda x: x+1)z ^)zL:1:4: Generator expression must be parenthesized if not sole argumentrz ^r) r$r%rrrVr2splitr rr)rrrrerrlinesexpected_errorrrrtest_stdinReportsErrorss    z"CheckTests.test_stdinReportsErrorsN)rrrr contextlibcontextmanagerrrrrrrrrrrrrr r platformrrrrrrrrrrrrr5s0    1#    rc@sZeZdZdZddZddZddZdd d Zd d Zd dZ ddZ ddZ ddZ dS)IntegrationTestszF Tests of the pyflakes script that actually spawn the script. cCs t|_tj|jd|_dS)Ntemp)rDrErFr&rOrP tempfilepathr0rrrrGs zIntegrationTests.setUpcCrHrrIr0rrrrLrMzIntegrationTests.tearDowncCs(ddl}tj|j}tj|dddS)z9 Return the path to the pyflakes binary. rNz..binr)rr&rOdirname__file__rP)rr package_dirrrrgetPyflakesBinarysz"IntegrationTests.getPyflakesBinaryNc Csttj}tjtj|d<tj|g}| ||r4t j ||t j t j t j d}| |d\}}nt j ||t j t j d}| \}}|}|d}|d}|||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. PYTHONPATH)envr)r+r r)rr+r r)dictr&environpathseprPr rO executablerextend subprocessPopenPIPE communicaterwaitr) rpathsr)rcommandpr+r rvrrr runPyflakess"       zIntegrationTests.runPyflakescCs.t|jd||jg}||ddS)z When a Python source file is all good, the return code is zero and no messages are printed to either stdout or stderr. rN)r!r!rN)rQrrRrrV)rdrrr test_goodFileszIntegrationTests.test_goodFilecCsvt|jd }|dWdn1swY||jg}t|jtdd}|||tjddfdS)z When a Python source file has warnings, the return code is non-zero and the warnings are printed to stdout. rsimport contraband Nrrr!) rQrrprrrrVr&r')rrrrrrrrtest_fileWithFlakess   z$IntegrationTests.test_fileWithFlakescCs4||jg}d|jtj}||d|dfdS) 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{}: No such file or directory{}r!rN)rrrr&r'rV)rr error_msgrrrtest_errors_ios zIntegrationTests.test_errors_iocCsjt|jd }|dWdn1swY||jg}d|jtj}||d|dfdS)rrsimportNz-{0}:1:7: invalid syntax{1}import{1} ^{1}r!r)rQrrprrr&r'rV)rrrrrrrrtest_errors_syntaxs z#IntegrationTests.test_errors_syntaxcCs>|jgdd}tdtdd}|||tjddfdS)zU If no arguments are passed to C{pyflakes} then it reads from stdin. zimport contraband)r)rrrr!N)rrrrVr&r')rrrrrrtest_readFromStdin s z#IntegrationTests.test_readFromStdinr) rrrrrGrLrrrrrrrrrrrrs    rc@seZdZdZdddZdS)TestMainz. Tests of the pyflakes main function. Nc Csz#t| }t|dWdn 1swYWt dWt dtyH}z||jtt|j}|j|j|fWYd}~Sd}~ww)N)rzSystemExit not raised) r r SystemExitassertIsInstancecodeboolintr3r4 RuntimeError)rrr)capturerkrrrrrs   zTestMain.runPyflakesr)rrrrrrrrrr rr ) rrr$r&r rJrrDpyflakes.checkerrpyflakes.messagesrpyflakes.reporterr pyflakes.apirrrrr pyflakes.test.harnessr r rrr r6rCrrrr rrrrs2     |Wb