File: //usr/local/lib/python3.10/dist-packages/greenlet/tests/__pycache__/test_greenlet.cpython-310.pyc
o
���g�� � @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dlZd dlmZ ddl
m Z ddl
m Z ddl
m
Z
dd lmZ G d
d � d e�Zd d
� Zdd� ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �Zedkr�e�� dS dS )� N)�ABCMeta)�abstractmethod)�greenlet� )�TestCase)�RUNNING_ON_MANYLINUX)�PY313)�fails_leakcheckc @ � e Zd ZdS )� SomeErrorN��__name__�
__module__� __qualname__� r r �G/usr/local/lib/python3.10/dist-packages/greenlet/tests/test_greenlet.pyr � r c C s0 z
t �� j�� W t� | �t�� d � � �Nr )r �
getcurrent�parent�switch�append�sys�exc_infor ��seenr r r �fmain s �r c C s" dd� }t || d�}|�|� d S )Nc S s | ��Nr )�excr r r �crasher, � z send_exception.<locals>.crasher�r �� RawGreenletr )�gr r �g1r r r �send_exception) s r& c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Ze�eo=ed�d d � �Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Z d.d/� Z d0d1� Z d2d3� Z d4d5� Z e!d]d7d8��Z"e!d9d:� �Z#d;d<� Z$e!d=d>� �Z%d?d@� Z&dAdB� Z'dCdD� Z(dEdF� Z)dGdH� Z*dIdJ� Z+e�eo�edK�dLdM� �Z,dNdO� Z-dPdQ� Z.dRdS� Z/dTdU� Z0dVdW� Z1dXdY� Z2dZd[� Z3d\S )^� TestGreenletc s^ g � � fdd�}t |�}� �d� |�� � �d� |�� � �d� | �� ttd��� d S )Nc s& � � d� t�� j�� � � d� d S )Nr � �r r r r r r ��lstr r �f7 s
z'TestGreenlet._do_simple_test.<locals>.fr � � � )r# r r � assertEqual�list�range)�selfr, r$ r r* r �_do_simple_test4 s
z TestGreenlet._do_simple_testc C � | � � d S r )r4 �r3 r r r � test_simpleC s zTestGreenlet.test_simplec C sN t � }| �t�� }|�� W d � n1 sw Y | �dt|j�� d S �N�run)r# � assertRaises�AttributeErrorr �assertIn�str� exception�r3 r$ r r r r �(test_switch_no_run_raises_AttributeErrorF s
�z5TestGreenlet.test_switch_no_run_raises_AttributeErrorc C sP t � }| �t��
}|�t� W d � n1 sw Y | �dt|j�� d S r8 )r# r: r; �throwr r< r= r> r? r r r �'test_throw_no_run_raises_AttributeErrorM s
�z4TestGreenlet.test_throw_no_run_raises_AttributeErrorc C s* t d d�}| �|� | �|jt�� � d S )Nr! )r# �assertIsNotNone�assertIsr r r �r3 r$ r r r �test_parent_equals_NoneT s
z$TestGreenlet.test_parent_equals_Nonec C s$ t d d�}| �|� | �|j� d S )N)r9 )r# rC � assertIsNoner9 rE r r r �test_run_equals_NoneY �
z!TestGreenlet.test_run_equals_Nonec s� g � � fdd�}t |�}t |�}|�� | �t� �d� |�� | �t� �d� |�� | �t� �d� | �|jd� |�� | �t� �d� | �|jd� d S )Nc s* � � d� t�� j�� � �ddg� d S )Nr )r r r r r �extendr r* r r r, a s
z)TestGreenlet.test_two_children.<locals>.fr r- r. T� )r# r r0 �len�dead)r3 r, r$ �hr r* r �test_two_children^ s z TestGreenlet.test_two_childrenc sf g ��fdd�� � �fdd�}t |�}| �t�|�d� |�� | ��g d�� | �t�|�d� d S )Nc s � � d� t�� j�� d S )N�br) r r* r r r, u s
z3TestGreenlet.test_two_recursive_children.<locals>.fc s( �� d� t� �} | �� �� d� d S )N�a�c)r r# r �r$ �r, r+ r r r$ y s
z3TestGreenlet.test_two_recursive_children.<locals>.gr- )rQ rP rR )r# r0 r � getrefcountr rE r rT r �test_two_recursive_childrenr s z(TestGreenlet.test_two_recursive_childrenc sf g ���fdd�� � fdd�t d�D �}|D ]}|�� q|D ]}|�d� q ��t��t|�� d S )Nc s � � � ��d� d S �NT)r4 r r )r3 �successr r r, � s z$TestGreenlet.test_threads.<locals>.fc s g | ]}t j� d ��qS )��target)� threading�Thread)�.0�i�r, r r �
<listcomp>� s z-TestGreenlet.test_threads.<locals>.<listcomp>�
)r2 �start�joinr0 rL )r3 �ths�thr )r, r3 rX r � test_threads� s
zTestGreenlet.test_threadsc C s� g }t t�}t t�}|�|� |�|� ||_| �|g � | �t|j� | �|tg� |�� }| �|d� | �|tg� |�d�}| �|d� | �|tg� d S )Nr � )r# r r r r0 r: r )r3 r r% �g2�valuer r r �test_exception� s
zTestGreenlet.test_exceptionc C s8 g }t t�}|�|� | �tt|t� | �|tg� d S r )r# r r r: �KeyErrorr& r0 )r3 r r% r r r �test_send_exception� s
z TestGreenlet.test_send_exceptionc C sp g }t t�}t t�}|�|� |�|� | �|g � ~t�� | �|tjg� ~t�� | �|tjtjg� d S r )r# r r r0 �gc�collectr � GreenletExit)r3 r r% rh r r r � test_dealloc� s
zTestGreenlet.test_deallocc C s| dd� }t |�}|�� tj}ddlm} |� }t_z~W |t_n|t_w |�� }| �d|� | �d|� | �d|� d S )Nc S s, z
t �� j�� W d S t jy td �w r )r r r r ro r r r r r r9 � s
�zHTestGreenlet.test_dealloc_catches_GreenletExit_throws_other.<locals>.runr )�StringIO� Exception�ignoredr )r# r r �stderr�iorq �getvaluer< )r3 r9 r$ � oldstderrrq rt �vr r r �.test_dealloc_catches_GreenletExit_throws_other� s z;TestGreenlet.test_dealloc_catches_GreenletExit_throws_otherz=Sometimes flaky (getting one GreenletExit in the second list)c s g �g �t �� � t �� �t �� �t �� �� �����fdd�}t j|d�}|�� � �d� | ��g � | �t��d� �d d �= t�� | ��g � �� � ��d� z | ��t
j g� W �� � |� d� �d d �= �d d �= d S �� � |� d� �d d �= �d d �= w )Nc sZ t t�} | ��� ��| � ~ t�� � �� ��d� t � ��� ��d� t � d S )Nr( )r# r r r rm rn �set�wait)r% ��/bg_glet_created_running_and_no_longer_ref_in_bg�bg_should_be_clear�fg_ref_released�ok_to_exit_bg_threadr �somerefr r r, � s
z1TestGreenlet.test_dealloc_other_thread.<locals>.frY ra r r( )
r[ �Eventr\ rb r{ r0 rL rm rn rz r ro rc )r3 r, �tr r| r �test_dealloc_other_thread� s6
�
z&TestGreenlet.test_dealloc_other_threadc sf � fdd�}t |�}|�� }� �||ju � � �|� |�� }� �|� � �|d� � �|jd � d S )Nc s, t �d�} � �| jd � t�� j�| � dS )Nr �meaning of life)r � _getframer0 �f_backr r r r r_ r6 r r �f1 s
z#TestGreenlet.test_frame.<locals>.f1r� )r# r �
assertTrue�gr_frame� assertFalser0 )r3 r� r$ �frame�from_gr r6 r �
test_frame s
zTestGreenlet.test_framec C sL dd� }t j|dd�}t j|dd�}|�� |�� |�d� |�d� d S )Nc s t � fdd��}|�� d S )Nc s
t �� �S r )�time�sleepr ��xr r �<lambda>$ �
z>TestGreenlet.test_thread_bug.<locals>.runner.<locals>.<lambda>r" )r� r$ r r� r �runner# s z,TestGreenlet.test_thread_bug.<locals>.runner)g�������?�rZ �args)g333333�?ra )r[ r\ rb rc )r3 r� �t1�t2r r r �test_thread_bug" s
z TestGreenlet.test_thread_bugc s. � fdd�}t |�jddd�}� �|d� d S )Nc s � � | d� � � |d� dS )Nr. r- �* )r0 �rQ rP r6 r r r9 . s z,TestGreenlet.test_switch_kwargs.<locals>.runr. r- r� r� �r# r r0 )r3 r9 r� r r6 r �test_switch_kwargs- s z TestGreenlet.test_switch_kwargsc C sR dd� }t |�}| �ddi|�d�� | �dddif|�� � | �d|�� � d S )Nc S s2 t �� jj| d� t �� jjddd� | | d fS )Nr� r- r( �r r r r r� r r r r9 6 s z6TestGreenlet.test_switch_kwargs_to_parent.<locals>.runr� r( )r- )r( � )r# r0 r )r3 r9 r$ r r r � test_switch_kwargs_to_parent5 s
z)TestGreenlet.test_switch_kwargs_to_parentc s� i �t �� � t �� �� ��fdd�}t j|d�}|�� � �d� | �tj�� �d �� W d � n1 s8w Y �� � |�
d� �� � d S )Nc �& t dd� ��d< � �� ��d� d S )Nc S � d S r r r r r r r� E � zITestGreenlet.test_switch_to_another_thread.<locals>.run.<locals>.<lambda>r$ ra �r# rz r{ r ��
created_event�data�
done_eventr r r9 D � z7TestGreenlet.test_switch_to_another_thread.<locals>.runrY ra r$ ) r[ r� r\ rb r{ r: r �errorr rz rc �clear)r3 r9 �threadr r� r � test_switch_to_another_thread? s
�
z*TestGreenlet.test_switch_to_another_threadc s* � �fdd�}�fdd�� t |��� d S )Nc s: zt d�� t�� } t� ��� ��| t�� � Y d S )N�fun)�
ValueErrorr r r# r r0 )r �rN r3 r r r, S s z&TestGreenlet.test_exc_state.<locals>.fc s � � t�� d� d S )N)NNN)r0 r r r r6 r r rN [ � z&TestGreenlet.test_exc_state.<locals>.hr" )r3 r, r r� r �test_exc_stateR s zTestGreenlet.test_exc_statec C s� dd� }dd� }dd� }t |�}| �|ji � |�� | �|jd� | �|jddi� |j|_| �|jddi� | �t||� | �t||d� d S ) Nc S s dt �� _d S �Nr� )r r �testr r r r r, a � z*TestGreenlet.test_instance_dict.<locals>.fc S s | ` d S r ��__dict__rS r r r �deldictc � z0TestGreenlet.test_instance_dict.<locals>.deldictc S s
|| _ d S r r� )r$ ri r r r �setdicte �
z0TestGreenlet.test_instance_dict.<locals>.setdictr� r� )r# r0 r� r r� r: � TypeError)r3 r, r� r� r$ r r r �test_instance_dict` s z TestGreenlet.test_instance_dictc s2 g � � fdd�}t |�}|�� | �� dg� d S )Nc s � � tt�� d�� d S r8 )r �hasattrr r r ��has_runr r �funcs s �z;TestGreenlet.test_running_greenlet_has_no_run.<locals>.funcFr� �r3 r� r$ r r� r � test_running_greenlet_has_no_runq s
z-TestGreenlet.test_running_greenlet_has_no_runc C s0 dd l }| �t|j t� � | �t|jt� � d S r )�copyr: r� r# �deepcopy)r3 r� r r r �
test_deepcopy| s zTestGreenlet.test_deepcopyc s� t dd� �� t�� �g �� ��fdd�}t |� d�}|�� ~| ��� | ��d �� | ��d j� � �d d �= d � d �d S )Nc S r� r r r r r r r� � r� z;TestGreenlet.test_parent_restored_on_kill.<locals>.<lambda>c sL z�� � W d S tjy% ��t�� j� ��t�� � � � � Y d S w r )r r ro r r r r ��hub�main�resultr r �worker� s �z9TestGreenlet.test_parent_restored_on_kill.<locals>.workerr! r r )r# r r r r� rD r )r3 r� r$ r r� r � test_parent_restored_on_kill� s
z)TestGreenlet.test_parent_restored_on_killc C sN t � }t dd� |d�}| �t��
|�� W d � d S 1 s w Y d S )Nc S r� r r r r r r r� � r� z9TestGreenlet.test_parent_return_failure.<locals>.<lambda>r! )r# r: r; r )r3 r% rh r r r �test_parent_return_failure� s
"�z'TestGreenlet.test_parent_return_failurec C s2 G dd� dt �}|dd� �}| �t|jt� � d S )Nc @ � e Zd Zdd� ZdS )z>TestGreenlet.test_throw_exception_not_lost.<locals>.mygreenletc S s zt � Y t�| |�S r )rr r# �__getattribute__�r3 �namer r r r� � s
zOTestGreenlet.test_throw_exception_not_lost.<locals>.mygreenlet.__getattribute__N�r
r r r� r r r r �
mygreenlet� � r� c S r� r r r r r r r� � r� z<TestGreenlet.test_throw_exception_not_lost.<locals>.<lambda>)r# r: r rA )r3 r� r$ r r r � test_throw_exception_not_lost� s z*TestGreenlet.test_throw_exception_not_lostFc sT g � dd� �� ��fdd�}t j|d�}|�� |�d� ~�r$| �� | �tj��}� d �t � W d � n1 s<w Y �sTt
|j �}| � |dkpQd |v � n | �
t
|j �d� t� d jd
�r�| �t�� � d j�� W d � n1 s|w Y �s�t� d jjd
�r�� d jj�� n| �� d j� ~d �� d d �= d | _d S ) Nc S � t �� j�� d S r r� r r r r r� � � zGTestGreenlet._do_test_throw_to_dead_thread_doesnt_crash.<locals>.workerc s6 t ��} | �� � �| � �r| �� t�� d S d S r )r# r r r r rS �r� �wait_for_cleanupr� r r �creator� s
�zHTestGreenlet._do_test_throw_to_dead_thread_doesnt_crash.<locals>.creatorrY ra r �Bcannot switch to a different thread (which happens to have exited)z
Cannot switchr� T)r[ r\ rb rc �wait_for_pending_cleanupsr: r r� rA r r= r> r� r0 r� r� � RuntimeErrorr� �f_localsrG �expect_greenlet_leak)r3 r� r� r� r �sr r� r �*_do_test_throw_to_dead_thread_doesnt_crash� sF
�
�� ��
z7TestGreenlet._do_test_throw_to_dead_thread_doesnt_crashc C r5 r �r� r6 r r r �&test_throw_to_dead_thread_doesnt_crash� s z3TestGreenlet.test_throw_to_dead_thread_doesnt_crashc C s | � d� d S rW r� r6 r r r �+test_throw_to_dead_thread_doesnt_crash_wait� r� z8TestGreenlet.test_throw_to_dead_thread_doesnt_crash_waitc C s2 G dd� dt �}|� }| �|�d�d� d| _d S )Nc @ �$ e Zd Zdd� Zdd� Zdd� ZdS )z7TestGreenlet.test_recursive_startup.<locals>.convolutedc S s t �| � d| _d S r )r# �__init__�countr6 r r r r� � s
z@TestGreenlet.test_recursive_startup.<locals>.convoluted.__init__c S s. |dkr| j dkrd| _ | �d� t�| |�S )Nr9 r r �+ )r� r r# r� r� r r r r� � s
zHTestGreenlet.test_recursive_startup.<locals>.convoluted.__getattribute__c S s | j �|� qr �r r �r3 ri r r r r9 � s �z;TestGreenlet.test_recursive_startup.<locals>.convoluted.runN)r
r r r� r� r9 r r r r �
convoluted� s r� r� r� T)r# r0 r r� )r3 r� r$ r r r �test_recursive_startup� s
z#TestGreenlet.test_recursive_startupc s� t �� ���� t �� ���� G ��fdd�dt�}dd� }� ��fdd�}t�� }t|�g� |� � d _� d �� t j |d�}|�
� �� � ��� | � t�� |� |�
d � d S )
Nc s e Zd Z� �fdd�ZdS )z;TestGreenlet.test_threaded_updatecurrent.<locals>.finalizedc s �� � � �� d S r ��release�acquirer6 ��lock1�lock2r r �__del__ s zCTestGreenlet.test_threaded_updatecurrent.<locals>.finalized.__del__N�r
r r r� r r� r r � finalized s r� c S r� r r� r r r r � deallocator r� z=TestGreenlet.test_threaded_updatecurrent.<locals>.deallocatorc s: �� � t�� � d= ��� �� � t�� ��� d S r �r� r r r� r �r$ r� r� r r �fthread s z9TestGreenlet.test_threaded_updatecurrent.<locals>.fthreadr rY ra )r[ �Lockr� �objectr r r# �bombr r\ rb r� r0 rc )r3 r� r� r� r� r� r r� r �test_threaded_updatecurrent s"
z(TestGreenlet.test_threaded_updatecurrentc sb g �� ��fdd��dd� � t ��g��d �� t � �d �� � �� }| ��� | �|d� d S )Nc
sV t �� j�� } �d= t �� j� _zt �� j�| � W ��t �� � d S ��t �� � w r )r r r r r )ri �� initiatorr r� r r r� 8 s "z>TestGreenlet.test_dealloc_switch_args_not_lost.<locals>.workerc S � dS r� r r r r r r E r zATestGreenlet.test_dealloc_switch_args_not_lost.<locals>.initiatorr r� )r# r r� r0 r� r r� r �!test_dealloc_switch_args_not_lost6 s
z.TestGreenlet.test_dealloc_switch_args_not_lostc sT dd� � G dd� dt �}|� �tdd��� ��fdd�}t|�}| �|�� �� d S ) Nc S s | |i |�� d S r r )r� rQ �kr r r �_apply^ r� z0TestGreenlet.test_tuple_subclass.<locals>._applyc @ r� )z1TestGreenlet.test_tuple_subclass.<locals>.mytuplec S s t �� �� t�| �S r )r r r �tuple�__len__r6 r r r r b s
z9TestGreenlet.test_tuple_subclass.<locals>.mytuple.__len__N)r
r r r r r r r �mytuplea r� r r� )rQ c s � t �� jj��� d S r r� r �r r� �kwargsr r � switchapplyg s z5TestGreenlet.test_tuple_subclass.<locals>.switchapply)r �dictr# r0 r )r3 r r
r$ r r r �test_tuple_subclassP s
z TestGreenlet.test_tuple_subclassc C sP t dtfdtdd� �i�}G dd� d|�}G dd� d|�}|� | �t|� d S ) N�AbstractSubclassr9 c S r� r r r6 r r r r� p r� z7TestGreenlet.test_abstract_subclasses.<locals>.<lambda>c @ r
)z:TestGreenlet.test_abstract_subclasses.<locals>.BadSubclassNr r r r r � BadSubclassr r r c @ r� )z;TestGreenlet.test_abstract_subclasses.<locals>.GoodSubclassc S r� r r r6 r r r r9 v r z?TestGreenlet.test_abstract_subclasses.<locals>.GoodSubclass.runN)r
r r r9 r r r r � GoodSubclassu r� r )r r# r r: r� )r3 r
r r r r r �test_abstract_subclassesl s �z%TestGreenlet.test_abstract_subclassesc sP t �� sd S t �� d � � dk rd S � �fdd�}td�D ]}|� r% d S q d S )Nr �2 c s t �� ���� t �� ���� dg���fdd�} t j| d�}|�� G ���fdd�dt�� G � fdd�dt�}g }t�d �}t�� }|� }|D ] }d }�d
r[|� d � dS t
� }�d
rd n|� |� qKt �
� �d
rw|� d � dS ��|j|� |D ] }��|j|� q�d S )
NFc s � � � t�� ��� d S r r� r r� r r �another_thread� � zWTestGreenlet.test_implicit_parent_with_threads.<locals>.attempt.<locals>.another_threadrY c s e Zd Z� ��fdd�ZdS )zTTestGreenlet.test_implicit_parent_with_threads.<locals>.attempt.<locals>.gc_callbackc s � � � ��� d�d< d S )NTr r� r6 �r� r� �recycledr r r� � r z\TestGreenlet.test_implicit_parent_with_threads.<locals>.attempt.<locals>.gc_callback.__del__Nr� r r r r � gc_callback� s r c � e Zd Z� fdd�ZdS )zPTestGreenlet.test_implicit_parent_with_threads.<locals>.attempt.<locals>.garbagec s | | _ � � | _d S r )�cycle�callbackr6 �r r r r� � s zYTestGreenlet.test_implicit_parent_with_threads.<locals>.attempt.<locals>.garbage.__init__N)r
r r r� r r r r �garbage� � r r- r ra T)r[ r� r� r\ rb r� r2 r r rc r# r rm rn r0 r ) r r� r �lr� �currentr$ �_�last��Nr3 )r r� r� r r �attempt� s>
z?TestGreenlet.test_implicit_parent_with_threads.<locals>.attemptr/ )rm � isenabled�
get_thresholdr2 )r3 r# r r r! r �!test_implicit_parent_with_threads| s -��z.TestGreenlet.test_implicit_parent_with_threadsc s| ddl m� ddl m� G dd� dt�}t�|�}g �_� ��fdd�}td�D ]}||��� q(�`�� t�|�|� d S ) Nr �r )ro c @ r
)zTTestGreenlet.test_issue_245_reference_counting_subclass_no_threads.<locals>.GreenletNr r r r r �Greenlet� r r( c s6 z �� j �� W d S � y �j��� � Y d S w r )r r �gletsr r �ro r r3 r r �
greenlet_main� s
�zYTestGreenlet.test_issue_245_reference_counting_subclass_no_threads.<locals>.greenlet_mainra )
r r ro r# r rU r) r2 r r0 )r3 r( � initial_refsr+ r r r* r �5test_issue_245_reference_counting_subclass_no_threads� s
zBTestGreenlet.test_issue_245_reference_counting_subclass_no_threadsz;The manylinux images appear to hang on this test on 3.13rc2c
s� ddl m} ddl m} ddlm� G dd� dt�� g �|� ��fdd��� ����fd d
�}t�� �}g }t|d �D ]}|� }|||fd �}|� |� |�
� q<|D ]} | � d
� qT�d d �= �� � | �
� | �t�� �|� d S )Nr )r\ )r� r' c @ r
)zSTestGreenlet.test_issue_245_reference_counting_subclass_threads.<locals>.MyGreenletNr r r r r �
MyGreenlet� r r. c s � � j �� d S r r� r r' r r r+ � s zVTestGreenlet.test_issue_245_reference_counting_subclass_threads.<locals>.greenlet_mainc s8 � ��}�� |� |�� ~| �� ��d� �� d S )Nra )r r rz r{ )�greenlet_running_event�mine�r. r r) r+ � ref_clearedr r � thread_main� s
zTTestGreenlet.test_issue_245_reference_counting_subclass_threads.<locals>.thread_main�- r� ra )r[ r\ r� r r r# r rU r2 r rb r{ rz r� r0 )
r3 r\ r� r3 r, �thread_ready_eventsr �eventr� r� r r1 r �2test_issue_245_reference_counting_subclass_threads� s.
�
z?TestGreenlet.test_issue_245_reference_counting_subclass_threadsc C sZ dd� }t |�}dd� }t ||�}| �t��
|�� W d � d S 1 s&w Y d S )Nc S r )N�
r r r r r �no_args r z\TestGreenlet.test_falling_off_end_switches_to_unstarted_parent_raises_error.<locals>.no_argsc S r r� r r r r r �leaf" r zYTestGreenlet.test_falling_off_end_switches_to_unstarted_parent_raises_error.<locals>.leaf)r# r: r� r )r3 r9 �parent_never_startedr: �childr r r �>test_falling_off_end_switches_to_unstarted_parent_raises_error s
"�zKTestGreenlet.test_falling_off_end_switches_to_unstarted_parent_raises_errorc C s: dd� }t |�}dd� }t ||�}|�� }| �|d� d S )Nc S s | dfS �N� r r� r r r �one_arg, r� zUTestGreenlet.test_falling_off_end_switches_to_unstarted_parent_works.<locals>.one_argc S r r� r r r r r r: 1 r zRTestGreenlet.test_falling_off_end_switches_to_unstarted_parent_works.<locals>.leaf)r� r? r� )r3 r@ r; r: r< r� r r r �7test_falling_off_end_switches_to_unstarted_parent_works+ s
zDTestGreenlet.test_falling_off_end_switches_to_unstarted_parent_worksc s� G dd� dt �}|� }g � t � fdd�|�}| �t�� |�� W d � n1 s*w Y | �� dg� | �t�� |�� W d � n1 sJw Y | �� dg� d S )Nc @ r� )zXTestGreenlet.test_switch_to_dead_greenlet_with_unstarted_perverse_parent.<locals>.Parentc S s |dkrt �d S r8 )r r� r r r r� ; s �ziTestGreenlet.test_switch_to_dead_greenlet_with_unstarted_perverse_parent.<locals>.Parent.__getattribute__Nr� r r r r �Parent: r� rB c �
� � d�S r� �r r r r r r� B r� zZTestGreenlet.test_switch_to_dead_greenlet_with_unstarted_perverse_parent.<locals>.<lambda>r� )r# r: r r r0 )r3 rB r; r< r r r �;test_switch_to_dead_greenlet_with_unstarted_perverse_parent9 s
�
�zHTestGreenlet.test_switch_to_dead_greenlet_with_unstarted_perverse_parentc sf g � t � fdd��}t � fdd��}|�� | �� dg� ||_|�� }| �|� | �� ddg� d S )Nc rC r> rD r r r r r� P r� zDTestGreenlet.test_switch_to_dead_greenlet_reparent.<locals>.<lambda>c rC r� rD r r r r r� Q r� r� r? )r# r r0 r rG )r3 r; r< r� r r r �%test_switch_to_dead_greenlet_reparentN s
z2TestGreenlet.test_switch_to_dead_greenlet_reparentc s� t �� �� fdd�}�fdd�� t|�}|�� tdd� �}|�� | �|j� | �|jjjd� | �|jj � | �|jj jjd� | �
|jj j � d S ) Nc s
� � d S r r r ��innerr r �outerb r� zHTestGreenlet.test_can_access_f_back_of_suspended_greenlet.<locals>.outerc s � � t�d�� d S r )r r r� r �r� r r rH e r� zHTestGreenlet.test_can_access_f_back_of_suspended_greenlet.<locals>.innerc S r� r r r r r r r� n r� zKTestGreenlet.test_can_access_f_back_of_suspended_greenlet.<locals>.<lambda>rH rI ) r r r# r rC r� r0 �f_code�co_namer� rG )r3 rI r� � unrelatedr �rH r� r �,test_can_access_f_back_of_suspended_greenlet\ s z9TestGreenlet.test_can_access_f_back_of_suspended_greenletc s� ddl m� ddlm� � ��fdd��t��}|�d� |j}td�D ]}| �|j d |� |j
}q$| �|j d d� | � |j
� | �|�d �d
� d S ) Nr )�partialr )�_test_extension_cppc s. | dkr| � � ��| d �� S t�� j�� S )Nr r )� test_callr r r r )rx �rQ rP �recurser r rT } s z@TestGreenlet.test_get_stack_with_nested_c_calls.<locals>.recurser/ rx ra i� ) � functoolsrP � rQ r# r r� r2 r0 r� r� rG )r3 �grr� r^ r rS r �"test_get_stack_with_nested_c_callsy s
z/TestGreenlet.test_get_stack_with_nested_c_callsc sd t �� �� fdd�}�fdd�� t|�}|�� }tdd� �}|�� | �|jjd� | �|j� d S )Nc s � t �d�� d S r )r r� r rG r r rI � r� z6TestGreenlet.test_frames_always_exposed.<locals>.outerc s � � | � d S r )r )r� rJ r r rH � r� z6TestGreenlet.test_frames_always_exposed.<locals>.innerc S r� r r r r r r r� � r� z9TestGreenlet.test_frames_always_exposed.<locals>.<lambda>rI ) r r r# r r0 rK rL rG r� )r3 rI rW r� rM r rN r �test_frames_always_exposed� s z'TestGreenlet.test_frames_always_exposedN)F)4r
r r r4 r7 r@ rB rF rH rO rV rf rj rl rp ry �unittest�skipIfr r r� r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r� r� r� r r r r& r- r7 r= rA rE rF rO rX rY r r r r r' 2 sp
�
,
9
/ 7 �
C r' c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�TestGreenletSetParentErrorsc s� i �t �� � t �� �� ��fdd�}dd� }t j|d�}|�� � �d� t|�}|�� | �t��
}�d |_ W d � n1 sBw Y ��
� |� d� | � t
|j�d� d S ) Nc r� )Nc S r� r r r r r r r� � r� zQTestGreenletSetParentErrors.test_threaded_reparent.<locals>.run.<locals>.<lambda>r$ ra r� r r� r r r9 � r� z?TestGreenletSetParentErrors.test_threaded_reparent.<locals>.runc S r� r r� r r r r �blank� r� zATestGreenletSetParentErrors.test_threaded_reparent.<locals>.blankrY ra r$ z&parent cannot be on a different thread)r[ r� r\ rb r{ r# r r: r� r rz rc r0 r= r> )r3 r9 r] r� r$ r r r� r �test_threaded_reparent� s
�
z2TestGreenletSetParentErrors.test_threaded_reparentc s� g � � fdd�}t j|d�}|�� |�d� | j| jd d� G � fdd�dt�}|d d
� �}| �tj �� }|�
� W d � n1 sEw Y | � t |j
�d � � d d �= d S ) Nc s" t dd� �} � �| � | �� d S )Nc S r� r r r r r r r� � r� zYTestGreenletSetParentErrors.test_unexpected_reparenting.<locals>.worker.<locals>.<lambda>)r# r r rS ��anotherr r r� � s
zGTestGreenletSetParentErrors.test_unexpected_reparenting.<locals>.workerrY ra r )�initial_main_greenletsc r )zKTestGreenletSetParentErrors.test_unexpected_reparenting.<locals>.convolutedc � |dkr � d | _ t�| |�S �Nr9 r �r r# r� r� r_ r r r� � �
z\TestGreenletSetParentErrors.test_unexpected_reparenting.<locals>.convoluted.__getattribute__Nr� r r_ r r r� � r r� c S r� r r r r r r r� � r� zITestGreenletSetParentErrors.test_unexpected_reparenting.<locals>.<lambda>r� )r[ r\ rb rc r� �main_greenlets_before_testr# r: r r� r r0 r= r> )r3 r� r� r� r$ r r r_ r �test_unexpected_reparenting� s
� �z7TestGreenletSetParentErrors.test_unexpected_reparentingc s g � t �� �t �� �� ��fdd�}G � fdd�dt�}t j|d�}|�� ��d� zL|dd� �}| �tj�� }|� � W d � n1 sGw Y | �
d t |j �� | �
d
t |j �� | �
d t |j �� W ��
� |�d� � d d �= d S ��
� |�d� � d d �= w ) Nc s4 t dd� �} � �| � | �� ��� ��d� d S )Nc S r� r r r r r r r� � r� zhTestGreenletSetParentErrors.test_unexpected_reparenting_thread_running.<locals>.worker.<locals>.<lambda>ra )r# r r rz r{ rS �r` �keep_main_alive�switched_to_greenletr r r� � s
zVTestGreenletSetParentErrors.test_unexpected_reparenting_thread_running.<locals>.workerc r )zZTestGreenletSetParentErrors.test_unexpected_reparenting_thread_running.<locals>.convolutedc rb rc rd r� r_ r r r� � re zkTestGreenletSetParentErrors.test_unexpected_reparenting_thread_running.<locals>.convoluted.__getattribute__Nr� r r_ r r r� � r r� rY ra c S r� r r r r r r r� � r� zXTestGreenletSetParentErrors.test_unexpected_reparenting_thread_running.<locals>.<lambda>z#Cannot switch to a different thread�Expected�Current)r[ r� r# r\ rb r{ r: r r� r r< r= r> rz rc )r3 r� r� r� r$ r r rh r �*test_unexpected_reparenting_thread_running� s,
�
�
zFTestGreenletSetParentErrors.test_unexpected_reparenting_thread_runningc C sb t dd� �}| �|jt�� � | �t��
}|`W d � n1 s!w Y | �t|j �d� d S )Nc S r� r r r r r r r�
r� zGTestGreenletSetParentErrors.test_cannot_delete_parent.<locals>.<lambda>�can't delete attribute)
r# rD r r r r: r; r0 r= r> )r3 r� r r r r �test_cannot_delete_parent s �z5TestGreenletSetParentErrors.test_cannot_delete_parentc C sH | � t�� }t�� `W d � n1 sw Y | �t|j�d� d S )Nrn �r: r; r r r r0 r= r> �r3 r r r r �!test_cannot_delete_parent_of_main s
�z=TestGreenletSetParentErrors.test_cannot_delete_parent_of_mainc C s | � t�� j� d S r )rG r r r r6 r r r �!test_main_greenlet_parent_is_none s z=TestGreenletSetParentErrors.test_main_greenlet_parent_is_nonec sT dd� }� fdd�}t |�}� �|� ||� |�� � �|� ||� |�� d S )Nc S r� r r� r r r r �bg s zCTestGreenletSetParentErrors.test_set_parent_wrong_types.<locals>.bgc sb d d� dfD ](}� � t�� }|| _W d � n1 sw Y � �t|j�dt|�j � qd S )Nr �42z4GreenletChecker: Expected any type of greenlet, not )r: r� r r0 r= r> �typer
)�glet�pr r6 r r �check# s � ��zFTestGreenletSetParentErrors.test_set_parent_wrong_types.<locals>.check)r# r� r r� )r3 rt ry r$ r r6 r �test_set_parent_wrong_types s
z7TestGreenletSetParentErrors.test_set_parent_wrong_typesc C sR t dd� �}| �t�� }||_W d � n1 sw Y | �t|j�d� d S )Nc S r� r r r r r r r� ; r� z@TestGreenletSetParentErrors.test_trivial_cycle.<locals>.<lambda>�cyclic parent chain)r# r: r� r r0 r= r> )r3 rw r r r r �test_trivial_cycle: s
�z.TestGreenletSetParentErrors.test_trivial_cyclec C sN | � t��}t�� t�� _W d � n1 sw Y | �t|j�d� d S )Nz(cannot set the parent of a main greenletrp rq r r r �test_trivial_cycle_main@ s �z3TestGreenletSetParentErrors.test_trivial_cycle_mainc C sv t dd� �}t dd� �}t dd� �}||_||_| �t�� }||_W d � n1 s+w Y | �t|j�d� d S )Nc S r� r r r r r r r� G r� z?TestGreenletSetParentErrors.test_deeper_cycle.<locals>.<lambda>c S r� r r r r r r r� H r� c S r� r r r r r r r� I r� r{ )r# r r: r� r0 r= r> )r3 r% rh �g3r r r r �test_deeper_cycleF s �z-TestGreenletSetParentErrors.test_deeper_cycleN)
r
r r r^ rg rm ro rr rs rz r| r} r r r r r r\ � s )$ r\ c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�TestReprc C s | � |�|�||f� d S r )r� �endswith)r3 �got�suffixr r r �assertEndsWithT s zTestRepr.assertEndsWithc C s t t�� �}| �|d� d S )N� current active started main>)�reprr r r� �r3 �rr r r �test_main_while_runningW s z TestRepr.test_main_while_runningc s4 t �� � � fdd�}t|�}|�� }| �|d� d S )Nc s t � �S r )r� r rJ r r r9 ] r� z-TestRepr.test_main_in_background.<locals>.run� suspended active started main>)r r r# r r� )r3 r9 r$ r� r rJ r �test_main_in_background[ s
z TestRepr.test_main_in_backgroundc C s t t� �}| �|d� d S )Nz pending>)r� r# r� r� r r r � test_initiald s
zTestRepr.test_initialc s� t �� � G � fdd�dtj�}|� }|�� |�d� | �|jd� | �|jd� t d�D ]}t
� d� q-t d�D ]}| � |j
j� t|j
�}| �|d� q9d S ) Nc s$ e Zd Zd ZZdZ� fdd�ZdS )z/TestRepr.test_main_from_other_thread.<locals>.TNc s$ t � �| _t�� | _t | j�| _d S r )r� �
original_mainr r � main_gletr3 r6 rJ r r r9 n rI z3TestRepr.test_main_from_other_thread.<locals>.T.run)r
r r r� r3 r� r9 r rJ r r �Tk s r� ra r� r� r( g����MbP?z (thread exited) dead>)r r r[ r\ rb rc r� r� r3 r2 r� r� r� r� rM r� )r3 r� r� r r� r rJ r �test_main_from_other_threadh s
�z$TestRepr.test_main_from_other_threadc C sX t dd� �}|�� | �t|�d� | �dt|�� | �dt|�� | �dt|�� d S )Nc S r� r r r r r r r� � r� z$TestRepr.test_dead.<locals>.<lambda>z dead>� suspended�started�active)r# r r� r� � assertNotInrE r r r � test_dead� s zTestRepr.test_deadc C sX t dd� �}t dd� �}t�� }|||fD ]}| �d|f t� | �d|f t� qd S )Nc S r� r r r r r r r� � r� z>TestRepr.test_formatting_produces_native_str.<locals>.<lambda>c S r� r r r r r r r� � r� z%sz%r)r# r r �assertIsInstancer= )r3 �g_dead�
g_not_started�g_curr$ r r r �#test_formatting_produces_native_str� s ���z,TestRepr.test_formatting_produces_native_strN)
r
r r r� r� r� r� r� r� r� r r r r r� R s # r� c @ r� )�TestMainGreenletc C s4 dt t�� �v s
J �tt�� �}dt |�vsJ �|S )Nr� )r� r r rv )r3 r� r r r �_check_current_is_main� s z'TestMainGreenlet._check_current_is_mainc C s$ | � � }td|fi �}| �|� d S )N�subclass)r� rv rC )r3 � main_typer� r r r �)test_main_greenlet_type_can_be_subclassed� s z:TestMainGreenlet.test_main_greenlet_type_can_be_subclassedc C s | � � | �t�� t� d S r )r� r� r r r# r6 r r r � test_main_greenlet_is_greenlet� s z/TestMainGreenlet.test_main_greenlet_is_greenletN)r
r r r� r� r� r r r r r� � s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�TestBrokenGreenletsc C sT dd� }t j�|�}d|_| �td��
|�� W d � d S 1 s#w Y d S )Nc S s t d��)NzNever get here)�AssertionErrorr r r r r� � r� z<TestBrokenGreenlets.test_failed_to_initialstub.<locals>.funcTz;Failed to switch stacks into a greenlet for the first time.)r � _greenlet�UnswitchableGreenlet�force_switch_error�assertRaisesRegex� SystemErrorr r� r r r �test_failed_to_initialstub� s �
"�z.TestBrokenGreenlets.test_failed_to_initialstubc s� g � � fdd�}t j�|�}|�� | �� dg� |�� | �� ddg� d|_| �td�� |�� W d � n1 s=w Y d|_|�� | �� g d�� d S ) Nc s> � � d� t�� j�� � � d� t�� j�� � � d� d S )Nr r- r( r) r ��runsr r r� � s
zDTestBrokenGreenlets.test_failed_to_switch_into_running.<locals>.funcr r- Tz0Failed to switch stacks into a running greenlet.F)r r- r( )r r� r� r r0 r� r� r� r� r r� r �"test_failed_to_switch_into_running� s �
�z6TestBrokenGreenlets.test_failed_to_switch_into_runningc C s. | � d�}| �d|j� | �|j| �� � d S )Nzfail_slp_switch.pyzfail_slp_switch is running)�assertScriptRaisesr< �output�
returncode�,get_expected_returncodes_for_aborted_process�r3 �exr r r �&test_failed_to_slp_switch_into_running� s
z:TestBrokenGreenlets.test_failed_to_slp_switch_into_runningc C sV | � d�}| �d|� | �d|� | �d|� | �d|� | �d|� | �d|� d S )Nz fail_switch_two_greenlets.pyz In g1_run�TRACEz LEAVE TRACEzFalling off end of mainzFalling off end of g1_runzFalling off end of g2��
run_scriptr< �r3 r� r r r �#test_reentrant_switch_two_greenlets� s
z7TestBrokenGreenlets.test_reentrant_switch_two_greenletsc C s. | j ddd�}| �d|j� | �d|j� d S )Nz fail_switch_three_greenlets.py)r )� exitcodesr� zpositional arguments)r� r< r� r� r r r �%test_reentrant_switch_three_greenlets� s z9TestBrokenGreenlets.test_reentrant_switch_three_greenletsc C � | � d�}| �d|� d S )Nz fail_switch_three_greenlets2.pya� RESULTS: [('trace', 'switch'), ('trace', 'switch'), ('g2 arg', 'g2 from tracefunc'), ('trace', 'switch'), ('main g1', 'from g2_run'), ('trace', 'switch'), ('g1 arg', 'g1 from main'), ('trace', 'switch'), ('main g2', 'from g1_run'), ('trace', 'switch'), ('g1 from parent', 'g1 from main 2'), ('trace', 'switch'), ('main g1.2', 'g1 done'), ('trace', 'switch'), ('g2 from parent', ()), ('trace', 'switch'), ('main g2.2', 'g2 done')]r� r� r r r �&test_reentrant_switch_three_greenlets2 s
�z:TestBrokenGreenlets.test_reentrant_switch_three_greenlets2c C r� )Nz#fail_initialstub_already_started.pya! RESULTS: ['Begin C', 'Switch to b from B.__getattribute__ in C', ('Begin B', ()), '_B_run switching to main', ('main from c', 'From B'), 'B.__getattribute__ back from main in C', ('Begin A', (None,)), ('A dead?', True, 'B dead?', True, 'C dead?', False), 'C done', ('main from c.2', None)]r� r� r r r �4test_reentrant_switch_GreenletAlreadyStartedInPython s
�zHTestBrokenGreenlets.test_reentrant_switch_GreenletAlreadyStartedInPythonc C r� )Nz fail_clearing_run_switches.pyz�RESULTS [('G.__getattribute__', 'run'), ('RunCallable', '__del__'), ('main: g.switch()', 'from RunCallable'), ('run_func', 'enter')]r� r� r r r �*test_reentrant_switch_run_callable_has_del! s
�z>TestBrokenGreenlets.test_reentrant_switch_run_callable_has_delN) r
r r r� r� r� r� r� r� r� r� r r r r r� � s
r� �__main__)rm r r� r[ rZ �abcr r r r# rV r r r � leakcheckr rr r r r& r' r\ r� r� r� r
r� r r r r �<module> s>
y .Um �