Your IP : 216.73.216.213
o
0�i� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlm Z dd� Z
dd� Z G dd� d�Z G d d
� d
e �Z
G d d � d e �ZG d
d� d�ZG dd� deej�ZG dd� deej�ZG dd� deej�ZG dd� deej�Zedkrze�� dS dS )� N)�findfilec C s t j�t| dd��S )NZ
dtracedata)Zsubdir)�os�path�abspathr )�filename� r �7/opt/alt/python310/lib64/python3.10/test/test_dtrace.pyr s r c C sn t �dd| �} z dd� | �� D �}|jdd� d� dd� |D �}d �|�W S ttfy6 td
�| ���w ) Nz\[[0-9]+ refs\]� c S s$ g | ]}|r|� d �s|�d��qS )�#� )�
startswith�split��.0�rowr r r �
<listcomp> s ���z*normalize_trace_output.<locals>.<listcomp>c S s t | d �S )Nr )�int)r r r r �<lambda> s z(normalize_trace_output.<locals>.<lambda>)�keyc S s g | ]}|d �qS )� r r r r r r ! s �
z%tracer produced unparsable output:
{}) �re�sub�
splitlines�sort�join�
IndexError�
ValueError�AssertionError�format)�output�resultr r r �normalize_trace_output s � ��r" c @ sH e Zd ZdZdZg Zd dd�Zd dd�Zd dd�Zd dd �Z d
d � Z
dS )
� TraceBackendNc C sv t | jt|| j �t|d �|d��}tt|| j d ���}|�� �� }W d � ||fS 1 s2w Y ||fS )Nz.py)� script_file� python_file�optimize_pythonz .expected)r" � trace_pythonr � EXTENSION�open�read�rstrip)�self�namer& �
actual_output�f�expected_outputr r r �run_case. s
�
��zTraceBackend.run_casec C s | j |g }|r|d|g7 }|S )Nz-c)�COMMAND)r, r$ �
subcommand�commandr r r �generate_trace_command9 s z#TraceBackend.generate_trace_commandc C s. | � ||�}tj|tjtjdd��� \}}|S )NT)�stdout�stderrZuniversal_newlines)r5 �
subprocess�Popen�PIPEZSTDOUTZ communicate)r, r$ r3 r4 r6 �_r r r �trace? s ��zTraceBackend.tracec C s< g }|r |� dg| � d�tjg| |g �}| �||�S )Nz-O� )�extendr �sys�
executabler< )r, r$ r% r& Z python_flagsr3 r r r r' G s
zTraceBackend.trace_pythonc
C sx z| � td| j ��}|�� }W n tttfy) } z
t|�}W Y d }~nd }~ww |dkr:t� d�
| j d |���d S )N�
assert_usablezprobe: successz{}(1) failed: {}r ) r< r r( �strip�FileNotFoundError�NotADirectoryError�PermissionError�str�unittestZSkipTestr r2 )r, r Zfnfer r r rA N s ����zTraceBackend.assert_usable�N) �__name__�
__module__� __qualname__r( r2 Z COMMAND_ARGSr1 r5 r< r' rA r r r r r# ) s
r# c @ s e Zd ZdZg d�ZdS )�
DTraceBackendz.d)Zdtracez-qz-sN�rI rJ rK r( r2 r r r r rL Z � rL c @ s e Zd ZdZddgZdS )�SystemTapBackendz.stpZstapz-gNrM r r r r rO _ rN rO c @ sL e Zd ZdZdZdZedd� �Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dS )�
TraceTestsNr c C s | j �� d S rH )�backendrA �r, r r r �
setUpClassl s zTraceTests.setUpClassc C s& | j j|| jd�\}}| �||� d S )N)r& )rQ r1 r& � assertEqual)r, r- r. r0 r r r r1 p s
�zTraceTests.run_casec C � | � d� d S )NZ
call_stack�r1 rR r r r �test_function_entry_returnu � z%TraceTests.test_function_entry_returnc s| t g d��}ttd��� }|�� � W d � n1 s w Y � �fdd�}|d�D ]}|�|j� q,��t � |� d S )N)Z
CALL_FUNCTIONZCALL_FUNCTION_EXZCALL_FUNCTION_KWz
call_stack.pyc sF t � dd�jd�}|jD ]}t|tj�r |j| kr t�|� S q g S )Nz<string>�exec)�sourcer �mode�optimize) �compiler& � co_consts�
isinstance�types�CodeType�co_name�disZget_instructions)�funcname�code�c�Z code_stringr, r r �get_function_instructions� s �
�zFTraceTests.test_verify_call_opcodes.<locals>.get_function_instructions�start)�setr) r r* �discard�opnamerT )r, Zopcodesr/ rh Z instructionr rg r �test_verify_call_opcodesx s
� z#TraceTests.test_verify_call_opcodesc C rU )NZgcrV rR r r r �test_gc� rX zTraceTests.test_gcc C rU )N�linerV rR r r r � test_line� rX zTraceTests.test_line)
rI rJ rK ZmaxDiffrQ r& � classmethodrS r1 rW rm rn rp r r r r rP d s
rP c @ � e Zd Ze� ZdZdS )�DTraceNormalTestsr N�rI rJ rK rL rQ r& r r r r rs � � rs c @ rr )�DTraceOptimizedTests� Nrt r r r r rv � ru rv c @ rr )�SystemTapNormalTestsr N�rI rJ rK rO rQ r& r r r r rx � ru rx c @ rr )�SystemTapOptimizedTestsrw Nry r r r r rz � ru rz �__main__)rc Zos.pathr r r8 r? r` rG Z test.supportr r r" r# rL rO rP ZTestCasers rv rx rz rI �mainr r r r �<module> s* 14 �