/usr/lib/python3.9/site-packages/certbot/_internal/tests/__pycache__
a }|�gR � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlZddlm Z ddl Z ddl m Z dd � Z d d � Zejd d � �Zdd� ZG dd� dej�ZG dd� de�Zedkr�e�e �ejdd� eg �� dS )z*Tests for certbot._internal.error_handler.� N)�Callable)�Dict)�Union)�mock)�osc C s dd� | D �S )z,Get the handlers for an iterable of signums.c S s i | ]}|t �|��qS � )�signal� getsignal��.0�sr r �N/usr/lib/python3.9/site-packages/certbot/_internal/tests/error_handler_test.py� <dictcomp> � z get_signals.<locals>.<dictcomp>r )�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 # s4 g � t | �}t� fdd�| D �� � V t|� dS )z Context manager to catch signalsc s i | ]}|� fd d��qS )c s � � | �S �N)�append)r �_��signalsr r �<lambda> r z,signal_receiver.<locals>.<dictcomp>.<lambda>r r r r r r r z#signal_receiver.<locals>.<dictcomp>N)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 )�ErrorHandlerTestz7Tests for certbot._internal.error_handler.ErrorHandler.c C sT ddl m} t�� | _dh| _ddi| _|j| jg| j�R i | j��| _|j | _ d S )Nr �� error_handler�* ZfooZbar) �certbot._internalr" r � MagicMock� init_func� init_args� init_kwargsZ ErrorHandler�handlerZ_SIGNALSr ��selfr" r r r �setUp- s ��zErrorHandlerTest.setUpc C sl d}z.| j � t�W d � n1 s&0 Y W n tyH d}Y n0 |sRJ �| jj| ji | j�� d S )NFT)r) � ValueErrorr& �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 ��P}| j�* d}t| j d � |d9 }W d � n1 s^0 Y W d � n1 s|0 Y d|ks�J �| j d g|ks�J �| jj| ji | j �� | j D ]}|| t � |�ks�J �q�d S )N�%Signals cannot be handled on Windows.��reasonr# r � ) r �skipTestr r r) r r& r. r' r( r r )r+ Z init_signals�signals_receivedZ should_be_42r r r r � test_context_manager_with_signalF s D � z1ErrorHandlerTest.test_context_manager_with_signalc C s~ t jtgd�}| j�|� z.| j� t�W d � n1 s<0 Y W n tyZ Y n0 | jj| ji | j�� |�� d S )N�Z side_effect) r r% r- r) �registerr& r. 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 d � n1 sv0 Y W d � n1 s�0 Y |� g|ks�J �| jj | j i | j �� |� � d S )Nr0 r1 r ���c s t � �S r )r r �Zsig1r r r k r z@ErrorHandlerTest.test_bad_recovery_with_signal.<locals>.<lambda>r7 ) r r4 r r% r) r8 r r r& r. r' r( )r+ Zsig2r9 r5 r r< r � test_bad_recovery_with_signalf s D �z.ErrorHandlerTest.test_bad_recovery_with_signalc C s\ z4| j � t�d� W d � n1 s(0 Y W n tyF Y n0 | jjdu sXJ �d S )Nr F)r) �sys�exit� SystemExitr& Zcalled)r+ r r r �test_sysexit_ignoredu s , z%ErrorHandlerTest.test_sysexit_ignoredc C sP t �� }| j�|� | j� W d � n1 s00 Y | j�� |�� d S r )r r% r) r8 r& Zassert_not_called�r+ �funcr r r �test_regular_exit} s z"ErrorHandlerTest.test_regular_exitN) �__name__� __module__� __qualname__�__doc__r, r/ r6 r: r= rA rD r r r r r * s r c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�ExitHandlerTestz6Tests for certbot._internal.error_handler.ExitHandler.c s: ddl m} t� �� |j| jg| j�R i | j��| _d S )Nr r! ) r$ r" �superr, Z ExitHandlerr& r' r( r) r* �� __class__r r r, � s ��zExitHandlerTest.setUpc C s\ t �� }| j�|� | j� W d � n1 s00 Y | jj| ji | j�� |�� d S r )r r% r) r8 r& r. r' r( rB r r r rD � s �z!ExitHandlerTest.test_regular_exit)rE rF rG rH r, rD � __classcell__r r rK r rI � s rI �__main__� )rH � contextlibr r> �typingr r r Zunittestr ZpytestZcertbot.compatr r r �contextmanagerr r ZTestCaser rI rE r? �main�argv�__file__r r r r �<module> s& \
.
Edit
..
Edit
__init__.cpython-39.opt-1.pyc
Edit
__init__.cpython-39.pyc
Edit
account_test.cpython-39.opt-1.pyc
Edit
account_test.cpython-39.pyc
Edit
auth_handler_test.cpython-39.opt-1.pyc
Edit
auth_handler_test.cpython-39.pyc
Edit
cert_manager_test.cpython-39.opt-1.pyc
Edit
cert_manager_test.cpython-39.pyc
Edit
cli_test.cpython-39.opt-1.pyc
Edit
cli_test.cpython-39.pyc
Edit
client_test.cpython-39.opt-1.pyc
Edit
client_test.cpython-39.pyc
Edit
configuration_test.cpython-39.opt-1.pyc
Edit
configuration_test.cpython-39.pyc
Edit
crypto_util_test.cpython-39.opt-1.pyc
Edit
crypto_util_test.cpython-39.pyc
Edit
eff_test.cpython-39.opt-1.pyc
Edit
eff_test.cpython-39.pyc
Edit
error_handler_test.cpython-39.opt-1.pyc
Edit
error_handler_test.cpython-39.pyc
Edit
errors_test.cpython-39.opt-1.pyc
Edit
errors_test.cpython-39.pyc
Edit
helpful_test.cpython-39.opt-1.pyc
Edit
helpful_test.cpython-39.pyc
Edit
hook_test.cpython-39.opt-1.pyc
Edit
hook_test.cpython-39.pyc
Edit
lock_test.cpython-39.opt-1.pyc
Edit
lock_test.cpython-39.pyc
Edit
log_test.cpython-39.opt-1.pyc
Edit
log_test.cpython-39.pyc
Edit
main_test.cpython-39.opt-1.pyc
Edit
main_test.cpython-39.pyc
Edit
ocsp_test.cpython-39.opt-1.pyc
Edit
ocsp_test.cpython-39.pyc
Edit
renewal_test.cpython-39.opt-1.pyc
Edit
renewal_test.cpython-39.pyc
Edit
renewupdater_test.cpython-39.opt-1.pyc
Edit
renewupdater_test.cpython-39.pyc
Edit
reverter_test.cpython-39.opt-1.pyc
Edit
reverter_test.cpython-39.pyc
Edit
storage_test.cpython-39.opt-1.pyc
Edit
storage_test.cpython-39.pyc
Edit
util_test.cpython-39.opt-1.pyc
Edit
util_test.cpython-39.pyc
Edit