shell bypass 403
o
a b�O � @ s� d dl Z d dlZd dlmZ d dlmZmZmZ d dlm Z m
Z
d dl Z d dl Z d dl
Z
d dl
mZ d dlmZ dd� Zd d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Z d d � Z d!d"� Z G d#d$� d$e �Z G d%d&� d&e j!�Z"G d'd(� d(e j!�Z#G d)d*� d*e j!�Z$G d+d,� d,e j!�Z%G d-d.� d.e j!�Z&G d/d0� d0e j!�Z'G d1d2� d2e j!�Z(G d3d4� d4e j!�Z)e*d5kr�e �+� dS dS )6� N)�captured_stdout)�TESTFN�rmtree�unlink��assert_python_ok�assert_python_failure)�Trace)�testmodc C s | � d�r | dd� } | S )z8Given a .pyc filename converts it to the appropriate .pyz.pycN���)�endswith)�filename� r �8/usr/local/python-3.10/lib/python3.10/test/test_trace.py�
fix_ext_py s
r c C s$ t j�t j�t��d } tt�| fS )z4The .py file and module name of this file (__file__)r )�os�path�splitext�basename�__file__r )�modnamer r r �my_file_and_modname s r c C s | j jS �N)�__code__�co_firstlineno��funcr r r �get_firstlineno � r c C s | }|}|| }|S r r )�x�y�a�b�cr r r �traced_func_linear% s r$ c C s | }t d�D ]}||7 }q|S �N� ��range)r r r# �ir r r �traced_func_loop+ s
r* c C s | | t �d� S )N� )r
r )r r r r r �traced_func_importing1 s r, c C s t | | �}|| S r �r$ )r r# r r r �traced_func_simple_caller4 s
r. c C s t | �}|t|| �7 }|S r )r. r, )r �kr r r � traced_func_importing_caller8 s r0 c c s" � d}t | �D ]}|| V qd S r% r' )�numr# r) r r r �traced_func_generator= s
� �r2 c C s d} t d�D ]}| |7 } qd S )Nr �
)r2 )r/ r) r r r � traced_func_calling_generatorB s
�r4 c C � | d S �N� r )r1 r r r �traced_doublerG r r8 c O s | |fS r r )�args�kwargsr r r �traced_capturerJ r r; c C s d} dd� t | �D �}|S )Nr3 c S s g | ]}t |��qS r )r8 )�.0r) r r r �
<listcomp>O s z4traced_caller_list_comprehension.<locals>.<listcomp>r' )r/ Zmylistr r r � traced_caller_list_comprehensionM s r> c C s, dd� } dd� }| |� dd� ��}|� d S )Nc S � | S r r ��fr r r �
decorator1S � z-traced_decorated_function.<locals>.decorator1c S s dd� } | S )Nc S r? r r r@ r r r �
decorator2V rC zGtraced_decorated_function.<locals>.decorator_fabric.<locals>.decorator2r �rD r r r �decorator_fabricU s z3traced_decorated_function.<locals>.decorator_fabricc S s d S r r r r r r r Y s z'traced_decorated_function.<locals>.funcr )rB rF r r r r �traced_decorated_functionR s
rG c @ s< e Zd Zdd� Zdd� Zdd� Zedd� �Zed d
� �Z d S ) � TracedClassc C s
|| _ d S r �r! )�selfr r r r �__init__a �
zTracedClass.__init__c C s
| j | S r rI )rJ r r r r �inst_method_lineard rL � TracedClass.inst_method_linearc C s | � |�}|t||� S r )rM r$ )rJ r r# r r r �inst_method_callingg �
� TracedClass.inst_method_callingc C s |d S r6 r )�clsr r r r �class_method_lineark � z TracedClass.class_method_linearc C r5 r6 r )r r r r �static_method_linearo rT z TracedClass.static_method_linearN)
�__name__�
__module__� __qualname__rK rM rO � classmethodrS � staticmethodrU r r r r rH ` s
rH c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�TestLineCountsz/White-box testing of line-counting, via runfuncc C s2 | � tjt�� � tddddd�| _tt�| _d S )Nr+ r ��count�trace�
countfuncs� countcallers) �
addCleanup�sys�settrace�gettracer �tracerr r �my_py_filename�rJ r r r �setUpy s zTestLineCounts.setUpc C sb | j �tdd�}| �|d� i }tt�}tdd�D ] }d|| j|| f< q| �| j �� j|� d S )Nr7 r& � r+ ) re �runfuncr$ � assertEqualr r( rf �results�counts)rJ �result�expected� firstlinenor) r r r �test_traced_func_linear~ s z&TestLineCounts.test_traced_func_linearc C sl | j �tdd� tt�}| j|d fd| j|d fd| j|d fd| j|d fdi}| �| j �� j|� d S )Nr7 � r+ � r& � )re rj r* r rf rk rl rm �rJ rp ro r r r �test_traced_func_loop� s �z$TestLineCounts.test_traced_func_loopc C s^ | j �tdd� tt�}| j|d fdttj�dfdttj�dfdi}| �| j � � j
|� d S )Nr7 r& r+ rr ) re rj r, r rf r r
r rk rl rm ru r r r �test_traced_func_importing� s �z)TestLineCounts.test_traced_func_importingc
C s� | j �t� tt�}tt�}| j|d fd| j|d fd| j|d fd| j|d fd| j|d fd| j|d fdi}| �| j �� j|� d S )Nr+ r7 � rr r3 ) re rj r4 r r2 rf rk rl rm )rJ �firstlineno_callingZfirstlineno_genro r r r �test_trace_func_generator� s �z(TestLineCounts.test_trace_func_generatorc C sp | j �t� tt�}tt�}| j|d fd| j|d fd| j|d fd| j|d fdi}| �| j �� j|� d S )Nr+ r7 � rr r3 ) re rj r> r r8 rf rk rl rm )rJ ry Zfirstlineno_calledro r r r � test_trace_list_comprehension� s �z,TestLineCounts.test_trace_list_comprehensionc C s� | j �t� tt�}| j|d fd| j|d fd| j|d fd| j|d fd| j|d fd| j|d fd| j|d fd| j|d fd| j|d fd| j|d
fd| j|d fdi }| �| j �� j|� d S ) Nr+ r7 rr rt r& rs ri � � r3 rx )re rj rG r rf rk rl rm ru r r r � test_traced_decorated_function� s �
z-TestLineCounts.test_traced_decorated_functionc C sh dD ]/}t ddddd�}td�}t||�}|�|d� t|�}| j|d fdi}| �|�� j|� qd S )N)rM r+ r r\ � � ) r rH �getattrrj r rf rk rl rm )rJ Zmethnamere Z
traced_obj�methodrp ro r r r �test_linear_methods� s
��z"TestLineCounts.test_linear_methodsN) rV rW rX �__doc__rh rq rv rw rz r| r r� r r r r r[ w s r[ c @ s e Zd ZdZdd� Zdd� ZdS )�TestRunExecCountsz8A simple sanity test of line-counting, via runctx (exec)c C s t t�| _| �tjt�� � d S r )r r rf ra rb rc rd rg r r r rh � s
zTestRunExecCounts.setUpc C s� t ddddd�| _d}t|td�}| j�|t� t� � tt�}| j |d fd| j |d fd| j |d fd | j |d
fdi}|�
� D ]}| � | j� � j
| || � qAd S ) Nr+ r r\ ztraced_func_loop(2, 5)�execr7 rs rr r& rt )r re �compiler Zrunctx�globals�varsr r* rf �keysrk rl rm )rJ �coderp ro r/ r r r �test_exec_counts� s � �z"TestRunExecCounts.test_exec_countsN)rV rW rX r� rh r� r r r r r� � s r� c @ sb e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Ze �
e e d �o$e �
� d
�dd� �Zdd� ZdS )� TestFuncsz"White-box testing of funcs tracingc C s8 | � tjt�� � tdddd�| _t� | _t�� | _d S )Nr r+ )r] r^ r_ ) ra rb rc rd r re r �filemod�_saved_tracefuncrg r r r rh s zTestFuncs.setUpc C s | j d ur
t�| j � d S d S r )r� rb rc rg r r r �tearDown s
�zTestFuncs.tearDownc C s> | j �td� | jd d| jd di}| �| j �� j|� d S )Nr+ �r. r- )re rj r. r� rk rl � calledfuncs�rJ ro r r r �test_simple_caller s
�z TestFuncs.test_simple_callerc C s� | j jtddddd�}| �|dddd�f� | �t�� | j jtdd� W d � n1 s.w Y | �t�� | j �� W d � d S 1 sIw Y d S )Nr+ r7 rr rt )rJ r )r+ r7 )r �arg)re rj r; rk Z assertRaises� TypeError)rJ �resr r r �test_arg_errors s � "�zTestFuncs.test_arg_errorsc C sb | j �td� | jd d| jd d| jd d| jd dttj�ddfdi}| �| j �� j |� d S )Nr+ r� r- �r0 �r, r
r )
re rj r0 r� r r
r rk rl r� r� r r r �test_loop_caller_importing s
�z$TestFuncs.test_loop_caller_importingrd �3pre-existing trace function throws off measurementsc C sR t d�}| j�|jd� | jd d| jd d| jd di}| �| j�� j|� d S )Nr� r+ )rQ )rN r- )rH re rj rO r� rk rl r� )rJ �objro r r r �test_inst_method_calling) s
�z"TestFuncs.test_inst_method_callingc
C sZ | j �t� | jd d| jd d| jd d| jd d| jd di}| �| j �� j|� d S )N)rG r+ )rF rE )rB r )re rj rG r� rk rl r� r� r r r r 6 s
�z(TestFuncs.test_traced_decorated_functionN)rV rW rX r� rh r� r� r� r� �unittest�skipIf�hasattrrb rd r� r r r r r r� s �
r� c @ s: e Zd ZdZdd� Ze�eed�oe� � d�dd� �Z
dS ) � TestCallersz$White-box testing of callers tracingc C s. | � tjt�� � tdddd�| _t� | _d S )Nr r+ )r] r^ r` )ra rb rc rd r re r r� rg r r r rh E s zTestCallers.setUprd r� c C s� | j �td� tj�tj�d d ddf| jd fd| jd | jd fd| jd | jd fd| jd | jd fd| jd t t
j�d
d ffdi}| � | j � � j
|� d S ) Nr+ r �.pyr^ z
Trace.runfuncr� r� r- r� r
r )re rj r0 r r r r^ r r� r r
rk rl Zcallersr� r r r r� J s. ������ z&TestCallers.test_loop_caller_importingN) rV rW rX r� rh r� r� r� rb rd r� r r r r r� C s �r� c @ s@ e Zd Zdd� Zdd� Z ddd�Zdd � Zd
d � Zd d
� ZdS )� TestCoveragec C s | � tjt�� � d S r )ra rb rc rd rg r r r rh ` s zTestCoverage.setUpc C s t t� tt� d S r )r r r rg r r r r� c s zTestCoverage.tearDown�_import test.support, test.test_pprint;test.support.run_unittest(test.test_pprint.QueryTestCase)c C s& |� |� |�� }|jddtd� d S )NT)Z show_missingZsummaryZcoverdir)�runrl Z
write_resultsr )rJ re �cmd�rr r r � _coverageg s
zTestCoverage._coveragec C s� t jddd�}t� �
}| �|� W d � n1 sw Y |�� }| �d|� | �d|� t�t�}| �d|� | �d|� d S )Nr r+ �r^ r] z pprint.pyzcase.pyz pprint.coverzunittest.case.cover) r^ r r r� �getvalue�assertInr �listdirr )rJ re �stdout�filesr r r �
test_coveragen s �
zTestCoverage.test_coveragec C s� t j�t j�t j��}tjtjtj |gddd�}t
� �
}| � |� W d � n1 s+w Y t j� t
�rDt �t
�}| �|dg� d S d S )Nr r+ )Z
ignoredirsr^ r] z_importlib.cover)r r �normpath�dirnamer r^ r rb � base_prefix�base_exec_prefixr r� �existsr r� rk )rJ Zlibpathre r� r� r r r �test_coverage_ignorey s �� �
�z!TestCoverage.test_coverage_ignorec
C s� t jddd�}d}|tjv rtj|= d}t� �}| �||� W d � n1 s(w Y |�d� |�� i }|D ]}|�� d d� \}}} t |�t |d d� �f|| < q:t �
tj| j �}| � ||� | �
|| d� d S ) Nr r+ r� ztest.tracedmodules.testmodz<import test.tracedmodules.testmod as t;t.func(0); t.func2();rr r )r& �d )r^ r rb �modulesr r� �seek�readline�split�intZ _fullmodnamer r� rk )
rJ re r r� r� Zcoverage�line�linesZcov�moduler r r �test_issue9936� s"
�
zTestCoverage.test_issue9936N)r� ) rV rW rX rh r� r� r� r� r� r r r r r� _ s
� r� c @ s e Zd Zdd� ZdS )� Test_Ignorec C s� t jj}t�ddg|dd�g�}| �|�dd�� | �|�dd�� | �|�dd �� | �|�|ddd
�d �� | �|�|dd �d
�� | �|�|dd
�d �� d S )Nr zy.zZfooZbarzx.pyzxy.pyZxyzy.pyr zbaz.pyZbazzz.py�z)r r �joinr^ Z_Ignore�
assertTrue�names� assertFalse)rJ Zjn�ignorer r r � test_ignored� s zTest_Ignore.test_ignoredN)rV rW rX r� r r r r r� � s r� c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) � TestCoverageCommandLineOutputztmp.pyz tmp.coverc C sF t | jddd��}|�t�d�� W d � d S 1 s w Y d S )N�w� iso-8859-15��encodingu� # coding: iso-8859-15
x = 'spœm'
if []:
print('unreachable')
)�open�codefile�write�textwrap�dedent)rJ rA r r r rh � s "�z#TestCoverageCommandLineOutput.setUpc C s t | j� t | j� d S r )r r� � coverfilerg r r r r� � rP z&TestCoverageCommandLineOutput.tearDownc C s� t j�t j�tj��}t j�|d�}t|� d�� | j g }t
|� \}}}| � |d� | � t j�
|�� | �t j�
| j�� t| jdd��}| � |�� d� W d � d S 1 sYw Y d S )Nz trace.coverz-m trace --count� r� r� u^ # coding: iso-8859-15
1: x = 'spœm'
1: if []:
print('unreachable')
)r r r� �abspathr^ r r� r r� r� r rk r� r� r� r� r� �read)rJ ZtracedirZtracecoverpath�argv�statusr� �stderrrA r r r �%test_cover_files_written_no_highlight� s
�"�zCTestCoverageCommandLineOutput.test_cover_files_written_no_highlightc C s| d� � | jg }t|� \}}}| �tj�| j�� t| jdd��}| � |�
� t � d�� W d � d S 1 s7w Y d S )Nz-m trace --count --missingr� r� u� # coding: iso-8859-15
1: x = 'spœm'
1: if []:
>>>>>> print('unreachable')
)
r� r� r r� r r r� r� r� rk r� r� r� )rJ r� r� r� r� rA r r r �'test_cover_files_written_with_highlight� s "�zETestCoverageCommandLineOutput.test_cover_files_written_with_highlightN) rV rW rX r� r� rh r� r� r� r r r r r� � s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestCommandLinec C s: d}|D ]^}}t ddg|�R � �^ }}| �||� qd S )N))s3 progname is missing: required with the main options�-lz-T)s8 cannot specify both --listfuncs and (--trace or --count)z-lc)s> argument -R/--no-report: not allowed with argument -r/--reportz-rR)sL must specify one of --trace, --count, --report, --listfuncs, or --trackcallsz-g)s -r/--report requires -f/--filez-r)s3 --summary can only be used with --count or --reportz-sT)s unrecognized arguments: -yz-y�-mr^ )r r� )rJ �_errors�messager9 �_r� r r r �
test_failures� s
�z TestCommandLine.test_failuresc C s� t d }tj�t �}t|ddd��7}| �t|� |�d� tddd|dd �\}}}| � d
|� d |� d |� d
�}| � |�
� |� W d � d S 1 sKw Y d S )Nr� r� �utf-8r� za = 1
r� r^ r� �ZPYTHONIOENCODINGs functions called:z
filename: z, modulename: z, funcname: <module>) r r r r r� ra r r� r r� �encode)rJ r
�
modulename�fdr� r� r� ro r r r �test_listfuncs_flag_success� s
� "�z+TestCommandLine.test_listfuncs_flag_successc C s� t tddd��}| �tt� |�d� |�d� W d � n1 s"w Y tt�\}}}tdddtdd �\}}}| �|�� |� d S )
Nr� r� r� z import sys
zprint(type(sys.argv))
r� r^ r� r� )r� r ra r r� r r� �strip)rJ r� r� Z
direct_stdoutr� Z trace_stdoutr r r �test_sys_argv_list� s
�
�z"TestCommandLine.test_sys_argv_listc C s� t � d�}t � d�}tj�t �}t|ddd�� }| �t|� | �t|� |�t� d�� W d � n1 s6w Y t
ddd |dd
�\}}}|� � }| � |d � | �
d |� | �
d
|� d|� d�|� d S )Nr� z.coverr� r� r� z� x = 1
y = 2
def f():
return x + y
for i in range(10):
f()
r� r^ z-csr� r z lines cov% module (path)z 6 100% z (�))r r r r r� ra r r� r� r� r �decoderk r� )rJ r
Z
coverfilenamer� r� r� r� r� r r r �test_count_and_summary s
�
� z&TestCommandLine.test_count_and_summaryc C s( t ddddddd� tddddd� d S ) Nr� r^ r� z--moduleZtimeitz-n�1Znot_a_module_zzzr rg r r r �test_run_as_module s z"TestCommandLine.test_run_as_moduleN)rV rW rX r� r� r� r� r� r r r r r� � s
r� �__main__),r rb Z test.supportr Ztest.support.os_helperr r r Ztest.support.script_helperr r r� r� r^ r Ztest.tracedmodulesr
r r r r$ r* r, r. r0 r2 r4 r8 r; r> rG �objectrH ZTestCaser[ r� r� r� r� r� r� r� rV �mainr r r r �<module> sH
m C >
2C �