Sindbad~EG File Manager
a
��g� � @ 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�Zedkr�e�� dS )� N)�findfilec C s t j�t| dd��S )NZ
dtracedata)Zsubdir)�os�path�abspathr )�filename� r �,/usr/local/lib/python3.9/test/test_dtrace.pyr s r c C st t �dd| �} z<dd� | �� D �}|jdd� d� dd� |D �}d �|�W S ttfyn td
�| ���Y n0 d S ) aM Normalize DTrace output for comparison.
DTrace keeps a per-CPU buffer, and when showing the fired probes, buffers
are concatenated. So if the operating system moves our thread around, the
straight result can be "non-causal". So we add timestamps to the probe
firing, sort by that field, then strip it from the outputz\[[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 �
z%tracer produced unparsable output:
{}N) �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 sp t | jt|| j �t|d �|d��}tt|| j d ��� }|�� �� }W d � n1 s^0 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�stderr�universal_newlines)r6 �
subprocess�Popen�PIPE�STDOUT� 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 sx z | � td| j ��}|�� }W n2 tttfyR } zt|�}W Y d }~n
d }~0 0 |dkrtt� 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 rE N s �zTraceBackend.assert_usable)N)N)N)N) �__name__�
__module__� __qualname__r) r3 Z COMMAND_ARGSr2 r6 r@ r( rE r r r r r$ ) s
r$ c @ s e Zd ZdZg d�ZdS )�
DTraceBackendz.d)Zdtracez-qz-sN�rL rM rN r) r3 r r r r rO Z s rO c @ s e Zd ZdZddgZdS )�SystemTapBackendz.stpZstapz-gNrP r r r r rQ _ s rQ 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)�backendrE �r- r r r �
setUpClassl s zTraceTests.setUpClassc C s& | j j|| jd�\}}| �||� d S )N)r' )rS 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 rT r r r �test_function_entry_returnu s z%TraceTests.test_function_entry_returnc sz t g d��}ttd���}|�� � W d � n1 s60 Y � �fdd�}|d�D ]}|�|j� qV��t � |� dS )z9Ensure our call stack test hits all function call opcodes)Z
CALL_FUNCTIONZCALL_FUNCTION_EXZCALL_FUNCTION_KWz
call_stack.pyNc sF t � dd�jd�}|jD ](}t|tj�r|j| krt�|� S qg 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�opnamerV )r- Zopcodesr0 rh Z instructionr rg r �test_verify_call_opcodesx s & z#TraceTests.test_verify_call_opcodesc C s | � d� d S )N�gcrW rT r r r �test_gc� s zTraceTests.test_gcc C s | � d� d S )N�linerW rT r r r � test_line� s zTraceTests.test_line)
rL rM rN ZmaxDiffrS r' � classmethodrU r2 rX rm ro rq r r r r rR d s
rR c @ s e Zd Ze� ZdZdS )�DTraceNormalTestsr N�rL rM rN rO rS r' r r r r rs � s rs c @ s e Zd Ze� ZdZdS )�DTraceOptimizedTests� Nrt r r r r ru � s ru c @ s e Zd Ze� ZdZdS )�SystemTapNormalTestsr N�rL rM rN rQ rS r' r r r r rw � s rw c @ s e Zd Ze� ZdZdS )�SystemTapOptimizedTestsrv Nrx r r r r ry � s ry �__main__)rc Zos.pathr r r: rC r` rK Z test.supportr r r# r$ rO rQ rR ZTestCasers ru rw ry rL �mainr r r r �<module> s& 14
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists