a
}|gR @ s d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlZddlm Z ddl
Z
ddl m Z dd Z
d
d Zejd d
Zdd ZG dd dejZG dd deZedkree
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.0sr r N/usr/lib/python3.9/site-packages/certbot/_internal/tests/error_handler_test.py
z get_signals..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 dqS )c s
| S N)append)r _signalsr r
r z,signal_receiver...r r
r r r
r r z#signal_receiver..N)r r )r Z
prev_handlersr r r
signal_receiver s
r c C s t t | dS )zSend the given signalN)r killgetpid)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| jR i | j| _|j | _
d S )Nr
error_handler* ZfooZbar) certbot._internalr" r MagicMock init_func init_args init_kwargsZ ErrorHandlerhandlerZ_SIGNALSr selfr" r r r
setUp- s
zErrorHandlerTest.setUpc C sl d}z.| j tW 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|ksJ | j d g|ksJ | jj| ji | j | j D ]}|| t
|ksJ qd S )N%Signals cannot be handled on Windows.reasonr# r
) r skipTestr r r) r r& r. r' r( r r )r+ Z init_signalssignals_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 tW d n1 s<0 Y W n tyZ Y n0 | jj| ji | j | d S )NZ 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 fddd}| j| t| j >}| j t| W d n1 sv0 Y W d n1 s0 Y | g|ksJ | 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..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 td W d n1 s(0 Y W n tyF Y n0 | jjdu sXJ d S )Nr F)r) sysexit
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_calledr+ 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 fddZdd Z ZS )ExitHandlerTestz6Tests for certbot._internal.error_handler.ExitHandler.c s: ddl m} t |j| jg| jR 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? mainargv__file__r r r r
s&
\