File: //usr/local/lib/python3.7/test/__pycache__/test_dtrace.cpython-37.opt-2.pyc
B
��g� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlm Z m
Z
dd� Z dd� Z G dd� d�Z
G d d
� d
e
�ZG d d � d e
�ZG d
d� dej�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Zdd� Zedkr�e� dS )� N)�findfile� run_unittestc C s t j�t| dd��S )NZ
dtracedata)Zsubdir)�os�path�abspathr )�filename� r �,/usr/local/lib/python3.7/test/test_dtrace.pyr s r c C st t �dd| �} y:dd� | �� D �}|jdd� d� dd� |D �}d �|�S ttfk
rn td
�| ���Y nX d S ) 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> � z(normalize_trace_output.<locals>.<lambda>)�keyc S s g | ] }|d �qS )� r )r r r r r r ! s �
z&tracer produced unparseable 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�Zddd�Zddd �Z d
d � Z
dS )� TraceBackendNc C s\ t | jt|| j �t|d �|d��}tt|| j d ���}|�� �� }W d Q R X ||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)r6 �
subprocess�Popen�PIPEZSTDOUTZ communicate)r- r% r4 r5 r7 �_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_flagsr4 r r r r( G s
zTraceBackend.trace_pythonc
C sv y | � td| j ��}|�� }W n0 tttfk
rP } z t|�}W d d }~X Y nX |dkrrt� d�
| j d |���d S )N�
assert_usablezprobe: successz{}(1) failed: {}r ) r= r r) �strip�FileNotFoundError�NotADirectoryError�PermissionError�str�unittestZSkipTestr r3 )r- r! Zfnfer r r rB N s zTraceBackend.assert_usable)N)N)N)N) �__name__�
__module__� __qualname__r) r3 Z COMMAND_ARGSr2 r6 r= r( rB r r r r r$ ) s
r$ c @ s e Zd ZdZdddgZdS )�
DTraceBackendz.dZdtracez-qz-sN)rI rJ rK r) r3 r r r r rL Z s rL c @ s e Zd ZdZddgZdS )�SystemTapBackendz.stpZstapz-gN)rI rJ rK r) r3 r r r r rM _ s rM 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 )N)�backendrB )r- r r r �
setUpClassl s zTraceTests.setUpClassc C s& | j j|| jd�\}}| �||� d S )N)r' )rO r2 r' � assertEqual)r- r. r/ r1 r r r r2 p s zTraceTests.run_casec C s | � d� d S )NZ
call_stack)r2 )r- r r r �test_function_entry_returnu s z%TraceTests.test_function_entry_returnc sl t dddg�}ttd���}|�� � W d Q R X � �fdd�}x |d�D ]}|�|j� qFW ��t � |� d S )NZ
CALL_FUNCTIONZCALL_FUNCTION_EXZCALL_FUNCTION_KWz
call_stack.pyc sF t � dd�jd�}x.|jD ]$}t|tj�r|j| krt�|�S qW g S )Nz<string>�exec)�sourcer �mode�optimize) �compiler' � co_consts�
isinstance�types�CodeType�co_name�disZget_instructions)�funcname�code�c)� code_stringr- r r �get_function_instructions� s
zFTraceTests.test_verify_call_opcodes.<locals>.get_function_instructions�start)�setr* r r+ �discard�opnamerQ )r- Zopcodesr0 rb Z instructionr )ra r- r �test_verify_call_opcodesx s z#TraceTests.test_verify_call_opcodesc C s | � d� d S )NZgc)r2 )r- r r r �test_gc� s zTraceTests.test_gcc C s | � d� d S )N�line)r2 )r- r r r � test_line� s zTraceTests.test_line)
rI rJ rK ZmaxDiffrO r' � classmethodrP r2 rR rg rh rj r r r r rN d s rN c @ s e Zd Ze� ZdZdS )�DTraceNormalTestsr N)rI rJ rK rL rO r' r r r r rl � s rl c @ s e Zd Ze� ZdZdS )�DTraceOptimizedTests� N)rI rJ rK rL rO r' r r r r rm � s rm c @ s e Zd Ze� ZdZdS )�SystemTapNormalTestsr N)rI rJ rK rM rO r' r r r r ro � s ro c @ s e Zd Ze� ZdZdS )�SystemTapOptimizedTestsrn N)rI rJ rK rM rO r' r r r r rp � s rp c C s t tttt� d S )N)r rl rm ro rp r r r r � test_main� s rq �__main__)r] Zos.pathr r r9 r@ rZ rH Z test.supportr r r r# r$ rL rM ZTestCaserN rl rm ro rp rq rI r r r r �<module> s( 14