File: //usr/local/lib/python3.7/test/__pycache__/test_atexit.cpython-37.opt-1.pyc
B
��g? � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ dd� Z dd� Z
dd � Z d
d � Z d d
� Z
dd� Zdd� ZG dd� dej�ZejG dd� dej��Zedkr�e�� dS )� N)�support)�
script_helperc C s t d� d S )N�h1)�print� r r �,/usr/local/lib/python3.7/test/test_atexit.pyr
s r c C s t d� d S )N�h2)r r r r r r
s r c C s t d� d S )N�h3)r r r r r r s r c O s t d| |� d S )N�h4)r )�args�kwargsr r r r
s r
c C s t �d S )N)� TypeErrorr r r r �raise1 s r c C s t �d S )N)� SystemErrorr r r r �raise2 s r c C s t �d S )N)�
SystemExitr r r r �exit 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� Zdd � Zd S ) � GeneralTestc C s4 t j| _t j| _t�� | _| j t _t _t� � d S )N)
�sys�stdout� save_stdout�stderr� save_stderr�io�StringIO�stream�atexit�_clear)�selfr r r �setUp" s
zGeneralTest.setUpc C s | j t_| jt_t�� d S )N)r r r r r r r )r r r r �tearDown) s zGeneralTest.tearDownc C sB t �t� t �t� t jtddd� t �� | �| j�� d� d S )N� �abc)�kwz"h4 (4,) {'kw': 'abc'}
h4 () {}
h1
)r �registerr r
�_run_exitfuncs� assertEqualr �getvalue)r r r r � test_args. s
zGeneralTest.test_argsc C s4 t �dd� dddd� dD �dd� | �tt j� d S )Nc S s dS )N� r r r r r �<lambda>9 � z*GeneralTest.test_badargs.<locals>.<lambda>r c s s | ]
}|V qd S )Nr )�.0�xr r r � <genexpr>9 s z+GeneralTest.test_badargs.<locals>.<genexpr>)r) � )r r$ � assertRaisesr
r% )r r r r � test_badargs8 s "zGeneralTest.test_badargsc C s< t �t� t �t� t �t� t �� | �| j�� d� d S )Nz h3
h2
h1
) r r$ r r r r% r& r r' )r r r r �
test_order< s
zGeneralTest.test_orderc C s& t �t� t �t� | �tt j� d S )N)r r$ r r r0 r
r% )r r r r �
test_raiseE s
zGeneralTest.test_raisec C s2 t �dd� � | �tt j� | �d| j�� � d S )Nc S s dd S )Nr) r r r r r r r* O r+ z5GeneralTest.test_raise_unnormalized.<locals>.<lambda>�ZeroDivisionError)r r$ r0 r4 r% �assertInr r' )r r r r �test_raise_unnormalizedL s z#GeneralTest.test_raise_unnormalizedc C s. t �t� | �tt j� | �| j�� d� d S )N� ) r r$ r r0 r r% r& r r' )r r r r � test_exitT s
zGeneralTest.test_exitc C s� dd� }dd� }dd� }t �|� t �|� t �|� | �tt j� | j�� }| �|�d�d� | � d |� | � d
|� | � d |� d S ) Nc S s dd d S )Nr) r r r r r r �f] s z,GeneralTest.test_print_tracebacks.<locals>.fc S s dd d S )Nr) r r r r r r �g_ s z,GeneralTest.test_print_tracebacks.<locals>.gc S s dd d S )Nr) r r r r r r �ha s z,GeneralTest.test_print_tracebacks.<locals>.hr4 � z# onez# twoz# three)
r r$ r0 r4 r% r r' r& �countr5 )r r9 r: r; r r r r �test_print_tracebacks[ s
z!GeneralTest.test_print_tracebacksc sJ dg� � fdd�}xt d�D ]}t�|� q W t�� | �� d d� d S )Nr c s � d d7 < d S )Nr r) r r )�ar r �incp s z$GeneralTest.test_stress.<locals>.inc� )�ranger r$ r% r& )r r@ �ir )r? r � test_stressn s zGeneralTest.test_stressc s@ dg� � fdd�}t �|� t �� t �� | �� d d� d S )Nr c s � d d7 < d S )Nr r) r r )r? r r r@ { s z#GeneralTest.test_clear.<locals>.inc)r r$ r r% r& )r r@ r )r? r �
test_cleary s
zGeneralTest.test_clearc sj dg� � fdd�}� fdd�}xt d�D ]}t�|� q(W t�|� t�|� t�� | �� d d� d S )Nr c s � d d7 < d S )Nr r) r r )r? r r r@ � s z(GeneralTest.test_unregister.<locals>.incc s � d d8 < d S )Nr r) r r )r? r r �dec� s z(GeneralTest.test_unregister.<locals>.decr! ���)rB r r$ �
unregisterr% r& )r r@ rF rC r )r? r �test_unregister� s
zGeneralTest.test_unregisterc C sN g }t �|jd� t �� | �|dg� t �|j� t �� | �|dg� d S )N� )r r$ �appendr% r& rH )r �lr r r �test_bound_methods� s z GeneralTest.test_bound_methodsc C s: d}t �d|�}| �|j�� �� ddg� | �|j� d S )Nz�if 1:
import atexit
def f(msg):
print(msg)
atexit.register(f, "one")
atexit.register(f, "two")
z-cZtwoZone)r Zassert_python_okr& �out�decode�
splitlinesZ assertFalse�err)r �codeZresr r r �
test_shutdown� s
zGeneralTest.test_shutdownN)�__name__�
__module__� __qualname__r r r( r1 r2 r3 r6 r8 r> rD rE rI rM rS r r r r r s
r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�SubinterpreterTestc C s6 t �� }d}t�|�}| �|d� | �t �� |� d S )Nz�if 1:
import atexit
def f():
pass
atexit.register(f)
del atexit
r )r � _ncallbacksr �run_in_subinterpr& )r �nrR �retr r r �test_callbacks_leak� s
z&SubinterpreterTest.test_callbacks_leakc C s6 t �� }d}t�|�}| �|d� | �t �� |� d S )Nz�if 1:
import atexit
def f():
pass
atexit.register(f)
atexit.__atexit = atexit
r )r rX r rY r& )r rZ rR r[ r r r � test_callbacks_leak_refcycle� s
z/SubinterpreterTest.test_callbacks_leak_refcyclec C sT d}t �� \}}d�|�}t�|�}t �|� | �t �|t|��|� t �|� d S )Ns The test has passed!z�if 1:
import os
import atexit
def callback():
os.write({:d}, b"The test has passed!")
atexit.register(callback)
) �os�pipe�formatr rY �closer& �read�len)r Zexpected�r�wrR r[ r r r �(test_callback_on_subinterpreter_teardown� s
z;SubinterpreterTest.test_callback_on_subinterpreter_teardownN)rT rU rV r\ r] rf r r r r rW � s rW �__main__)r Zunittestr r r^ Ztestr Z test.supportr r r r r
r r r ZTestCaser Z cpython_onlyrW rT �mainr r r r �<module> s&
4