a
}|g3O @ s d Z ddlZddlZddlmZ ddlZddlZddlmZ ddl m
Z
ddl m Z ddl
mZ ddlmZ dd lmZ dd
lmZ dd lmZ ddlm mZ eje d Z G d
d dej Z G dd dej Z!G dd dej Z"G dd dej Z#G dd dej Z$G dd dej Z%G dd dej Z&G dd d ej Z'G d d d ej Z(e)d e)d G d!d" d"ej Z*e+d#kre,e-ej.d$d e/g dS )%zTest certbot.display.ops. N)mock)messages)errors)account)obj)
filesystem)os)ops)utilzrsa512_key.pemc @ s| e Zd ZdZedd Ze dd Ze dd Z e dd Z
e d
d Z e d d
Z e dd Z
dS ) GetEmailTestz(Tests for certbot.display.ops.get_email.c K s ddl m} |f i |S )Nr ) get_email)certbot.display.opsr )clskwargsr r L/usr/lib/python3.9/site-packages/certbot/_internal/tests/display/ops_test.py_call s zGetEmailTest._callc C s | j }tjdf|_ttj | W d n1 s>0 Y ttj | jdd W d n1 sv0 Y d S )N foo@bar.bazFZoptional) input display_utilCANCEL return_valuepytestraisesr Errorr )selfmock_get_utility
mock_inputr r r test_cancel_none s &z GetEmailTest.test_cancel_nonec C sX | j }tjdf|_td&}d|_| dks6J W d n1 sJ0 Y d S )Nr #certbot.display.ops.util.safe_emailT)r r OKr r patchr r r r mock_safe_emailr r r test_ok_safe( s
zGetEmailTest.test_ok_safec C s\ | j }tjdf|_td*}ddg|_| dks:J W d n1 sN0 Y d S )Nr r FT)r r r! r r r" side_effectr r# r r r test_ok_not_safe0 s
z GetEmailTest.test_ok_not_safec C s d}| j }tjdf|_tdV}d|_| ||jd d vsHJ | jdd ||jd d v sjJ W d n1 s~0 Y d S )NThere seem to be problemsr r Tr Zinvalid)r r r! r r r" r call_argsr r Z invalid_txtr r$ r r r test_invalid_flag8 s z GetEmailTest.test_invalid_flagc C sx | j }tjdf|_tdF}ddg|_| jdd |jD ]}d|d d vs z;ChooseNamesTest.test_sort_names_trivial..c s g | ] }| qS r r r r r r r r )r
ry sorted)r ry Z some_domainsunsorted_shortZ
unsorted_long sorted_shortZ sorted_longr r r test_sort_names_trivial s z'ChooseNamesTest.test_sort_names_trivialc C s ddl m} g d}g d}g d}g }|D ] }|D ]}||| q4q,g }t|D ] }|D ]}||| q^qV|||ksJ d S )Nr rx )z.cx.comz.bx.comz.ax.comz.dx.com)wwwbnother.long.subdomainr~ a.long.subdomainr b)r~ r r r r r )r
ry appendr ) r ry Zunsorted_domainsr r Zto_sortZshortrv Zsortdr r r test_sort_names_many s z$ChooseNamesTest.test_sort_names_manyc C sN dh| j j_tjdgf| j_| | j }|dgks8J | jjdksJJ d S rs )rm ru r r r! checklistr rp r rS namesr r r test_filter_names_valid_return s
z.ChooseNamesTest.test_filter_names_valid_returnc C sj dh| j j_tjdgf| j_| | j d}|dgks:J | jjdksLJ | jjd d dksfJ d S )Nra ZCustomrV r ) rm ru r r r! r r rp r* r r r r $test_filter_namees_override_question s z4ChooseNamesTest.test_filter_namees_override_questionc C s4 dh| j j_tjg f| j_| | j g ks0J d S Nra )rm ru r r r! r r rR r r r "test_filter_names_nothing_selected s z2ChooseNamesTest.test_filter_names_nothing_selectedc C s6 dh| j j_tjdgf| j_| | j g ks2J d S r )rm ru r r r r r rR r r r test_filter_names_cancel s
z(ChooseNamesTest.test_filter_names_cancelc C s\ ddl m} g d}ddg}g d}|||ks4J ||g ksDJ t||dksXJ d S )Nr )get_valid_domains)ra zsecond.example.comalso.example.comunder_score.example.comjusttldz*.wildcard.comu öóòps.net
uniçodé.com)ra u úniçøde.comr )r
r len)r r Z all_validZ all_invalidZ two_validr r r test_get_valid_domains s z&ChooseNamesTest.test_get_valid_domainsc C s ddl m} | }d|j_tjdf|j_| g ks6J td4}t ddddd}||_
| g ksfJ W d n1 sz0 Y tjd f|j_| g d
ksJ d S ) Nr _choose_names_manuallyFr z=certbot.display.ops.internal_display_util.separate_list_inputr r^ rV z=example.com,under_score.example.com,justtld,valid.example.com)ra r r zvalid.example.com) r
r yesnor r r! r r r" UnicodeEncodeErrorr& )r rS r utility_mockZmock_sliZ
unicode_errorr r r test_choose_manually# s( ,z$ChooseNamesTest.test_choose_manuallyc C sF ddl m} | }tjdf|j_g d|j_| |jjdksBJ d S )Nr r r )TTF ) r
r r r! r r r r& rp )r rS r r r r r test_choose_manually_retry> s z*ChooseNamesTest.test_choose_manually_retry)N)r1 r2 r3 r4 r? r5 r r r" rq r6 r7 rr rw r r r r r r r r r r r r r rl s2
rl c @ s6 e Zd ZdZedd Ze e ddd Z
dS )SuccessInstallationTestz&Test the success installation message.c C s ddl m} || d S )Nr )success_installation)r
r )r r r r r r r L s z SuccessInstallationTest._callcertbot.display.util.notifyc C sV d | j _ddg}| | |jdks*J |jd d d }|D ]}||v s@J q@d S )Nra abc.comrV r ) notificationr r rp r- )r mock_notify mock_displayr argnamer r r test_success_installationQ s
z1SuccessInstallationTest.test_success_installationN) r1 r2 r3 r4 r5 r r6 r7 r r" r r r r r r J s
r c @ s6 e Zd ZdZedd Ze e ddd Z
dS )SuccessRenewalTestz!Test the success renewal message.c C s ddl m} || d S )Nr )success_renewal)r
r )r r r r r r r b s zSuccessRenewalTest._callr c C s. d | j _ddg}| | |jdks*J d S )Nra r rV )r r r rp )r r r r r r r test_success_renewalg s
z'SuccessRenewalTest.test_success_renewalN) r1 r2 r3 r4 r5 r r6 r7 r r" r r r r r r ` s
r c @ s6 e Zd ZdZedd Ze e ddd Z
dS )SuccessRevocationTestz$Test the success revocation message.c C s ddl m} || d S )Nr )success_revocation)r
r )r rE r r r r r t s zSuccessRevocationTest._callr c C s" d}| | |d| d S )Nz/path/to/cert.pemzWCongratulations! You have successfully revoked the certificate that was located at {0}.)r Zassert_called_once_withrf )r r Zunused_mock_displayrE r r r test_success_revocationy s
z-SuccessRevocationTest.test_success_revocationN) r1 r2 r3 r4 r5 r r6 r7 r r" r r r r r r r s
r c @ s e Zd ZdZdZdZdZedd Ze
dd Z e
d d
Z e
d d Z
e
d
d Ze
dd Ze
dd Ze
dd ZdS )ValidatorTestsz6Tests for `validated_input` and `validated_directory`.zMust be non-emptyZasdfz
/var/www/htmlc C s | dkrt tjd S r] )r Z PluginErrorr _ValidatorTests__ERROR)mr r r Z __validator s zValidatorTests.__validatorc C sr t jdft jdft jdft j| jfg| j_tj| jddd}tj | j
j d d ksZJ |t j| jfksnJ d S )Nr^ messageTZforce_interactiver ) r r! valid_inputr r& r validated_input_ValidatorTests__validatorr r r r* r rS Zreturnedr r r test_input_blank_with_validator s
z.ValidatorTests.test_input_blank_with_validatorc C s> t j| jfg| j_tj| jddd}|t j| jfks:J d S Nmsgotherdefault)r r! r r r& r r r r r r r "test_input_validation_with_default s z1ValidatorTests.test_input_validation_with_defaultc C sT t j| jfg| j_tt" tj | j
ddd W d n1 sF0 Y d S Nr r^ r ) r r! r r r& r r AssertionErrorr r r rR r r r &test_input_validation_with_bad_default s z5ValidatorTests.test_input_validation_with_bad_defaultc C s: t jdfg| j_tj| jddd\}}|t jks6J d S )Nr^ r Tr )r r r r& r r r )r rS codeZ
unused_rawr r r test_input_cancel_with_validator s z/ValidatorTests.test_input_cancel_with_validatorc C sb t jdft j| jfg| j_tj| jddd}tj | j
j d d ksJJ |t j| jfks^J d S )Nr^ r Tr r ) r r! valid_directorydirectory_selectr& r validated_directoryr r r r r* r r r r test_directory_select_validation s
z/ValidatorTests.test_directory_select_validationc C s> t j| jfg| j_tj| jddd}|t j| jfks:J d S r )r r! r r r& r r r r r r r -test_directory_select_validation_with_default s z s< B+- E%