관리-도구
편집 파일: test_greenlet.cpython-38.pyc
U �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 �Zedk�re�� dS )� N)�ABCMeta)�abstractmethod)�greenlet� )�TestCase)�RUNNING_ON_MANYLINUX)�PY313)�fails_leakcheckc @ s e Zd ZdS )� SomeErrorN��__name__� __module__� __qualname__� r r �L/opt/hc_python/lib64/python3.8/site-packages/greenlet/tests/test_greenlet.pyr s r c C s< zt �� j�� W n | �t�� d � � Y nX t�d S �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 | �d S �Nr )�excr r r �crasher, s 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�eozed�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�e�oRedK�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 s | � � d S r )r1 �r0 r r r � test_simpleC s zTestGreenlet.test_simplec C s: t � }| �t��}|�� W 5 Q R X | �dt|j�� d S �N�run)r � assertRaises�AttributeErrorr �assertIn�str� exception�r0 r! r r r r �(test_switch_no_run_raises_AttributeErrorF s z5TestGreenlet.test_switch_no_run_raises_AttributeErrorc C s< t � }| �t��}|�t� W 5 Q R X | �dt|j�� d S r4 )r r6 r7 �throwr r8 r9 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 �r0 r! r r r �test_parent_equals_NoneT s z$TestGreenlet.test_parent_equals_Nonec C s$ t d d�}| �|� | �|j� d S )N)r5 )r r? � assertIsNoner5 rA r r r �test_run_equals_NoneY s 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 r- �len�dead)r0 r) r! �hr r' r �test_two_children^ s z TestGreenlet.test_two_childrenc sh g ��fdd�� � �fdd�}t |�}| �t�|�d� |�� | ��dddg� | �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* rL rK rM )r r- r � getrefcountr rA r rO 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)r1 r r )r0 �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>� )r/ �start�joinr- rG )r0 Zths�thr )r) r0 rS 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 r- r6 r )r0 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 r6 �KeyErrorr# r- )r0 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 r- �gc�collectr � GreenletExit)r0 r r" rb r r r � test_dealloc� s zTestGreenlet.test_deallocc C sv dd� }t |�}|�� tj}ddlm} |� }t_z~W 5 |t_X |�� }| �d|� | �d|� | �d|� d S )Nc S s4 zt �� j�� W n t jk r. td �Y nX d S r )r r r r ri r r r r r r5 � s zHTestGreenlet.test_dealloc_catches_GreenletExit_throws_other.<locals>.runr )�StringIO� ExceptionZignoredr )r r r �stderr�iork �getvaluer8 )r0 r5 r! Z oldstderrrk rm �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 5 �� � |� d� �d d �= �d d �= X d S )Nc sZ t t�} | ��� ��| � ~ t�� � �� ��d� t � ��� ��d� t � d S )Nr% )r r r r rg rh �set�wait)r" �Z/bg_glet_created_running_and_no_longer_ref_in_bgZbg_should_be_clearZfg_ref_releasedZok_to_exit_bg_threadr Zsomerefr r r) � s z1TestGreenlet.test_dealloc_other_thread.<locals>.frT r\ r r% ) rV �EventrW r] rs r- rG rg rh rr r^ r ri )r0 r) �tr rt r �test_dealloc_other_thread� s. z&TestGreenlet.test_dealloc_other_threadc sf � fdd�}t |�}|�� }� �||jk� � �|� |�� }� �|� � �|d� � �|jd � d S )Nc s, t �d�} � �| jd � t�� j�| � dS )Nr �meaning of life)r � _getframer- �f_backr r r r rZ r2 r r �f1 s z#TestGreenlet.test_frame.<locals>.f1rx )r r � assertTrue�gr_frame� assertFalser- )r0 r{ r! �frameZfrom_gr r2 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�������?�rU �args)g333333�?r\ )rV rW r] r^ )r0 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* �* )r- �rL rK r2 r r r5 . s z,TestGreenlet.test_switch_kwargs.<locals>.runr+ r* r� r� �r r r- )r0 r5 r� r r2 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 r5 6 s z6TestGreenlet.test_switch_kwargs_to_parent.<locals>.runr� r% )r* )r% � )r r- r )r0 r5 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 5 Q R X �� � |� d� �� � d S )Nc s& t dd� ��d< � �� ��d� d S )Nc S s d S r r r r r r r� E r� zITestGreenlet.test_switch_to_another_thread.<locals>.run.<locals>.<lambda>r! r\ �r rr rs r �Z created_event�data� done_eventr r r5 D s z7TestGreenlet.test_switch_to_another_thread.<locals>.runrT r\ r! ) rV ru rW r] rs r6 r �errorr rr r^ �clear)r0 r5 �threadr r� r � test_switch_to_another_thread? s z*TestGreenlet.test_switch_to_another_threadc s* � �fdd�}�fdd�� t |��� d S )Nc sB z t d��W n0 t�� } t� ��� ��| t�� � Y nX d S )NZfun)� ValueErrorr r r r r- )r �rI r0 r r r) S s z&TestGreenlet.test_exc_state.<locals>.fc s � � t�� d� d S )N)NNN)r- r r r r2 r r rI [ s z&TestGreenlet.test_exc_state.<locals>.hr )r0 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 s z*TestGreenlet.test_instance_dict.<locals>.fc S s | ` d S r ��__dict__rN r r r �deldictc s z0TestGreenlet.test_instance_dict.<locals>.deldictc S s || _ d S r r� )r! rc r r r �setdicte s z0TestGreenlet.test_instance_dict.<locals>.setdictr� r� )r r- r� r r� r6 � TypeError)r0 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 r4 )r �hasattrr r r �Zhas_runr r �funcs s �z;TestGreenlet.test_running_greenlet_has_no_run.<locals>.funcFr� �r0 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 )�copyr6 r� r �deepcopy)r0 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 s d S r r r r r r r� � r� z;TestGreenlet.test_parent_restored_on_kill.<locals>.<lambda>c sN z �� � W n< tjk rH ��t�� j� ��t�� � � � � Y nX d S r )r r ri 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| r@ r )r0 r� r! r r� r � test_parent_restored_on_kill� s z)TestGreenlet.test_parent_restored_on_killc C s8 t � }t dd� |d�}| �t�� |�� W 5 Q R X d S )Nc S s d S r r r r r r r� � r� z9TestGreenlet.test_parent_return_failure.<locals>.<lambda>r )r r6 r7 r )r0 r" rb 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 @ s e Zd Zdd� ZdS )z>TestGreenlet.test_throw_exception_not_lost.<locals>.mygreenletc S s" zt �W n Y nX t�| |�S r )rl r �__getattribute__�r0 �namer r r r� � s zOTestGreenlet.test_throw_exception_not_lost.<locals>.mygreenlet.__getattribute__N�r r r r� r r r r � mygreenlet� s r� c S s d S r r r r r r r� � r� z<TestGreenlet.test_throw_exception_not_lost.<locals>.<lambda>)r r6 r r= )r0 r� r! r r r � test_throw_exception_not_lost� s z*TestGreenlet.test_throw_exception_not_lostFc s0 g � dd� �� ��fdd�}t j|d�}|�� |�d� ~�rH| �� | �tj��}� d �t � W 5 Q R X �s�t |j �}| � |dkp�d |k� n| � t |j �d� t� d jd �r�| �t�� � d j�� W 5 Q R X ��st� d jjd ��r� d jj�� n| �� d j� ~d �� d d �= d | _d S ) Nc S s t �� j�� d S r r� r r r r r� � s zGTestGreenlet._do_test_throw_to_dead_thread_doesnt_crash.<locals>.workerc s2 t ��} | �� � �| � �r.| �� t�� d S r )r r r r r rN �r� �wait_for_cleanupr� r r �creator� s zHTestGreenlet._do_test_throw_to_dead_thread_doesnt_crash.<locals>.creatorrT r\ r �Bcannot switch to a different thread (which happens to have exited)z Cannot switchr� T)rV rW r] r^ �wait_for_pending_cleanupsr6 r r� r= r r9 r: r| r- r� r} � RuntimeErrorr� �f_localsrC �expect_greenlet_leak)r0 r� r� rv r �sr r� r �*_do_test_throw_to_dead_thread_doesnt_crash� s@ �� z7TestGreenlet._do_test_throw_to_dead_thread_doesnt_crashc C s | � � d S r �r� r2 r r r �&test_throw_to_dead_thread_doesnt_crash� s z3TestGreenlet.test_throw_to_dead_thread_doesnt_crashc C s | � d� d S rR r� r2 r r r �+test_throw_to_dead_thread_doesnt_crash_wait� s z8TestGreenlet.test_throw_to_dead_thread_doesnt_crash_waitc C s2 G dd� dt �}|� }| �|�d�d� d| _d S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z7TestGreenlet.test_recursive_startup.<locals>.convolutedc S s t �| � d| _d S r )r �__init__�countr2 r r r r� � s z@TestGreenlet.test_recursive_startup.<locals>.convoluted.__init__c S s. |dkr"| j dkr"d| _ | �d� t�| |�S )Nr5 r r �+ )r� r r r� r� r r r r� � s zHTestGreenlet.test_recursive_startup.<locals>.convoluted.__getattribute__c S s | j �|� q d S r �r r �r0 rc r r r r5 � s z;TestGreenlet.test_recursive_startup.<locals>.convoluted.runN)r r r r� r� r5 r r r r � convoluted� s r� r� r� T)r r- r r� )r0 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�acquirer2 ��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 s t �� j�� d S r r� r r r r � deallocator s 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 rT r\ )rV �Lockr� �objectr r r �bombr rW r] r� r- r^ )r0 r� r� r� r� rv 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 sJ t �� j�� } �d= t �� j� _zt �� j�| � W 5 ��t �� � X d S r )r r r r r )rc �� initiatorr r� r r r� 8 s z>TestGreenlet.test_dealloc_switch_args_not_lost.<locals>.workerc S s dS r� r r r r r r� E s zATestGreenlet.test_dealloc_switch_args_not_lost.<locals>.initiatorr r� )r r r| r- 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 | ||� d S r r )r� rL �kr r r �_apply^ s z0TestGreenlet.test_tuple_subclass.<locals>._applyc @ s e Zd Zdd� ZdS )z1TestGreenlet.test_tuple_subclass.<locals>.mytuplec S s t �� �� t�| �S r )r r r �tuple�__len__r2 r r r r� b s z9TestGreenlet.test_tuple_subclass.<locals>.mytuple.__len__N)r r r r� r r r r �mytuplea s r� r� )rL c s � t �� jj��� d S r r� r �r� r� �kwargsr r � switchapplyg s z5TestGreenlet.test_tuple_subclass.<locals>.switchapply)r� �dictr r- r )r0 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�AbstractSubclassr5 c S s d S r r r2 r r r r� p r� z7TestGreenlet.test_abstract_subclasses.<locals>.<lambda>c @ s e Zd ZdS )z:TestGreenlet.test_abstract_subclasses.<locals>.BadSubclassNr r r r r � BadSubclassr s r� c @ s e Zd Zdd� ZdS )z;TestGreenlet.test_abstract_subclasses.<locals>.GoodSubclassc S s d S r r r2 r r r r5 v s z?TestGreenlet.test_abstract_subclasses.<locals>.GoodSubclass.runN)r r r r5 r r r r � GoodSubclassu s r� )r r r r6 r� )r0 r� r� r� r r r �test_abstract_subclassesl s �z%TestGreenlet.test_abstract_subclassesc sN t �� s d S t �� d � � dk r$d S � �fdd�}td�D ]}|� r: qJq: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 r� q�|� |� q�t � � �d r�|� d � dS ��|j|� |D ]}��|j|� �qd S ) NFc s � � � t�� ��� d S r r� r r� r r �another_thread� s zWTestGreenlet.test_implicit_parent_with_threads.<locals>.attempt.<locals>.another_threadrT 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� r2 �r� r� �recycledr r r� � s z\TestGreenlet.test_implicit_parent_with_threads.<locals>.attempt.<locals>.gc_callback.__del__Nr� r r� r r � gc_callback� s r� c s e Zd Z� fdd�ZdS )zPTestGreenlet.test_implicit_parent_with_threads.<locals>.attempt.<locals>.garbagec s | | _ � � | _d S r )�cycle�callbackr2 �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� s r� r* r r\ T)rV r� r� rW r] r� r/ r r r^ r r rg rh r- r ) r� rv r� �lr� �currentr! �_�last��Nr0 )r� r� r� r� r �attempt� s> z?TestGreenlet.test_implicit_parent_with_threads.<locals>.attemptr, )rg � isenabledZ get_thresholdr/ )r0 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 ]}||��� qP�`�� t�|�|� d S ) Nr �r )ri c @ s e Zd ZdS )zTTestGreenlet.test_issue_245_reference_counting_subclass_no_threads.<locals>.GreenletNr r r r r �Greenlet� s r c s8 z�� j �� W n" � k r2 �j��� � Y nX d S r )r r �gletsr r �ri r r0 r r � greenlet_main� s zYTestGreenlet.test_issue_245_reference_counting_subclass_no_threads.<locals>.greenlet_mainr\ ) r r ri r r rP r r/ r r- )r0 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 �}|� |� |� � qx|D ]} | � d � q��d d �= �� � | � � | �t�� �|� d S )Nr )rW )ru r c @ s e Zd ZdS )zSTestGreenlet.test_issue_245_reference_counting_subclass_threads.<locals>.MyGreenletNr r r r r � MyGreenlet� s 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 )Nr\ )r r rr rs )Zgreenlet_running_eventZmine�r r r r Z ref_clearedr r � thread_main� s zTTestGreenlet.test_issue_245_reference_counting_subclass_threads.<locals>.thread_main�- r� r\ )rV rW ru r r r r rP r/ r r] rs rr r� r- ) r0 rW ru r r Zthread_ready_eventsr� �eventr� r� r r r �2test_issue_245_reference_counting_subclass_threads� s. � z?TestGreenlet.test_issue_245_reference_counting_subclass_threadsc C sD dd� }t |�}dd� }t ||�}| �t�� |�� W 5 Q R X d S )Nc S s dS )N� r r r r r �no_args s z\TestGreenlet.test_falling_off_end_switches_to_unstarted_parent_raises_error.<locals>.no_argsc S s dS r� r r r r r �leaf" s zYTestGreenlet.test_falling_off_end_switches_to_unstarted_parent_raises_error.<locals>.leaf)r r6 r� r )r0 r �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, s zUTestGreenlet.test_falling_off_end_switches_to_unstarted_parent_works.<locals>.one_argc S s dS r� r r r r r r 1 s zRTestGreenlet.test_falling_off_end_switches_to_unstarted_parent_works.<locals>.leaf)r� r r� )r0 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 5 Q R X | �� dg� | �t�� |�� W 5 Q R X | �� dg� d S )Nc @ s e Zd Zdd� ZdS )zXTestGreenlet.test_switch_to_dead_greenlet_with_unstarted_perverse_parent.<locals>.Parentc S s |dkr t �d S r4 )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: s r c s � � 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 r6 r r r- )r0 r 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 s � � d�S r r r r r r r� P r� zDTestGreenlet.test_switch_to_dead_greenlet_reparent.<locals>.<lambda>c s � � d�S r� r r r r r r� Q r� r� r )r r r- r rC )r0 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 s zHTestGreenlet.test_can_access_f_back_of_suspended_greenlet.<locals>.outerc s � � t�d�� d S r )r r ry r �r� r r r e s zHTestGreenlet.test_can_access_f_back_of_suspended_greenlet.<locals>.innerc S s d S r r r r r r r� n r� zKTestGreenlet.test_can_access_f_back_of_suspended_greenlet.<locals>.<lambda>r r! ) r r r r r? r} r- �f_code�co_namerz rC )r0 r! r� � unrelatedr �r 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 }qH| �|j d d� | � |j � | �|�d �d � d S ) Nr )�partialr )�_test_extension_cppc s. | dkr | � � ��| d �� S t�� j�� S )Nr r )Z test_callr r r r )rp �r) r( �recurser r r+ } s z@TestGreenlet.test_get_stack_with_nested_c_calls.<locals>.recurser, rp r\ i� ) � functoolsr( � r) r r r} r/ r- r� rz rC )r0 �grr rY r r* 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 ry r r r r r! � s z6TestGreenlet.test_frames_always_exposed.<locals>.outerc s � � | � d S r )r )r r" r r r � s z6TestGreenlet.test_frames_always_exposed.<locals>.innerc S s d S r r r r r r r� � r� z9TestGreenlet.test_frames_always_exposed.<locals>.<lambda>r! ) r r r r r- r# r$ rC rz )r0 r! r. r r% r r& r �test_frames_always_exposed� s z'TestGreenlet.test_frames_always_exposedN)F)4r r r r1 r3 r< r> rB rD rJ rQ r` rd rf rj rq �unittestZskipIfr r rw 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 r r r r' r/ r0 r r r r r$ 2 sn � , 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 5 Q R X �� � |� d� | � t |j�d� d S ) Nc s& t dd� ��d< � �� ��d� d S )Nc S s d S r r r r r r r� � r� zQTestGreenletSetParentErrors.test_threaded_reparent.<locals>.run.<locals>.<lambda>r! r\ r� r r� r r r5 � s z?TestGreenletSetParentErrors.test_threaded_reparent.<locals>.runc S s t �� j�� d S r r� r r r r �blank� s zATestGreenletSetParentErrors.test_threaded_reparent.<locals>.blankrT r\ r! z&parent cannot be on a different thread)rV ru rW r] rs r r r6 r� r rr r^ r- r9 r: )r0 r5 r3 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 5 Q R X | � t |j �d � � d d �= d S ) Nc s" t dd� �} � �| � | �� d S )Nc S s d S r r r r r r r� � r� zYTestGreenletSetParentErrors.test_unexpected_reparenting.<locals>.worker.<locals>.<lambda>)r r r rN ��anotherr r r� � s zGTestGreenletSetParentErrors.test_unexpected_reparenting.<locals>.workerrT r\ r )Zinitial_main_greenletsc s e Zd Z� fdd�ZdS )zKTestGreenletSetParentErrors.test_unexpected_reparenting.<locals>.convolutedc s |dkr� d | _ t�| |�S �Nr5 r �r r r� r� r5 r r r� � s z\TestGreenletSetParentErrors.test_unexpected_reparenting.<locals>.convoluted.__getattribute__Nr� r r5 r r r� � s r� c S s d S r r r r r r r� � r� zITestGreenletSetParentErrors.test_unexpected_reparenting.<locals>.<lambda>r� )rV rW r] r^ r� Zmain_greenlets_before_testr r6 r r� r r- r9 r: )r0 r� rv r� r! r r r5 r �test_unexpected_reparenting� s �z7TestGreenletSetParentErrors.test_unexpected_reparentingc s� g � t �� �t �� �� ��fdd�}G � fdd�dt�}t j|d�}|�� ��d� zf|dd� �}| �t j ��}|� � W 5 Q R X | � d t |j�� | � d t |j�� | � d t |j�� W 5 ��� |�d� � d d �= X d S ) Nc s4 t dd� �} � �| � | �� ��� ��d� d S )Nc S s d S r r r r r r r� � r� zhTestGreenletSetParentErrors.test_unexpected_reparenting_thread_running.<locals>.worker.<locals>.<lambda>r\ )r r r rr rs rN �r6 Zkeep_main_aliveZswitched_to_greenletr r r� � s zVTestGreenletSetParentErrors.test_unexpected_reparenting_thread_running.<locals>.workerc s e Zd Z� fdd�ZdS )zZTestGreenletSetParentErrors.test_unexpected_reparenting_thread_running.<locals>.convolutedc s |dkr� d | _ t�| |�S r7 r8 r� r5 r r r� � s zkTestGreenletSetParentErrors.test_unexpected_reparenting_thread_running.<locals>.convoluted.__getattribute__Nr� r r5 r r r� � s r� rT r\ c S s d S r r r r r r r� � r� zXTestGreenletSetParentErrors.test_unexpected_reparenting_thread_running.<locals>.<lambda>z#Cannot switch to a different threadZExpectedZCurrent)rV ru r rW r] rs rr r^ r6 r r� r r8 r9 r: )r0 r� r� rv r! r r r: r �*test_unexpected_reparenting_thread_running� s$ zFTestGreenletSetParentErrors.test_unexpected_reparenting_thread_runningc C sN t dd� �}| �|jt�� � | �t�� }|`W 5 Q R X | �t|j �d� d S )Nc S s d S r r r r r r r� r� zGTestGreenletSetParentErrors.test_cannot_delete_parent.<locals>.<lambda>�can't delete attribute) r r@ r r r r6 r7 r- r9 r: )r0 r� r r r r �test_cannot_delete_parent s z5TestGreenletSetParentErrors.test_cannot_delete_parentc C s4 | � t��}t�� `W 5 Q R X | �t|j�d� d S )Nr< �r6 r7 r r r r- r9 r: �r0 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 )rC r r r r2 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 s t �� j�� d S r r� r r r r �bg s zCTestGreenletSetParentErrors.test_set_parent_wrong_types.<locals>.bgc sN d d� dfD ]<}� � t�� }|| _W 5 Q R X � �t|j�dt|�j � q d S )Nr Z42z4GreenletChecker: Expected any type of greenlet, not )r6 r� r r- r9 r: �typer )�glet�pr r2 r r �check# s �zFTestGreenletSetParentErrors.test_set_parent_wrong_types.<locals>.check)r r~ r r| )r0 rB rF r! r r2 r �test_set_parent_wrong_types s z7TestGreenletSetParentErrors.test_set_parent_wrong_typesc C s> t dd� �}| �t�� }||_W 5 Q R X | �t|j�d� d S )Nc S s d S r r r r r r r� ; r� z@TestGreenletSetParentErrors.test_trivial_cycle.<locals>.<lambda>�cyclic parent chain)r r6 r� r r- r9 r: )r0 rD r r r r �test_trivial_cycle: s z.TestGreenletSetParentErrors.test_trivial_cyclec C s: | � t��}t�� t�� _W 5 Q R X | �t|j�d� d S )Nz(cannot set the parent of a main greenletr> r? r r r �test_trivial_cycle_main@ s z3TestGreenletSetParentErrors.test_trivial_cycle_mainc C sb t dd� �}t dd� �}t dd� �}||_||_| �t�� }||_W 5 Q R X | �t|j�d� d S )Nc S s d S r r r r r r r� G r� z?TestGreenletSetParentErrors.test_deeper_cycle.<locals>.<lambda>c S s d S r r r r r r r� H r� c S s d S r r r r r r r� I r� rH )r r r6 r� r- r9 r: )r0 r" rb Zg3r r r r �test_deeper_cycleF s z-TestGreenletSetParentErrors.test_deeper_cycleN) r r r r4 r9 r; r= r@ rA rG rI rJ rK r r r r r2 � s )$ r2 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)r0 �got�suffixr r r �assertEndsWithT s zTestRepr.assertEndsWithc C s t t�� �}| �|d� d S )N� current active started main>)�reprr r rP �r0 �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 )rR r r" r r r5 ] s z-TestRepr.test_main_in_background.<locals>.run� suspended active started main>)r r r r rP )r0 r5 r! rT r r" r �test_main_in_background[ s z TestRepr.test_main_in_backgroundc C s t t� �}| �|d� d S )Nz pending>)rR r rP rS r r r � test_initiald s zTestRepr.test_initialc s� t �� � G � fdd�dtj�}|� }|�� |�d� | �|jd� | �|jd� t d�D ]}t � d� qZt d�D ](}| � |j j� t|j �}| �|d� qrd 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 )rR � original_mainr r � main_gletr r2 r" r r r5 n s z3TestRepr.test_main_from_other_thread.<locals>.T.run)r r r rY r rZ r5 r r" r r �Tk s r[ r\ rV rQ r% g����MbP?z (thread exited) dead>)r r rV rW r] r^ rP rY r r/ r� r� r| rZ rH rR )r0 r[ rv r� rT r r" 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 s d S r r r r r r r� � r� z$TestRepr.test_dead.<locals>.<lambda>z dead>Z suspended�startedZactive)r r rP rR Z assertNotInrA r r r � test_dead� s zTestRepr.test_deadc C sX t dd� �}t dd� �}t�� }|||fD ](}| �d|f t� | �d|f t� q*d S )Nc S s d S r r r r r r r� � r� z>TestRepr.test_formatting_produces_native_str.<locals>.<lambda>c S s d S r r r r r r r� � r� z%sz%r)r r r �assertIsInstancer9 )r0 Zg_deadZ g_not_startedZg_curr! r r r �#test_formatting_produces_native_str� s ��z,TestRepr.test_formatting_produces_native_strN) r r r rP rU rW rX r\ r^ r` r r r r rL R s #rL c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestMainGreenletc C s4 dt t�� �kst�tt�� �}dt |�ks0t�|S )Nr� )rR r r �AssertionErrorrC )r0 rv r r r �_check_current_is_main� s z'TestMainGreenlet._check_current_is_mainc C s$ | � � }td|fi �}| �|� d S )N�subclass)rc rC r? )r0 Z main_typerd 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 )rc r_ r r r r2 r r r � test_main_greenlet_is_greenlet� s z/TestMainGreenlet.test_main_greenlet_is_greenletN)r r r rc re rf r r r r ra � s ra c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�TestBrokenGreenletsc C s> dd� }t j�|�}d|_| �td�� |�� W 5 Q R X d S )Nc S s t d��d S )NzNever get here)rb r r r r r� � s 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 5 Q R X d|_|�� | �� dddg� 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.Fr% )r rh ri r r- rj rk rl r� r rn 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)�assertScriptRaisesr8 �output� returncodeZ,get_expected_returncodes_for_aborted_process�r0 �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_scriptr8 �r0 rr 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 )Z exitcodesr� zpositional arguments)rq r8 rr rt r r r �%test_reentrant_switch_three_greenlets� s z9TestBrokenGreenlets.test_reentrant_switch_three_greenletsc C s | � 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')]rx rz r r r �&test_reentrant_switch_three_greenlets2 s �z:TestBrokenGreenlets.test_reentrant_switch_three_greenlets2c C s | � d�}| �d|� d S )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)]rx rz r r r �4test_reentrant_switch_GreenletAlreadyStartedInPython s �zHTestBrokenGreenlets.test_reentrant_switch_GreenletAlreadyStartedInPythonc C s | � d�}| �d|� d S )Nz fail_clearing_run_switches.pyz�RESULTS [('G.__getattribute__', 'run'), ('RunCallable', '__del__'), ('main: g.switch()', 'from RunCallable'), ('run_func', 'enter')]rx rz r r r �*test_reentrant_switch_run_callable_has_del! s �z>TestBrokenGreenlets.test_reentrant_switch_run_callable_has_delN) r r r rm rp rv r{ r| r} r~ r r r r r rg � s rg �__main__)rg r r� rV r1 �abcr r r r r- r r r Z leakcheckr rl r r r# r$ r2 rL ra rg r r� r r r r �<module> s: y .Um