← Back
Editing: error_handler_test.cpython-38.pyc
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 \
Save File
Cancel