File: //usr/local/lib/python3.7/test/__pycache__/test_cmd_line_script.cpython-37.pyc
B
��g�t � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
d dlmZmZmZmZmZmZmZmZ e
jZdddgZdZefdd �Zed
fd d �ZG d
d� dej �Z dd� Z e dkr�e � dS )� N)�support)�make_pkg� make_script� make_zip_pkg�make_zip_script�assert_python_ok�assert_python_failure� spawn_python� kill_pythonZtest1Ztest2Ztest3a� # Script may be run with optimisation enabled, so don't rely on assert
# statements being executed
def assertEqual(lhs, rhs):
if lhs != rhs:
raise AssertionError('%r != %r' % (lhs, rhs))
def assertIdentical(lhs, rhs):
if lhs is not rhs:
raise AssertionError('%r is not %r' % (lhs, rhs))
# Check basic code execution
result = ['Top level assignment']
def f():
result.append('Lower level reference')
f()
assertEqual(result, ['Top level assignment', 'Lower level reference'])
# Check population of magic variables
assertEqual(__name__, '__main__')
from importlib.machinery import BuiltinImporter
_loader = __loader__ if __loader__ is BuiltinImporter else type(__loader__)
print('__loader__==%a' % _loader)
print('__file__==%a' % __file__)
print('__cached__==%a' % __cached__)
print('__package__==%r' % __package__)
# Check PEP 451 details
import os.path
if __package__ is not None:
print('__main__ was located through the import system')
assertIdentical(__spec__.loader, __loader__)
expected_spec_name = os.path.splitext(os.path.basename(__file__))[0]
if __package__:
expected_spec_name = __package__ + "." + expected_spec_name
assertEqual(__spec__.name, expected_spec_name)
assertEqual(__spec__.parent, __package__)
assertIdentical(__spec__.submodule_search_locations, None)
assertEqual(__spec__.origin, __file__)
if __spec__.cached is not None:
assertEqual(__spec__.cached, __cached__)
# Check the sys module
import sys
assertIdentical(globals(), sys.modules[__name__].__dict__)
if __spec__ is not None:
# XXX: We're not currently making __main__ available under its real name
pass # assertIdentical(globals(), sys.modules[__spec__.name].__dict__)
from test import test_cmd_line_script
example_args_list = test_cmd_line_script.example_args
assertEqual(sys.argv[1:], example_args_list)
print('sys.argv[0]==%a' % sys.argv[0])
print('sys.path[0]==%a' % sys.path[0])
# Check the working directory
import os
print('cwd==%a' % os.getcwd())
c C s t | ||�}t�� |S )N)r � importlib�invalidate_caches)�
script_dir�script_basename�source� to_return� r �5/usr/local/lib/python3.7/test/test_cmd_line_script.py�_make_test_scriptN s r � c C s t | |||||�}t�� |S )N)r r r )Zzip_dirZ zip_basenameZpkg_namer r �depthr r r r �_make_test_zip_pkgS s
r c @ s� e Zd Zd`dd�Zdd�dd�Zdd�dd�Zd d
� Zd d � Zej dadd��Z
dbdd�Z dcdd�Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#ej dBdC� �Z$dDdE� Z%dFdG� Z&dHdI� Z'dJdK� Z(dLdM� Z)dNdO� Z*dPdQ� Z+dRdS� Z,dTdU� Z-dVdW� Z.dXdY� Z/dZd[� Z0d\d]� Z1d^d_� Z2dS )d� CmdLineTestNc
C s t dkr td| � tt|�� | �|d� d| }
d| } d| } d| }
d| }| d krdt�� } d | }t dkr�td
� t| � t| � t|
� t|� | �|
�d �|� | �| �d �|� | �| �d �|� | �|
�d �|� | �|�d �|� | �|�d �|� d S ) Nr zOutput from test script %r:r z__loader__==%az __file__==%az__package__==%rzsys.argv[0]==%azsys.path[0]==%azcwd==%azExpected output:zutf-8)�verbose�print�repr� assertEqual�os�getcwd�assertIn�encode)�self� script_nameZ exit_code�data�
expected_file�expected_argv0�expected_path0�expected_package�expected_loaderZ expected_cwdZprinted_loaderZ printed_fileZprinted_packageZ
printed_argv0Z
printed_path0Z printed_cwdr r r �
_check_output[ s0 zCmdLineTest._check_output)�cwdc O s\ t |t�r|g}t�� ||t�}
t|
d|d�| ��\} } }
| �|| | |
||||||� d S )NF)�
__isolated�__cwd)�
isinstance�strr Z!optim_args_from_interpreter_flags� example_argsr r( )r �script_exec_argsr# r$ r% r& r' r) �cmd_line_switches�env_vars�run_args�rc�out�errr r r �
_check_scriptx s
zCmdLineTest._check_scriptc
O s~ t |t�r|f}nt|�}|| }t|d|d�|��\}}} tdkrhtd| � tt| �� td| � | �|�d�| � d S )NF)r* r+ r zOutput from test script %r:zExpected output: %rzutf-8) r, r- �tupler r r r r r )
r r/ Z expected_msgr) r0 r1 r2 r3 r4 r5 r r r �_check_import_error� s
z CmdLineTest._check_import_errorc C s2 t dd�\}}}ttjj��d�}| �||� d S )Nz-czprint(__loader__)zutf-8)r r r � machinery�BuiltinImporterr r )r r3 r4 r5 �expectedr r r �test_dash_c_loader� s z CmdLineTest.test_dash_c_loaderc C sN t � }z|j�d� |j�� W d t|�}X ttjj�� d�}| �
||� d S )Ns print(__loader__)
zutf-8) r �stdin�write�flushr
r r r9 r: r r )r �pr4 r; r r r �test_stdin_loader� s
z CmdLineTest.test_stdin_loaderFc c sv |r t ddtjd�}|j}nt ddtjd�}|j}z,x |�d�}|dkrJP |�� q6W |V W d t|� |� � X d S )Nz-ir )�bufsize�stderr� s >>> )
r �
subprocess�PIPErC ZSTDOUT�stdout�read�readliner
�close)r �separate_stderrr@ rC r" r r r �interactive_python� s
z CmdLineTest.interactive_pythonc C sF | � |��2}|j�d� |j�� | �d|j�� �� � W d Q R X d S )Ns
print('foo')
s foo)rL r= r> r? r rG rI �strip)r rK r@ r r r �check_repl_stdout_flush� s
z#CmdLineTest.check_repl_stdout_flushc C sp | � |��\}|j�d� |j�� |r,|jn|j}| �d|�� � | �d|�� � | �d|�� � W d Q R X d S )Ns 1/0
s
Traceback s File "<stdin>"s ZeroDivisionError)rL r= r> r? rC rG r rI )r rK r@ rC r r r �check_repl_stderr_flush� s
z#CmdLineTest.check_repl_stderr_flushc C s | � � d S )N)rN )r r r r �test_repl_stdout_flush� s z"CmdLineTest.test_repl_stdout_flushc C s | � d� d S )NT)rN )r r r r �&test_repl_stdout_flush_separate_stderr� s z2CmdLineTest.test_repl_stdout_flush_separate_stderrc C s | � � d S )N)rO )r r r r �test_repl_stderr_flush� s z"CmdLineTest.test_repl_stderr_flushc C s | � d� d S )NT)rO )r r r r �&test_repl_stderr_flush_separate_stderr� s z2CmdLineTest.test_repl_stderr_flush_separate_stderrc C s: t �� �(}t|d�}| �||||d tjj� W d Q R X d S )N�script)r �temp_dirr r6 r r9 �SourceFileLoader)r r
r! r r r �test_basic_script� s
z CmdLineTest.test_basic_scriptc C s\ t �� �J}t|d�}tj|dd� t�|� t �|�}| �||||d t j
j � W d Q R X d S )NrT T)�doraise) r rU r �
py_compile�compiler �remove�make_legacy_pycr6 r r9 �SourcelessFileLoader)r r
r! �pyc_filer r r �test_script_compiled� s
z CmdLineTest.test_script_compiledc C s: t �� �(}t|d�}| �||||dtjj� W d Q R X d S )N�__main__� )r rU r r6 r r9 rV )r r
r! r r r �test_directory� s
zCmdLineTest.test_directoryc C s\ t �� �J}t|d�}tj|dd� t�|� t �|�}| �||||dt j
j � W d Q R X d S )Nr` T)rX ra ) r rU r rY rZ r r[ r\ r6 r r9 r] )r r
r! r^ r r r �test_directory_compiled� s
z#CmdLineTest.test_directory_compiledc C s, t �� �}d| }| �||� W d Q R X d S )Nz"can't find '__main__' module in %r)r rU r8 )r r
�msgr r r �test_directory_error� s
z CmdLineTest.test_directory_errorc C sH t �� �6}t|d�}t|d|�\}}| �||||dtj� W d Q R X d S )Nr` �test_zipra )r rU r r r6 � zipimport� zipimporter)r r
r! �zip_name�run_namer r r � test_zipfile� s
zCmdLineTest.test_zipfilec C sV t �� �D}t|d�}tj|dd�}t|d|�\}}| �||||dtj� W d Q R X d S )Nr` T)rX rf ra ) r rU r rY rZ r r6 rg rh )r r
r! �
compiled_nameri rj r r r �test_zipfile_compiled s
z!CmdLineTest.test_zipfile_compiledc C sF t �� �4}t|d�}t|d|�\}}d| }| �||� W d Q R X d S )NZnot_mainrf z"can't find '__main__' module in %r)r rU r r r8 )r r
r! ri rj rd r r r �test_zipfile_error s
z CmdLineTest.test_zipfile_errorc
C sX t �� �F}tj�|d�}t|� t|d�}| jddg|||dtj j
|d� W d Q R X d S )N�test_pkgrT z-mztest_pkg.script)r) ) r rU r �path�joinr r r6 r r9 rV )r r
�pkg_dirr! r r r �test_module_in_package s
z"CmdLineTest.test_module_in_packagec C sJ t �� �8}t|ddd�\}}| jddg|||dtj||d� W d Q R X d S )Nrf ro rT z-mztest_pkg.script)�
PYTHONPATHr) )r rU r r6 rg rh )r r
ri rj r r r �!test_module_in_package_in_zipfile s
z-CmdLineTest.test_module_in_package_in_zipfilec C sN t �� �<}t|ddddd�\}}| jddg|||dtj||d � W d Q R X d S )
Nrf ro rT � )r z-mztest_pkg.test_pkg.scriptztest_pkg.test_pkg)rt r) )r rU r r6 rg rh )r r
ri rj r r r �$test_module_in_subpackage_in_zipfile& s
z0CmdLineTest.test_module_in_subpackage_in_zipfilec
C sX t �� �F}tj�|d�}t|� t|d�}| jddg|||dtj j
|d� W d Q R X d S )Nro r` z-m)r) ) r rU r rp rq r r r6 r r9 rV )r r
rr r! r r r � test_package. s
zCmdLineTest.test_packagec
C sz t �� �h}tj�|d�}t|� t|d�}tj|dd�}t� |� t �
|�}| j ddg|||dt j
j|d� W d Q R X d S )Nro r` T)rX z-m)r) )r rU r rp rq r r rY rZ r[ r\ r6 r r9 r] )r r
rr r! rl r^ r r r �test_package_compiled8 s
z!CmdLineTest.test_package_compiledc C sF t �� �4}tj�|d�}t|� d}| jddg||d� W d Q R X d S )Nro z7'test_pkg' is a package and cannot be directly executedz-m)r) )r rU r rp rq r r8 )r r
rr rd r r r �test_package_errorE s
z CmdLineTest.test_package_errorc C s\ t �� �J}tj�|d�}t|� tj�|d�}t|� d}| jddg||d� W d Q R X d S )Nro r` z^Cannot use package as __main__ module; 'test_pkg' is a package and cannot be directly executedz-m)r) )r rU r rp rq r r8 )r r
rr Zmain_dirrd r r r �test_package_recursionM s
z"CmdLineTest.test_package_recursionc C s� t �� ��}t j|d��� tj�|d�}t|d� t|d�}td t �ddi�\}}}t
d krft t |�� d
d }| �
|�d �|� | �||||||dtjj� W d Q R X W d Q R X d S )
N)rp ro z1import sys; print('init_argv0==%r' % sys.argv[0])rT �-m�test_pkg.scriptr* Fr zinit_argv0==%rzutf-8)r| r} )r rU �
change_cwdr rp rq r r r r. r r r r r r( r r9 rV )r r
rr r! r3 r4 r5 r; r r r �test_issue8202X s
zCmdLineTest.test_issue8202c C s� t �� ��}t j|d��h tdd��R}|�d� tdddd�\}}}tdkrVtt|�� d d
}| � |�
d �|� W d Q R X W d Q R X W d Q R X d S ) N)rp z-c�wr" z2import sys; print("sys.path[0]==%r" % sys.path[0])F)r* r zsys.path[0]==%rra zutf-8) r rU r~ �openr> r r r r r r )r r
�fr3 r4 r5 r; r r r �"test_issue8202_dash_c_file_ignoredi s
z.CmdLineTest.test_issue8202_dash_c_file_ignoredc
C s� t �� �|}t|d�}t j|d��Z tdd��D}|�d� td t�ddi�\}}}| �||||||dt j
j � W d Q R X W d Q R X W d Q R X d S )
N�other)rp �-mr� r" r* Fra )r� r� ) r rU r r~ r� r> r r. r( r r9 rV )r r
r! r� r3 r4 r5 r r r �"test_issue8202_dash_m_file_ignoredx s
z.CmdLineTest.test_issue8202_dash_m_file_ignoredc
C s� t �� ��}tj�|d�}t|ddd��@}|�d� |�d� xtd�D ]}|�d� qFW |�d� W d Q R X t j|d �� t |�\}}}W d Q R X | �
d
|� | �
d
|� W d Q R X d S ) Nz
issue20884.pyr� �
)�newlinez#coding: iso-8859-1
z"""
� zQxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
)rp � ) r rU r rp rq r� r> �ranger~ r r )r r
r! r� �_r3 r4 r5 r r r �test_issue20884� s
zCmdLineTest.test_issue20884c
g sR t �� �@}t j|d��( tj�|d�}t|f|�� |V W d Q R X W d Q R X d S )N)rp ro )r rU r~ r rp rq r )r �argsr
rr r r r �setup_test_pkg� s
zCmdLineTest.setup_test_pkgc G s< t d|�ddi�\}}}tdkr,tt|�� | �|d� |S )N�-mr* Fr )r� )r r r r r )r r� r3 r4 r5 r r r �check_dash_m_failure� s
z CmdLineTest.check_dash_m_failurec C s> | � � �,}t|dd�}| jdt�� }| �d|� W d Q R X d S )Nr� z+if __name__ == '__main__': raise ValueError�test_pkg.others
ValueError)r� )r� r r� r. r )r rr r! r5 r r r � test_dash_m_error_code_is_one� s
z)CmdLineTest.test_dash_m_error_code_is_onec
C sd d}xZ|D ]R\}}| � |��: td|�\}}}| �|d� | �||� | �d|� W d Q R X q
W d S )N))�builtinss No code object available)z
builtins.xs= Error while finding module specification.*ModuleNotFoundError)z builtins.x.ys] Error while finding module specification.*ModuleNotFoundError.*No module named.*not a package)zos.paths loader.*cannot handle)r s= No module named.*is a package and cannot be directly executed)zimportlib.nonexistants No module named)z .unittests# Relative module names not supportedz-mr s Traceback)�subTestr r � assertRegex� assertNotIn)r Ztests�nameZregexr3 r� r5 r r r �test_dash_m_errors� s z CmdLineTest.test_dash_m_errorsc C s~ t �� �l}t j|d��T t�d� tdd�� W d Q R X | �d�}| �|d� | �d|� | �d|� W d Q R X W d Q R X d S )N)rp ro ztest_pkg/__init__.pyc�wbsG Error while finding module specification.*ImportError.*bad magic numbers is a packages Traceback) r rU r~ r �mkdirr� r� r� r� )r r
r5 r r r �test_dash_m_bad_pyc� s
z CmdLineTest.test_dash_m_bad_pycc C s� t tttf}x||D ]t}|j}d�|�}| �|��P | �|��:}| �d�}| � |�
d�|� | � d|� | � d|� W d Q R X W d Q R X qW d S )Nz%raise {0}('Exception in __init__.py')ro �asciis Exception in __init__.pys Traceback) � ImportError�AttributeError� TypeError�
ValueError�__name__�formatr� r� r� r r )r �
exceptionsZ exceptionZinitrr r5 r r r �test_dash_m_init_traceback� s
z&CmdLineTest.test_dash_m_init_tracebackc C sV | � � �D}d}t|d|� | �d�}| �d|� | �d|� | �d|� W d Q R X d S )Nz1raise ImportError('Exception in __main__ module')r` ro s ImportErrors Exception in __main__ modules Traceback)r� r r� r )r rr �mainr5 r r r �test_dash_m_main_traceback� s
z&CmdLineTest.test_dash_m_main_tracebackc C s� t �d�}t�� �|}t|d|�}t|�\}}}|�d��d�}| �t |�d� | �
|d � d�� | �
|d � d �� | �
|d
� d �� W d Q R X d S ) Nz| try:
raise ValueError
except:
raise NameError from None
rT r� r� rD r Z Tracebackr z File � � NameError) �textwrap�dedentr rU r r �decode�splitr �lenZ
assertTrue�
startswith)r rT r
r! �exitcoderG rC �textr r r �test_pep_409_verbiage� s
z!CmdLineTest.test_pep_409_verbiagec C s� t jr tjdkr t�t j�}nt jr,t j}n
| �d� d}ttj ||�}| �
t j |� t |�\}}}| �
t|�|�� �d�d||f � | �
d|� d S )N)�win32�darwinz need support.TESTFN_NONASCIIzprint(ascii(__file__))
r� zstdout=%r stderr=%rr )r ZTESTFN_UNDECODABLE�sys�platformr �fsdecodeZTESTFN_NONASCIIZskipTestr �curdirZ
addCleanup�unlinkr r r� �rstripr� )r r� r r! r3 rG rC r r r �test_non_ascii� s
zCmdLineTest.test_non_asciic C sR t �d�}t�� �6}t|d|�}t|�\}}}|�d�}| �|d� W d Q R X d S )Nz� import sys
error = None
try:
raise ValueError('some text')
except ValueError as err:
error = err
if error:
sys.exit(error)
rT r� z some text)r� r� r rU r r r� r )r rT r
r! r� rG rC r� r r r �)test_issue20500_exit_with_exception_value s
z5CmdLineTest.test_issue20500_exit_with_exception_valuec C sX d}t �� �B}t|d|�}t|�\}}}t�t�|�d��� }| �d|� W d Q R X d S )Nz
1 + 1 = 2
rT r� z
1 + 1 = 2
^) r rU r r �io�
TextIOWrapper�BytesIOrH r )r rT r
r! r� rG rC r� r r r �*test_syntaxerror_unindented_caret_position& s
z6CmdLineTest.test_syntaxerror_unindented_caret_positionc C s� t �d�}t�� ��}t|d|�}t|�\}}}t�t�|�d�� � }| �
d|� d}t|d|�}t|�\}}}t�t�|�d�� � }| � d|� | �
d|� W d Q R X d S )Nz; if True:
1 + 1 = 2
rT r� z
1 + 1 = 2
^zif True:
1 + 1 = 2
� ) r� r� r rU r r r� r� r� rH r r� )r rT r
r! r� rG rC r� r r r �(test_syntaxerror_indented_caret_position/ s
z4CmdLineTest.test_syntaxerror_indented_caret_positionc
C s� t �d�}d | _t�� ��}t�� ��}t|d|�}td||d�}t|��� � � }| �
|d |� | � ||� td||d�}t|��� � � }| �
||� td||d�}t|��� � � }| �
|||� W d Q R X W d Q R X d S )Nzc import sys
for entry in sys.path:
print(entry)
r` z-Es)r) r z-I) r� r� �maxDiffr rU r r r
r� �
splitlinesr r� ) r rT �work_dirr
r! r@ Z out_by_nameZ
out_by_dirZout_by_dir_isolatedr r r �-test_consistent_sys_path_for_direct_executionF s z9CmdLineTest.test_consistent_sys_path_for_direct_executionc C s� t �d�}d | _t�� ��}tj�|d�}t�|� t |d|�}t
dd|d�}t |�� � �
� }| �|d |� | �||� t
dd|d�}t |�� � �
� }| �||� tdd|d�\}} }
|
� � �
� } | �d | d
� W d Q R X d S ) Nzc import sys
for entry in sys.path:
print(entry)
Z
script_pkgr` z-smzscript_pkg.__main__)r) r z-ImzNo module named script_pkg���)r� r� r� r rU r rp rq r� r r r
r� r� r r� r r ) r rT r� r
r! r@ Z
out_by_moduleZout_by_packager� rG rC Ztraceback_linesr r r �-test_consistent_sys_path_for_module_executione s$
z9CmdLineTest.test_consistent_sys_path_for_module_executionc C sT d}| � tj�|�� t|dtjtjd�}|�� \}}| �d|� | � |j
d� d S )Nznonexistingscript.pyT)r� rG rC z: can't open file r ) Z assertFalser rp �existsr rE rF Z communicater ZassertNotEqual�
returncode)r rT �procr4 r5 r r r �test_nonexisting_script� s
z#CmdLineTest.test_nonexisting_script)N)F)F)F)3r� �
__module__� __qualname__r( r6 r8 r< rA �
contextlib�contextmanagerrL rN rO rP rQ rR rS rW r_ rb rc re rk rm rn rs ru rw rx ry rz r{ r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r Z s^
%r c C s t �t� t �� d S )N)r Z run_unittestr Z
reap_childrenr r r r � test_main� s
r� r` ) r� r Zimportlib.machineryrg Zunittestr� r Zos.pathrY rE r� r� Ztestr Ztest.support.script_helperr r r r r r r r
r r. Z test_sourcer r ZTestCaser r� r� r r r r �<module> s4 (
5 D