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 sH | j }tjdf|_td}d|_W d n1 s:0 Y d S )Nr #certbot.display.ops.util.safe_emailT)r r OKr r patchr r r mock_safe_emailr r r test_ok_safe( s
zGetEmailTest.test_ok_safec C sL | j }tjdf|_td}ddg|_W d n1 s>0 Y d S )Nr r FT)r r r! r r r" side_effectr# r r r test_ok_not_safe0 s
z GetEmailTest.test_ok_not_safec C s` d}| j }tjdf|_td*}d|_| | jdd W d n1 sR0 Y d S )NThere seem to be problemsr r TZinvalid)r r r! r r r" r r r Z invalid_txtr r$ r r r test_invalid_flag8 s z GetEmailTest.test_invalid_flagc C sd | j }tjdf|_td2}ddg|_| jdd |jD ]}q [ s0
zChooseAccountTest.setUpc C s
t |S N)r Zchoose_account)r Zaccountsr r r r p s zChooseAccountTest._callc C s t jdf| j_d S )Nr r r! menur r mock_utilr r r test_onet s zChooseAccountTest.test_onec C s t jdf| j_d S N rN rP r r r test_twoy s zChooseAccountTest.test_twoc C s t jdf| j_d S rS )r r rO r rP r r r test_cancel~ s z ChooseAccountTest.test_cancel)
r0 r1 r2 r3 r> r4 r r5 r6 rR rU rV
__classcell__r r rK r r7 Y s
r7 c @ sL e Zd ZdZdd Zedd Zdd Zdd Zd
d Z d d
Z
dd Z dS )GenHttpsNamesTestzTest _gen_https_names.c C s t t tjd d S NF)r? r@ rA rB rC rJ r r r r> s zGenHttpsNamesTest.setUpc C s ddl m} ||S )Nr )_gen_https_names)r
rZ )r ZdomainsrZ r r r r s zGenHttpsNamesTest._callc C s d S rM r rJ r r r test_zero s zGenHttpsNamesTest.test_zeroc C s ddg}|D ]}q d S )N example.comzasllkjsadfljasdf.cr )r domsdomr r r rR s
zGenHttpsNamesTest.test_onec C s ddgddgg}|D ]}qd S )Nz foo.bar.orgzbar.orgz paypal.google.facebook.live.comz*.zombo.example.comr )r Z domains_listr] r r r rU s
zGenHttpsNamesTest.test_twoc C s g d}d S )N)a.orgb.orgc.orgr )r r] r r r
test_three s z GenHttpsNamesTest.test_threec C s g d}dj |d}d S )N)r_ r` ra zd.orgzJhttps://{dom[0]}, https://{dom[1]}, https://{dom[2]}, and https://{dom[3]})r^ )format)r r] Zexpr r r test_four s
zGenHttpsNamesTest.test_fourN) r0 r1 r2 r3 r> r4 r r[ rR rU rb rd r r r r rX s
rX c @ s e Zd ZdZdd Zed ddZeddd Z e
d
d Z e
d d
Z
dd Zdd Ze
dd Ze
dd Ze
dd Ze
dd Zdd Ze
d d Ze
d d ZdS )!ChooseNamesTestzTest choose names.c C s" t t tjd t | _d S rY )r? r@ rA rB rC r rG mock_installrJ r r r r> s zChooseNamesTest.setUpNc C s ddl m} |||S )Nr ) choose_names)r
rg )r Z installerquestionrg r r r r s zChooseNamesTest._callz*certbot.display.ops._choose_names_manuallyc C s | d d S rM r )r Z mock_manualr r r test_no_installer s
z!ChooseNamesTest.test_no_installerc C s t jg f| j_d S rM )r r r r rP r r r test_no_installer_cancel s z(ChooseNamesTest.test_no_installer_cancelc C s2 t | j_d}tj|f| j_| | j}d S Nr\ )setrf
get_all_namesr r r! r r )r rQ domainZ actual_domsr r r test_no_names_choose s
z$ChooseNamesTest.test_no_names_choosec sP ddl m} g d}d g d} fdd|D }t|} fdd|D }d S ) Nr _sort_names)zex.comzzx.comzax.comz.ex.com)eazyc s g | ] }| qS r r .0usro r r
z;ChooseNamesTest.test_sort_names_trivial..c s g | ] }| qS r r rw rz r r r{ r| )r
rr sorted)r rr Z some_domainsunsorted_shortZ
unsorted_long sorted_shortZ sorted_longr rz r test_sort_names_trivial s z'ChooseNamesTest.test_sort_names_trivialc C sx ddl m} g d}g d}g d}g }|D ] }|D ]}||| q4q,g }t|D ] }|D ]}||| q^qVd S )Nr rq )z.cx.comz.bx.comz.ax.comz.dx.com)wwwbnother.long.subdomainrt a.long.subdomainru b)rt r r r r ru )r
rr appendr} ) r rr Zunsorted_domainsr~ r Zto_sortZshortro Zsortdr r r test_sort_names_many s z$ChooseNamesTest.test_sort_names_manyc C s. dh| j j_tjdgf| j_| | j }d S rl rf rn r r r! checklistr r rQ namesr r r test_filter_names_valid_return s z.ChooseNamesTest.test_filter_names_valid_returnc C s0 dh| j j_tjdgf| j_| | j d}d S )Nr\ ZCustomr r r r r $test_filter_namees_override_question s z4ChooseNamesTest.test_filter_namees_override_questionc C s dh| j j_tjg f| j_d S rl )rf rn r r r! r rP r r r "test_filter_names_nothing_selected s z2ChooseNamesTest.test_filter_names_nothing_selectedc C s" dh| j j_tjdgf| j_d S rl )rf rn r r r r rP r r r test_filter_names_cancel s
z(ChooseNamesTest.test_filter_names_cancelc C s( ddl m} g d}ddg}g d}d S )Nr )get_valid_domains)r\ zsecond.example.comalso.example.comzunder_score.example.comZjusttldz*.wildcard.comu öóòps.net
uniçodé.com)r\ u úniçøde.comr )r
r )r r Z all_validZ all_invalidZ two_validr r r test_get_valid_domains s
z&ChooseNamesTest.test_get_valid_domainsc C sz ddl m} | }d|j_tjdf|j_td&}t ddddd}||_
W d n1 s^0 Y tjd f|j_d S )
Nr _choose_names_manuallyFr z=certbot.display.ops.internal_display_util.separate_list_inputr rT z=example.com,under_score.example.com,justtld,valid.example.com) r
r yesnor r r! r r r" UnicodeEncodeErrorr& )r rQ r utility_mockZmock_sliZ
unicode_errorr r r test_choose_manually# s" z$ChooseNamesTest.test_choose_manuallyc C s6 ddl m} | }tjdf|j_g d|j_| d S )Nr r r )TTF)r
r r r! r r r r& )r rQ r r r r r test_choose_manually_retry> s z*ChooseNamesTest.test_choose_manually_retry)N)r0 r1 r2 r3 r> r4 r r r" rj r5 r6 rk rp r r r r r r r r r r r r r re s2
re 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 s< d | j _ddg}| | |jd d d }|D ]}q2d S )Nr\ abc.comr ) notificationr r r, )r mock_notify mock_displayr argnamer r r test_success_installationQ s
z1SuccessInstallationTest.test_success_installationN) r0 r1 r2 r3 r4 r r5 r6 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}| | d S )Nr\ r )r r r )r r r r r r r test_success_renewalg s
z'SuccessRenewalTest.test_success_renewalN) r0 r1 r2 r3 r4 r r5 r6 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 rD 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_withrc )r r Zunused_mock_displayrD r r r test_success_revocationy s
z-SuccessRevocationTest.test_success_revocationN) r0 r1 r2 r3 r4 r r5 r6 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 )Nr )r Z PluginErrorr _ValidatorTests__ERROR)mr r r Z __validator s zValidatorTests.__validatorc C sB t jdft jdft jdft j| jfg| j_tj| jddd}d S Nr messageTZforce_interactiver r! valid_inputr r& r validated_input_ValidatorTests__validatorr rQ Zreturnedr r r test_input_blank_with_validator s
z.ValidatorTests.test_input_blank_with_validatorc C s* t j| jfg| j_tj| jddd}d S Nmsgotherdefaultr 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 rP r r r &test_input_validation_with_bad_default s z5ValidatorTests.test_input_validation_with_bad_defaultc C s, t jdfg| j_tj| jddd\}}d S r )r r r r& r r r )r rQ codeZ
unused_rawr r r test_input_cancel_with_validator s z/ValidatorTests.test_input_cancel_with_validatorc C s2 t jdft j| jfg| j_tj| jddd}d S )Nr r Tr r r! valid_directorydirectory_selectr& r validated_directoryr r r r r test_directory_select_validation s
z/ValidatorTests.test_directory_select_validationc C s* t j| jfg| j_tj| jddd}d S r r r r r r -test_directory_select_validation_with_default s z s< B+- E%