Your IP : 216.73.217.129
3
��2a� � @ 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 jjt| dd��S )NZ
dtracedata)Zsubdir)�os�path�abspathr )�filename� r �0/opt/alt/python36/lib64/python3.6/test_dtrace.pyr s r c C st t jdd| �} y:dd� | j� D �}|jdd� d� dd� |D �}d j|�S ttfk
rn td
j| ���Y nX d S ) Nz\[[0-9]+ refs\]� c S s& g | ] }|r|j d � r|jd��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 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 ���}|j� j� }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. | j ||�}tj|tjtjdd�j� \}}|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|j dg| � djtjg| |g �}| j||�S )Nz-O� )�extendr �sys�
executabler< )r, r$ r% r& Z python_flagsr3 r r r r' G s
zTraceBackend.trace_pythonc C sv y | j td| j ��}|j� }W n0 tttfk
rP } zt|�}W Y d d }~X nX |dkrrtj dj
| 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)N)N)N) �__name__�
__module__� __qualname__r( r2 Z COMMAND_ARGSr1 r5 r< r' rA r r r r r# ) s
r# c @ s e Zd ZdZdddgZdS )�
DTraceBackendz.dZdtracez-qz-sN)rH rI rJ r( r2 r r r r rK Z s rK c @ s e Zd ZdZddgZdS )�SystemTapBackendz.stpZstapz-gN)rH rI rJ r( r2 r r r r rL _ s rL c @ sL e Zd ZdZdZdZedd� �Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dS )�
TraceTestsNr c C s | j j� d S )N)�backendrA )r, r r r �
setUpClassl s zTraceTests.setUpClassc C s& | j j|| jd�\}}| j||� d S )N)r& )rN r1 r& � assertEqual)r, r- r. r0 r r r r1 p s zTraceTests.run_casec C s | j d� d S )NZ
call_stack)r1 )r, r r r �test_function_entry_returnu s z%TraceTests.test_function_entry_returnc
sl t dddg�}ttd���}|j� � W d Q R X � �fdd�}x |d�D ]}|j|j� qFW �jt � |� d S )NZ
CALL_FUNCTIONZCALL_FUNCTION_EXZCALL_FUNCTION_KWz
call_stack.pyc sF t � dd�jd�}x.|jD ]$}t|tj�r|j| krtj|�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�opnamerP )r, Zopcodesr/ ra Z instructionr )r` r, r �test_verify_call_opcodesx s z#TraceTests.test_verify_call_opcodesc C s | j d� d S )N�gc)r1 )r, r r r �test_gc� s zTraceTests.test_gcc C s | j d� d S )N�line)r1 )r, r r r � test_line� s zTraceTests.test_line)
rH rI rJ ZmaxDiffrN r& � classmethodrO r1 rQ rf rh rj r r r r rM d s rM c @ s e Zd Ze� ZdZdS )�DTraceNormalTestsr N)rH rI rJ rK rN r& r r r r rl � s rl c @ s e Zd Ze� ZdZdS )�DTraceOptimizedTests� N)rH rI rJ rK rN r& r r r r rm � s rm c @ s e Zd Ze� ZdZdS )�SystemTapNormalTestsr N)rH rI rJ rL rN r& r r r r ro � s ro c @ s e Zd Ze� ZdZdS )�SystemTapOptimizedTestsrn N)rH rI rJ rL rN 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 r8 r? rY rG Z test.supportr r r r" r# rK rL ZTestCaserM rl rm ro rp rq rH r r r r �<module> s( 14