U
    [¹Öh°
  ã                   @   sx   d dl mZ d dl mZ d dlZd dlZd dlZddlmZ ddlmZ ddlm	Z	 G dd	„ d	eƒZ
ed
krte ¡  dS )é    )Úprint_function)Úabsolute_importNé   )Ú_test_extension_cpp)ÚTestCase)ÚWINc                   @   sH   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Ze e	d	¡d
d„ ƒZ
dd„ ZdS )ÚCPPTestsc                 C   sX   g }t dƒD ]$}t tj¡}| |¡ | |¡ qt|ƒD ]\}}|  | ¡ |¡ q:d S )Né   )ÚrangeÚgreenletr   Útest_exception_switchZswitchÚappendÚ	enumerateZassertEqual)ÚselfZ	greenletsÚiÚg© r   ú=./venv/lib/python3.8/site-packages/greenlet/tests/test_cpp.pyr      s    
zCPPTests.test_exception_switchc           
   	   C   s    dd l }dd l}|j |j t¡d¡}|j|t|tƒs<|j	n|g}|}|  
tj¡}tj|dtjd W 5 Q R X |j}|  ¡ }	|  |j|	¡ |  d|j¡ |jS )Nr   zfail_cpp_exception.pyzutf-8)ÚencodingÚstderrzfail_cpp_exception is running)ÚosÚsysÚpathÚjoinÚdirnameÚ__file__Ú
executableÚ
isinstanceÚstrÚ__name__ZassertRaisesÚ
subprocessZCalledProcessErrorZcheck_outputZSTDOUTZ	exceptionZ,get_expected_returncodes_for_aborted_processÚassertInÚ
returncodeÚoutput)
r   Útargetr   r   ZscriptÚargsZ__traceback_info__ÚexcÚexZexpected_exitr   r   r   Ú_do_test_unhandled_exception   s&    
þýz%CPPTests._do_test_unhandled_exceptionc                 C   s   |   tj¡ d S ©N)r(   r   Ztest_exception_throw_nonstd©r   r   r   r   Ú&test_unhandled_nonstd_exception_aborts-   s    z/CPPTests.test_unhandled_nonstd_exception_abortsc                 C   s   |   tj¡ d S r)   )r(   r   Ztest_exception_throw_stdr*   r   r   r   Ú#test_unhandled_std_exception_aborts1   s    z,CPPTests.test_unhandled_std_exception_abortsz#XXX: This does not crash on Windowsc                 C   s   |   d¡}|  d|¡ d S )NZrun_as_greenlet_targetzThrown from an extension.)r(   r!   )r   r#   r   r   r   Ú8test_unhandled_std_exception_as_greenlet_function_aborts5   s
    
úzACPPTests.test_unhandled_std_exception_as_greenlet_function_abortsc                 C   s   |   d¡ d S )NZ*run_unhandled_exception_in_greenlet_aborts)r(   r*   r   r   r   Ú+test_unhandled_exception_in_greenlet_abortsC   s    z4CPPTests.test_unhandled_exception_in_greenlet_abortsN)r   Ú
__module__Ú__qualname__r   r(   r+   r,   ÚunittestZskipIfr   r-   r.   r   r   r   r   r      s   	

r   Ú__main__)Z
__future__r   r   r    r1   r   Ú r   r   r   r   r   Úmainr   r   r   r   Ú<module>   s   <