Your IP : 216.73.216.172
U
�]� � @ s� d Z ddlZddlZddlZddlZddlZddlmZmZm Z ddl
m Z dd� Z dd� Z
ejd d
� �Zd d � ZG d
d� dej�ZG dd� de�Zedkr�e�� dS )z Tests for certbot.error_handler.� N)�Callable�Dict�Union)�osc C s t dd� | D ��S )z,Get the handlers for an iterable of signums.c s s | ]}|t �|�fV qd S �N)�signal� getsignal��.0�s� r �B/usr/lib/python3/dist-packages/certbot/tests/error_handler_test.py� <genexpr> s z get_signals.<locals>.<genexpr>)�dict)�signumsr r r
� get_signals s r c C s" | � � D ]\}}t�||� qdS )zDSet the signal (keys) with the handler (values) from the input dict.N)�itemsr )Zsig_handler_dictr �hr r r
� set_signals s r c # s8 g � t | �}tt� fdd�| D ��� � V t|� dS )z Context manager to catch signalsc 3 s | ]}|� fd d�fV qdS )c s
� � | �S r )�append)r �_��signalsr r
�<lambda> � z+signal_receiver.<locals>.<genexpr>.<lambda>Nr r r r r
r s z"signal_receiver.<locals>.<genexpr>N)r r r )r Z
prev_handlersr r r
�signal_receiver s
r c C s t �t �� | � dS )zSend the given signalN)r �kill�getpid)�signumr r r
� send_signal% s r c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�ErrorHandlerTestz-Tests for certbot.error_handler.ErrorHandler.c C sP ddl m} t�� | _td�| _ddi| _|j| jf| j�| j�| _ |j
| _ d S )Nr ��
error_handler)�* ZfooZbar) �certbotr" �mock� MagicMock� init_func�set� init_args� init_kwargsZ ErrorHandler�handlerZ_SIGNALSr ��selfr" r r r
�setUp- s
��zErrorHandlerTest.setUpc C sX d}z| j �
t�W 5 Q R X W n tk
r6 d}Y nX | �|� | jj| j| j� d S )NFT)r+ �
ValueErrorZ
assertTruer' �assert_called_once_withr) r* )r- Zexception_raisedr r r
�test_context_manager: s
�z%ErrorHandlerTest.test_context_managerc
C s� | j s| jdd� t| j �}t| j ��2}| j� d}t| j d � |d9 }W 5 Q R X W 5 Q R X | �d|� | �| j d g|� | jj| j | j
� | j D ]}| �|| t � |�� q�d S )N�%Signals cannot be handled on Windows.��reasonr# r �
)
r �skipTestr r r+ r � assertEqualr' r0 r) r* r r )r- Z init_signals�signals_receivedZ should_be_42r r r r
� test_context_manager_with_signalF s
�
z1ErrorHandlerTest.test_context_manager_with_signalc C sh t jtgd�}| j�|� z| j�
t�W 5 Q R X W n tk
rH Y nX | jj| j| j� |�� d S )N�Z side_effect) r% r& r/ r+ �registerr' r0 r) r* )r- �bad_funcr r r
�test_bad_recoveryZ s
�z"ErrorHandlerTest.test_bad_recoveryc
s� | j s| jdd� | j d � | j d }tj� fdd�d�}| j�|� t| j �� }| j� t|� W 5 Q R X W 5 Q R X | �|� g|� | j j
| j | j � |�
� d S )Nr2 r3 r ���c s t � �S r )r r �Zsig1r r
r k r z@ErrorHandlerTest.test_bad_recovery_with_signal.<locals>.<lambda>r: )
r r6 r% r& r+ r; r r r7 r' r0 r) r* )r- Zsig2r< r8 r r? r
� test_bad_recovery_with_signalf s
�z.ErrorHandlerTest.test_bad_recovery_with_signalc C sH z | j � t�d� W 5 Q R X W n tk
r4 Y nX | �| jj� d S )Nr )r+ �sys�exit�
SystemExitZ assertFalser' Zcalled)r- r r r
�test_sysexit_ignoredu s z%ErrorHandlerTest.test_sysexit_ignoredc C s< t �� }| j�|� | j� W 5 Q R X | j�� |�� d S r )r% r& r+ r; r' Zassert_not_called�r- �funcr r r
�test_regular_exit} s
z"ErrorHandlerTest.test_regular_exitN) �__name__�
__module__� __qualname__�__doc__r. r1 r9 r= r@ rD rG r r r r
r * s
r c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�ExitHandlerTestz,Tests for certbot.error_handler.ExitHandler.c s8 ddl m} tt| ��� |j| jf| j�| j�| _ d S )Nr r! )
r$ r" �superrL r. Z ExitHandlerr' r) r* r+ r, �� __class__r r
r. � s
��zExitHandlerTest.setUpc C sD t �� }| j�|� | j� W 5 Q R X | jj| j| j� |�� d S r )r% r& r+ r; r' r0 r) r* rE r r r
rG � s
�z!ExitHandlerTest.test_regular_exit)rH rI rJ rK r. rG �
__classcell__r r rN r
rL � s rL �__main__)rK �
contextlibr rA Zunittestr% Zacme.magic_typingr r r Zcertbot.compatr r r �contextmanagerr r ZTestCaser rL rH �mainr r r r
�<module> s
\