a
}|g& @ s d Z ddlZddlmZ ddlZddlmZ ddlZddlmZ ddlm Z ddl
m Z ddl
mZ dd lmZ dd lmZ G d
d d ejZG d d
d
ejZG dd dejZG dd dejZG dd dejZedkre e ej dd e g dS )z'Tests for letsencrypt.plugins.selection N)List)mock)errors)
interfaces)obj)PluginsRegistry)utilc @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
S ) ConveniencePickPluginTestz5Tests for certbot._internal.plugins.selection.pick_*.c C st t }t }t }t d,}d|_d||||dks@J W d n1 sT0 Y ||||d| d S )Nz/certbot._internal.plugins.selection.pick_pluginfoo Question?)r Mockpatch return_valueassert_called_once_with)selfZfunifacesconfigdefaultpluginsZmock_p r R/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/selection_test.py_test s 4
z ConveniencePickPluginTest._testc C s ddl m} | |tjf d S )Nr )pick_authenticator)#certbot._internal.plugins.selectionr r r
Authenticator)r r r r r test_authenticator s z,ConveniencePickPluginTest.test_authenticatorc C s ddl m} | |tjf d S )Nr )pick_installer)r r r r Installer)r r r r r test_installer# s z(ConveniencePickPluginTest.test_installerc C s$ ddl m} | |tjtjf d S )Nr )pick_configurator)r r r r r r )r r r r r test_configurator' s
z+ConveniencePickPluginTest.test_configuratorN)__name__
__module__ __qualname____doc__r r r r r r r r r s
r c @ s` e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dd Z
dS )PickPluginTestz:Tests for certbot._internal.plugins.selection.pick_plugin.c C s. t jdd| _d | _t | _d| _g | _d S )NF)Znoninteractive_moder )r r r r MagicMockregquestionr r r r r setUp0 s
zPickPluginTest.setUpc C s& ddl m} || j| j| j| j| jS )Nr ) pick_plugin)r r+ r r r' r( r )r r+ r r r _call7 s zPickPluginTest._callc C s$ d| _ | d| jjjks J d S Nr
)r r, r' filter
call_countr) r r r test_default_provided< s z$PickPluginTest.test_default_providedc C s" | d| j jjks J d S Nr. )r, r' visibler r0 r) r r r test_no_defaultA s z PickPluginTest.test_no_defaultc C s | d u sJ d S N)r, r) r r r test_no_candidateE s z PickPluginTest.test_no_candidatec C s@ t }d|j_d|_d|i| j j_d| ks r? rC rD rE r r r r r% - s
r% c @ sP e Zd ZdZdd Zdd Ze dd Ze dd Z e d
d Z
d S )
ChoosePluginTestzJ d S )Nr r. ) display_utilOKmenu side_effectrO r, Z notificationr0 r Z mock_utilr r r test_selection s
z ChoosePluginTest.test_selectionc C s( t jdfg| j_| j| ks$J d S r2 )rQ rR rS rT rO r, rU r r r test_more_info s
z ChoosePluginTest.test_more_infoc C s$ t jdf| j_| d u s J d S )Nr )rQ ZCANCELrS r r, rU r r r test_no_choice s z ChoosePluginTest.test_no_choiceN) r! r" r# r$ r* r, test_utilZpatch_display_utilrV rW rX r r r r rF z s
rF c sH e Zd ZdZ fddZdd Zdd Zdd Zd
d Zd d
Z Z
S )GetUnpreparedInstallerTestzGTests for certbot._internal.plugins.selection.get_unprepared_installer.c sl t tjdd| _dd | j_tjdd| _dd | j_t | _| j| jj _
t | j| jd| _ d S )NafailrH c S s | dkS )Nr[ r rN r r r z2GetUnpreparedInstallerTest.setUp..rI c S s | dkS NrI r r] r r r r^ r_ )r[ rI )
superr* r r mock_apache_fail_ep
check_nameZmock_apache_epr& mock_apache_pluginr9 r r r r) __class__r r r* s
z GetUnpreparedInstallerTest.setUpc C s ddl m} || j| jS )Nr )get_unprepared_installer)r rg r r )r rg r r r r, s z GetUnpreparedInstallerTest._callc C s d | j _| d u sJ d S r5 )r configuratorr, r) r r r test_no_installer_defined s z4GetUnpreparedInstallerTest.test_no_installer_definedc C sJ d| j _ti | _ttj | W d n1 s<0 Y d S r` ) r rh r r pytestraisesr PluginSelectionErrorr, r) r r r test_no_available_installers s
z7GetUnpreparedInstallerTest.test_no_available_installersc C s" d| j _| }|| ju s J d S r` )r rh r, rd )r Z installerr r r test_get_plugin s z*GetUnpreparedInstallerTest.test_get_pluginc C sL d| j _dd | j_ttj | W d n1 s>0 Y d S )NrI c S s | dkS r` r r] r r r r^ r_ zNGetUnpreparedInstallerTest.test_multiple_installers_returned..) r rh rb rc rj rk r rl r, r) r r r !test_multiple_installers_returned s z.F)r r rc r& rN r9 r r: )r rN Zmock_epZ mock_pluginr r] r _setupMockPlugin s z.TestChooseConfiguratorPlugins._setupMockPluginc C s* ddl m} ddlm} || j| S )Nr )
configuration)cli)certbotrt Zcertbot._internalru Zprepare_and_parse_argsr split)r argsrt ru r r r
_parseArgs s z(TestChooseConfiguratorPlugins._parseArgsc C s( t | d| d| dd| _d S )NnginxrI manual)rz rI r{ )r rs r r) r r r r* s
z#TestChooseConfiguratorPlugins.setUpc C s ddl m} || || jdS )Nr )choose_configurator_pluginsZcertonly)r r| ry r )r rx r| r r r _runWithArgs s z*TestChooseConfiguratorPlugins._runWithArgsc C sX | d\}}|jdks J |jdks*J | d\}}|jdksFJ |jdksTJ d S )Nzcertonly --nginxrz zcertonly --apacherI r} rN r instZauthr r r test_noninteractive_configurator s z>TestChooseConfiguratorPlugins.test_noninteractive_configuratorc C s | d\}}|jdks J |jdks*J | d\}}|jdksFJ |jdksTJ | d\}}|d u snJ |jdks|J | d\}}|d u sJ |jdksJ | d\}}|d u sJ |jdksJ | d\}}|d u sJ |jdksJ d S ) Nzcertonly -a nginx -i nginxrz z certonly -a apache -i apacherI zcertonly -a nginxzcertonly -a nginx -i nonezcertonly -a apachezcertonly -a apache -i noner~ r r r r test_noninteractive_inst_arg s$ z:TestChooseConfiguratorPlugins.test_noninteractive_inst_argN)
r! r" r# r$ rs ry r* r} r r r r r r rq s
rq __main__r. ) r$ rL typingr Zunittestr rj rv r r Zcertbot._internal.displayr rK Z certbot._internal.plugins.discor Zcertbot.displayr rQ Z
certbot.testsrY ZTestCaser r% rF ZConfigTestCaserZ rq r! exitmainargv__file__r r r r s$ M*-B