File: //opt/cppython/lib/python3.8/test/__pycache__/test_cmd_line_script.cpython-38.opt-1.pyc
U
.D�g�} � @ 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 �8/opt/cppython/lib/python3.8/test/test_cmd_line_script.py�_make_test_scriptN s r � c C s t | |||||�}t�� |S r )r r r
)Zzip_dirZ zip_basenameZpkg_namer r �depthr r r r �_make_test_zip_pkgS s
�r c @ s� e Zd Zdhdd�Zdd�dd�Zdd�dd�Zd d
� Zd d � Zej didd��Z
djdd�Z dkdd�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#dBdC� Z$dDdE� Z%ej 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_� Z2d`da� Z3dbdc� Z4ddde� Z5dfdg� Z6dS )l� 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:�utf-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��
__isolatedZ__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�|��\}}} tdkrltd|� ��� tt| �� td| � | �|�d�| � d S )NFr- r zOutput from test script zExpected output: %rr ) r/ r0 �tupler r r r r r! )
r" r2 Z expected_msgr, r3 r4 r5 r6 r7 r8 r r r �_check_import_error� s
�� z CmdLineTest._check_import_errorc C s2 t dd�\}}}ttjj��d�}| �||� d S )N�-czprint(__loader__)r )r r r � machinery�BuiltinImporterr! r )r" r6 r7 r8 �expectedr r r �test_dash_c_loader� s z CmdLineTest.test_dash_c_loaderc C sN t � }z|j�d� |j�� W 5 t|�}X ttjj�� d�}| �
||� d S )Ns print(__loader__)
r ) r r
�stdin�write�flushr r r= r> r! r )r" �pr7 r? r r r �test_stdin_loader� s
z CmdLineTest.test_stdin_loaderFc c sn |rt dtjd�}|j}nt dtjd�}|j}z(|�d�}|dkrDqN|� � q0|V W 5 t|� |�� X d S )Nz-i)�stderr� s >>> )
r �
subprocess�PIPErF �STDOUT�stdoutr
�close�read�readline)r" �separate_stderrrD rF r$ r r r �interactive_python� s
z CmdLineTest.interactive_pythonc C sF | � |��2}|j�d� |j�� | �d|j�� �� � W 5 Q R X d S )Ns
print('foo')
s foo)rP rA rB rC r rK rN �strip)r" rO rD 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 5 Q R X d S )Ns 1/0
s
Traceback s File "<stdin>"s ZeroDivisionError)rP rA rB rC rF rK r rN )r" rO rD rF r r r �check_repl_stderr_flush� s
z#CmdLineTest.check_repl_stderr_flushc C s | � � d S r �rR �r" r r r �test_repl_stdout_flush� s z"CmdLineTest.test_repl_stdout_flushc C s | � d� d S �NTrT rU r r r �&test_repl_stdout_flush_separate_stderr� s z2CmdLineTest.test_repl_stdout_flush_separate_stderrc C s | � � d S r �rS rU r r r �test_repl_stderr_flush� s z"CmdLineTest.test_repl_stderr_flushc C s | � d� d S rW rY rU 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 5 Q R X d S )N�script�r �temp_dirr r9 r r= �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 5 Q R X d S )Nr\ T��doraise� r r^ r �
py_compile�compiler �remove�make_legacy_pycr9 r r= �SourcelessFileLoader�r" r r# �pyc_filer r r �test_script_compiled� s
�z CmdLineTest.test_script_compiledc C s: t �� �(}t|d�}| �||||dtjj� W 5 Q R X d S )N�__main__� 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 5 Q R X d S )Nrm Trb rn rd rj r r r �test_directory_compiled� s
�z#CmdLineTest.test_directory_compiledc C s, t �� �}d| }| �||� W 5 Q R X d S )N�"can't find '__main__' module in %r)r r^ r; )r" r �msgr r r �test_directory_error� s
z CmdLineTest.test_directory_errorc C sH t �� �6}t|d�}t|d|�\}}| �||||dtj� W 5 Q R X d S )Nrm �test_ziprn )r r^ r r r9 � zipimport� zipimporter)r" r r# �zip_name�run_namer r r � test_zipfile� s
�zCmdLineTest.test_zipfilec C s\ t �� �J}t|d�}tj|dtjjd�}t|d|�\}}| �||||dt j
� W 5 Q R X d S �Nrm T)rc �invalidation_modert rn ) r r^ r re rf �PycInvalidationMode� TIMESTAMPr r9 ru rv �r" r r# �
compiled_namerw rx r r r � test_zipfile_compiled_timestamp s
��z+CmdLineTest.test_zipfile_compiled_timestampc C s\ t �� �J}t|d�}tj|dtjjd�}t|d|�\}}| �||||dt j
� W 5 Q R X d S rz ) r r^ r re rf r| � CHECKED_HASHr r9 ru rv r~ r r r �"test_zipfile_compiled_checked_hash s
��z.CmdLineTest.test_zipfile_compiled_checked_hashc C s\ t �� �J}t|d�}tj|dtjjd�}t|d|�\}}| �||||dt j
� W 5 Q R X d S rz ) r r^ r re rf r| �UNCHECKED_HASHr r9 ru rv r~ r r r �$test_zipfile_compiled_unchecked_hash s
��z0CmdLineTest.test_zipfile_compiled_unchecked_hashc C sF t �� �4}t|d�}t|d|�\}}d| }| �||� W 5 Q R X d S )NZnot_mainrt rq )r r^ r r r; )r" r r# rw rx rr 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 5 Q R X d S )N�test_pkgr\ �-m�test_pkg.scriptr+ � r r^ r �path�joinr r r9 r r= r_ �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 5 Q R X d S )Nrt r� r\ r� r� �Z
PYTHONPATHr, �r r^ r r9 ru rv �r" r rw rx r r r �!test_module_in_package_in_zipfile5 s
�z-CmdLineTest.test_module_in_package_in_zipfilec C sN t �� �<}t|ddddd�\}}| jddg|||dtj||d � W 5 Q R X d S )
Nrt r� r\ � )r r� ztest_pkg.test_pkg.scriptztest_pkg.test_pkgr� r� r� 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 5 Q R X d S )Nr� rm r� r+ r� r� r r r � test_packageD s
�zCmdLineTest.test_packagec
C sz t �� �h}tj�|d�}t|� t|d�}tj|dd�}t� |� t �
|�}| j ddg|||dt j
j|d� W 5 Q R X d S )Nr� rm Trb r� r+ )r r^ r r� r� r r re rf rg rh r9 r r= ri )r" r r� r# r rk r r r �test_package_compiledN s
�z!CmdLineTest.test_package_compiledc C sF t �� �4}tj�|d�}t|� d}| jddg||d� W 5 Q R X d S )Nr� z7'test_pkg' is a package and cannot be directly executedr� r+ �r r^ r r� r� r r; )r" r r� rr r r r �test_package_error[ s
z CmdLineTest.test_package_errorc C s\ t �� �J}tj�|d�}t|� tj�|d�}t|� d}| jddg||d� W 5 Q R X d S )Nr� rm z^Cannot use package as __main__ module; 'test_pkg' is a package and cannot be directly executedr� r+ r� )r" r r� Zmain_dirrr r r r �test_package_recursionc 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 5 Q R X W 5 Q R X d S )
N�r� r� z1import sys; print('init_argv0==%r' % sys.argv[0])r\ r� r� r. Fr zinit_argv0==%rr )r� r� )r r^ �
change_cwdr r� r� r r r r1 r r r r r! r* r r= r_ )r" r r� r# r6 r7 r8 r? r r r �test_issue8202n s"
�zCmdLineTest.test_issue8202c C s� t �� ��}t j|d��h tdd��R}|�d� tdddd�\}}}tdkrVtt|�� d d
}| � |�
d �|� W 5 Q R X W 5 Q R X W 5 Q R X d S ) Nr� r< �wr$ z2import sys; print("sys.path[0]==%r" % sys.path[0])F)r. r zsys.path[0]==%rrn r ) r r^ r� �openrB r r r r r r! )r" r �fr6 r7 r8 r? r r r �"test_issue8202_dash_c_file_ignored 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 5 Q R X W 5 Q R X W 5 Q R X d S )
N�otherr� r� r� r$ r. Frn )r� r� ) r r^ r r� r� rB r r1 r* r r= r_ )r" r r# r� r6 r7 r8 r r r �"test_issue8202_dash_m_file_ignored� s
�
�z.CmdLineTest.test_issue8202_dash_m_file_ignoredc
C s� t �� ��}tj�|d�}t|ddd��<}|�d� |�d� td�D ]}|�d� qD|�d� W 5 Q R X t j|d �� t |�\}}}W 5 Q R X | �
d
|� | �
d
|� W 5 Q R X d S ) Nz
issue20884.pyr� �
)�newlinez#coding: iso-8859-1
z"""
� zQxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
r� � ) r r^ r r� r� r� rB �ranger� r r )r" r r# r� �_r6 r7 r8 r r r �test_issue20884� s
zCmdLineTest.test_issue20884c
g sR t �� �@}t j|d��( tj�|d�}t|f|�� |V W 5 Q R X W 5 Q R X d S )Nr� r� )r r^ r� r r� r� r )r" �argsr r� r r r �setup_test_pkg� s
�zCmdLineTest.setup_test_pkgc G s< t d|�ddi�\}}}tdkr,tt|�� | �|d� |S )Nr� r. Fr )r� )r r r r r )r" r� r6 r7 r8 r r r �check_dash_m_failure� s
z CmdLineTest.check_dash_m_failurec C s> | � � �,}t|dd�}| jdt�� }| �d|� W 5 Q R X d S )Nr� z+if __name__ == '__main__': raise ValueError�test_pkg.others
ValueError)r� )r� r r� r1 r )r" r� r# r8 r r r � test_dash_m_error_code_is_one� s
�z)CmdLineTest.test_dash_m_error_code_is_onec
C s` d}|D ]R\}}| � |��: td|�\}}}| �|d� | �||� | �d|� W 5 Q R X qd 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.nonexistents No module named)z .unittests# Relative module names not supportedr� r � Traceback)�subTestr r � assertRegex� assertNotIn)r" Ztests�nameZregexr6 r� r8 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 5 Q R X | �d�}| �|d� | �d|� | �d|� W 5 Q R X W 5 Q R X d S )Nr� r� ztest_pkg/__init__.pyc�wbsG Error while finding module specification.*ImportError.*bad magic numbers is a packager� ) r r^ r� r �mkdirr� r� r� r� )r" r r8 r r r �test_dash_m_bad_pyc� s
�
� z CmdLineTest.test_dash_m_bad_pycc C s� t tttf}|D ]t}|j}d�|�}| �|��P | �|��:}| �d�}| � |�
d�|� | � d|� | � d|� W 5 Q R X W 5 Q R X qd S )Nz%raise {0}('Exception in __init__.py')r� �asciis Exception in __init__.pyr� ) � ImportError�AttributeError� TypeError�
ValueError�__name__�formatr� r� r� r r! )r" �
exceptions� exceptionZinitr� r8 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 5 Q R X d S )Nz1raise ImportError('Exception in __main__ module')rm r� s ImportErrors Exception in __main__ moduler� )r� r r� r )r" r� �mainr8 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 5 Q R X d S ) Nz| try:
raise ValueError
except:
raise NameError from None
r\ r� r� rG r Z Tracebackr z File � � NameError) �textwrap�dedentr r^ r r �decode�splitr �lenZ
assertTrue�
startswith�r" r\ r r# �exitcoderK rF �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# r6 rK rF r r r �test_non_ascii s" �
�zCmdLineTest.test_non_asciic C sR t �d�}t�� �6}t|d|�}t|�\}}}|�d�}| �|d� W 5 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)
r\ r� z some text)r� r� r r^ r r r� r 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 5 Q R X d S )Nz
1 + 1 = 2
r\ r� �
1 + 1 = 2
^) r r^ r r �io�
TextIOWrapper�BytesIOrM r 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 5 Q R X d S )Nz; if True:
1 + 1 = 2
r\ r� r� zif True:
1 + 1 = 2
� ) r� r� r r^ r r r� r� r� rM r r� r� r r r �(test_syntaxerror_indented_caret_positionE s
� z4CmdLineTest.test_syntaxerror_indented_caret_positionc C sT d}t �� �>}t|d|�}t|�\}}}| �|�� dd � dddg� W 5 Q R X d S )Nzfoo = f"""{}
foo"""
r\ ���s foo = f"""{}� ^s3 SyntaxError: f-string: empty expression not allowed�r r^ r r r �
splitlines�r" r\ r r# r� rK rF r r r �#test_syntaxerror_multi_line_fstring\ s
��z/CmdLineTest.test_syntaxerror_multi_line_fstringc C sV d}t �� �@}t|d|�}td|�\}}}| �|�� dd � dddg� W 5 Q R X d S )Nzfoo = """\q
"""
r\ z-Werrorr� s foo = """\qr� s' SyntaxError: invalid escape sequence \qr� r� r r r �3test_syntaxerror_invalid_escape_sequence_multi_linej s
�
��z?CmdLineTest.test_syntaxerror_invalid_escape_sequence_multi_linec
C s� t �d�}d | _t�� ��}t�� ��}t|d|�}td||d�}t|��� � � }| �
|d |� | � ||� td||d�}t|��� � � }| �
||� td||d�}t|��� � � }| �
|||� W 5 Q R X W 5 Q R X d S )N�c import sys
for entry in sys.path:
print(entry)
rm z-Esr+ r z-I) r� r� �maxDiffr r^ r r r
r� r� r r� ) r" r\ �work_dirr r# rD Z out_by_nameZ
out_by_dirZout_by_dir_isolatedr r r �-test_consistent_sys_path_for_direct_executionz 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 5 Q R X d S ) Nr� Z
script_pkgrm z-smzscript_pkg.__main__r+ r z-ImzNo module named script_pkg���)r� r� r� r r^ r r� r� r� r r r
r� r� r r� r r ) r" r\ r� r r# rD Z
out_by_moduleZout_by_packager� rK rF Ztraceback_linesr r r �-test_consistent_sys_path_for_module_execution� s(
� z9CmdLineTest.test_consistent_sys_path_for_module_executionc C sT d}| � tj�|�� t|dtjtjd�}|�� \}}| �d|� | � |j
d� d S )Nznonexistingscript.pyT)r� rK rF z: can't open file r ) Z assertFalser r� �existsr rH rI � communicater ZassertNotEqual�
returncode)r" r\ �procr7 r8 r r r �test_nonexisting_script� s � z#CmdLineTest.test_nonexisting_script)N)F)F)F)7r� �
__module__� __qualname__r* r9 r; r@ rE �
contextlib�contextmanagerrP rR rS rV rX rZ r[ ra rl ro rp rs ry 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� r� r� r� r� r� r� r� r r r r r Z sn �
� �
%r c C s t �t� t �� d S r )r Z run_unittestr �
reap_childrenr r r r � test_main� s
r� rm ) r� r Zimportlib.machineryru Zunittestr� r Zos.pathre rH r� r� �testr Ztest.support.script_helperr r r r r r r r
r r1 Z test_sourcer r ZTestCaser r� r� r r r r �<module> s8 (
5 �
x