File: //usr/local/lib/python3.10/dist-packages/greenlet/tests/__pycache__/test_tracing.cpython-310.pyc
o
���g: � @ s� d dl mZ d dlZd dlZd dlZddlmZ ddlmZ eo$eed�dfZ G dd � d e
�Z G d
d � d e �Z
G d d
� d
e�ZG dd� de �Zdd� ZG dd� de�Zedkr]e�� dS dS )� )�print_functionN� )�TestCase)�PY312�gettotalrefcountz%Broken on debug builds of Python 3.12c @ s e Zd ZdS )� SomeErrorN)�__name__�
__module__� __qualname__� r r �F/usr/local/lib/python3.10/dist-packages/greenlet/tests/test_tracing.pyr s r c @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d
� ZdS ) �GreenletTracerNFc C s g | _ || _d S �N)�actions�error_on_trace)�selfr r r r �__init__ s
zGreenletTracer.__init__c G s | j �|� | jr t�d S r )r �appendr r �r �argsr r r �__call__ s �zGreenletTracer.__call__c C � t �| �| _| jS r )�greenlet�settrace�oldtracer �r r r r � __enter__ � zGreenletTracer.__enter__c G � t �| j� d S r )r r r r r r r �__exit__" � zGreenletTracer.__exit__)F�r r r
r r r r r r r r r r
s
r
c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �TestGreenletTracingz*
Tests of ``greenlet.settrace()``
c C s� t �� }dd� }dd� }t� � }t � |�}|�� t � |�}| �t|j� W d � n1 s/w Y | �|d||ffd||ffd||ffd||ffg� d S )Nc S s d S r r r r r r �dummy- � z:TestGreenletTracing.test_a_greenlet_tracing.<locals>.dummyc S s t � �r )r r r r r �dummyexc/ � z=TestGreenletTracing.test_a_greenlet_tracing.<locals>.dummyexc�switch�throw)r �
getcurrentr
r' � assertRaisesr � assertEqual)r �mainr# r% r �g1�g2r r r �test_a_greenlet_tracing+ s
�
�z+TestGreenletTracing.test_a_greenlet_tracingc s� t �� � � fdd�}t � |�}|�� tdd��}| �t|j� | �t �� d � W d � n1 s2w Y | �|d� |ffg� d S )Nc s � � � d S r )r' r �r, r r r# A s zDTestGreenletTracing.test_b_exception_disables_tracing.<locals>.dummyT)r r' )r r) r' r
r* r r+ �gettrace)r r# �gr r r0 r �!test_b_exception_disables_tracing? s
�
�z5TestGreenletTracing.test_b_exception_disables_tracingc C s: t � }|� t�|� W d � d S 1 sw Y d S r )r
r r )r �tracerr r r �test_set_same_tracer_twiceM s "�z.TestGreenletTracing.test_set_same_tracer_twiceN)r r r
�__doc__r/ r3 r5 r r r r r" & s
r" c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � ZdS )
� PythonTracerNc C s
g | _ d S r )r r r r r r Z s
zPythonTracer.__init__c C s | j �||jjf� d S r )r r �f_code�co_name)r �frame�event�argr r r r ] s zPythonTracer.__call__c C r r )�sys�
setprofiler r r r r r r a r zPythonTracer.__enter__c G r r )r= r> r r r r r r e r zPythonTracer.__exit__r! r r r r r7 W s r7 c C s dS �N�* r r r r r � tpt_callbackh r$ rA c @ sx e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z e
je� dd� �Ze
je� dd� �ZdS )�TestPythonTracingz�
Tests of the interaction of ``sys.settrace()``
with greenlet facilities.
NOTE: Most of this is probably CPython specific.
Nc C s@ t � � }t� W d � n1 sw Y | �|g d�� d S )N)��returnr ��callrA �rD rA �rF r ��c_callr )r7 rA r+ )r r r r r �test_trace_events_trivialu s �
z+TestPythonTracing.test_trace_events_trivialc C s4 t � �
}|�� W d � |S 1 sw Y |S r )r7 r' �r �gletr r r r �
_trace_switch� s
��z TestPythonTracing._trace_switchc C s | � |�}| �|g d�� d S )N) rC )rJ rN )rF �runrE rG �rD rO )�c_returnrN rH rI )rN r+ rL r r r �$_check_trace_events_func_already_set� s
z6TestPythonTracing._check_trace_events_func_already_setc C s dd� }| � t�|�� d S )Nc S � t � S r �rA r r r r rO � r& zOTestPythonTracing.test_trace_events_into_greenlet_func_already_set.<locals>.run)rR r �r rO r r r �0test_trace_events_into_greenlet_func_already_set� s zBTestPythonTracing.test_trace_events_into_greenlet_func_already_setc C s" G dd� dt j �}| �|� � d S )Nc @ s e Zd Zdd� ZdS )zQTestPythonTracing.test_trace_events_into_greenlet_subclass_already_set.<locals>.Xc S rS r rT r r r r rO � r& zUTestPythonTracing.test_trace_events_into_greenlet_subclass_already_set.<locals>.X.runN�r r r
rO r r r r �X� s rX )r rR �r rX r r r �4test_trace_events_into_greenlet_subclass_already_set� s zFTestPythonTracing.test_trace_events_into_greenlet_subclass_already_setc C s, |� � t� |�� | �|jg d�� d S )N)rC rE rG rP rE rG rH rI )r' rA r r+ r )r r2 r4 r r r �/_check_trace_events_from_greenlet_sets_profiler� s zATestPythonTracing._check_trace_events_from_greenlet_sets_profilerc s( t � � � fdd�}| �t�|�� � d S )Nc � � � � t� S r �r rA r �r4 r r rO � � zQTestPythonTracing.test_trace_events_from_greenlet_func_sets_profiler.<locals>.run)r7 r[ r rU r r^ r �2test_trace_events_from_greenlet_func_sets_profiler� s
�zDTestPythonTracing.test_trace_events_from_greenlet_func_sets_profilerc s. t � � G � fdd�dtj�}| �|� � � d S )Nc s e Zd Z� fdd�ZdS )zSTestPythonTracing.test_trace_events_from_greenlet_subclass_sets_profiler.<locals>.Xc r\ r r] r r^ r r rO � r_ zWTestPythonTracing.test_trace_events_from_greenlet_subclass_sets_profiler.<locals>.X.runNrW r r^ r r rX � s rX )r7 r r[ rY r r^ r �6test_trace_events_from_greenlet_subclass_sets_profiler� s zHTestPythonTracing.test_trace_events_from_greenlet_subclass_sets_profilerc sn t � �d � d ���fdd�}� �fdd�}t�|�� t�|��� �� }| �|d� t� | ��jg d�� d S )Nc s �� � t� � �� t� dS r? )r rA r' r �r. r4 r r �g1_run� s
zPTestPythonTracing.test_trace_events_multiple_greenlets_switching.<locals>.g1_runc s t � ��� t � � �� d S r )rA r r' r �r- r4 r r �g2_run� s zPTestPythonTracing.test_trace_events_multiple_greenlets_switching.<locals>.g2_runr@ ) rC rE rG �rJ rc )rF re rE rG rH rI �r7 r r' r+ rA r �r rc re �xr �r- r. r4 r �.test_trace_events_multiple_greenlets_switching� s
z@TestPythonTracing.test_trace_events_multiple_greenlets_switchingc s~ t � �d � d ���fdd�}� �fdd�}t�|�� t�|��� �� ��� � �� }| �|d� t� | ��jg d�� d S )Nc s. t �� j�� ��� t� � �� t� dS r? )r r) �parentr' r rA r rb r r rc � s zYTestPythonTracing.test_trace_events_multiple_greenlets_switching_siblings.<locals>.g1_runc s. t �� j�� t� ��� t� � �� d S r )r r) rl r' rA r r rd r r re s
zYTestPythonTracing.test_trace_events_multiple_greenlets_switching_siblings.<locals>.g2_runr@ )rC rE rG rf rE rG rH rI rg rh r rj r �7test_trace_events_multiple_greenlets_switching_siblings� s
zITestPythonTracing.test_trace_events_multiple_greenlets_switching_siblings)r r r
r6 �maxDiffrK rN rR rV rZ r[ r` ra �unittest�skipIf�DEBUG_BUILD_PY312rk rm r r r r rB k s
%rB �__main__)�
__future__r r= r ro � r r �hasattrrq � Exceptionr �objectr
r" r7 rA rB r r, r r r r �<module> s& �1 8 �