JFIF $
d d 7
Viewing File: /usr/lib/python3.9/site-packages/certbot/_internal/tests/__pycache__/cli_test.cpython-39.opt-1.pyc
a
}|�g`f � @ s� d Z ddlZddlZddlmZ ddlZddlZddlZddl m
Z
m Z ddl Z ddl m
Z
ddlZddlmZ ddlmZ ddlmZmZ dd lmZ dd
lmZ dd lmZ dd lm Z dd
l m Z ddl m Z ddl m! m"Z# ddl m$Z$ e j%�&� Z'G dd� de$�Z(G dd� de j)�Z*e+e
d�dd�Z,e+e
ed�dd�Z-G dd� de j)�Z.e/d k�r|e�0e�1ej2d d� e3g �� dS ) z Tests for certbot._internal.cli.� N)�reload)�Any�List)�mock)�
challenges)�errors)�ArgumentSource�NamespaceConfig)�cli)� constants)� flag_default)�disco)�
filesystem)�os)�TempDirTestCasec @ s e Zd ZdZdd� ZdS )� TestReadFilezTest cli.read_filec C s� t �� }z�t �| j� t�t j�| jd��}t j�|�}t �
t j �� t
�|� W d � n1 sd0 Y d}t|d��}|�|� W d � n1 s�0 Y t
�|�\}}W t �|� n t �|� 0 d S )N�foos bar
�wb)r �getcwd�chdirZtempdirr �realpath�path�join�relpath�pytest�raises�argparse�ArgumentTypeErrorr
Z read_file�open�write)�selfZcurr_dirZ real_path�
relative_pathZ
test_contents�fr �contents� r$ �D/usr/lib/python3.9/site-packages/certbot/_internal/tests/cli_test.py�test_read_file s ( (zTestReadFile.test_read_fileN)�__name__�
__module__� __qualname__�__doc__r& r$ r$ r$ r% r s r c @ s e Zd ZdZdd� ZdS )�FlagDefaultTestzTests cli.flag_defaultc C s t jdkr n d S )N�nt)r �name�r r$ r$ r% �test_default_directories@ s
z(FlagDefaultTest.test_default_directoriesN)r' r( r) r* r/ r$ r$ r$ r% r+ = s r+ )�attr�valuec C s d S �Nr$ )� namespacer0 r1 r$ r$ r% � assert_set_by_user_with_valueK s r4 )r0 r1 �sourcec C s d S r2 r$ )r3 r0 r1 r5 r$ r$ r% �assert_value_and_sourceP s r6 c @ s� e Zd ZdZeee ed�dd��Zeee ed�dd��Z ee d�dd �Z
e � d
�d d � �Z
d
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Z d+d,� Z d-d.� Z e � d/�d0d1� �Z e � d2�d3d4� �Z d5d6� Z!d7d8� Z"d9d:� Z#d;d<� Z$d=d>� Z%d?d@� Z&dAdB� Z'dCdD� Z(dEdF� Z)dGdH� Z*dIdJ� Z+dKdL� Z,dMdN� Z-dOdP� Z.dQdR� Z/dSdT� Z0dUdV� Z1dWdX� Z2dYdZ� Z3d[d\� Z4d]d^� Z5e � d/�d_d`� �Z6dadb� Z7dcdd� Z8deS )f� ParseTestz Test the cli args entrypoint)�args�returnc C s t �t| �S )z)Get result of cli.prepare_and_parse_args.)r
�prepare_and_parse_args�PLUGINS�r8 r$ r$ r% �_unmocked_parseX s zParseTest._unmocked_parsec C s6 t �� � t�| �W d � S 1 s(0 Y dS )zJMocks certbot._internal.display.obj.get_display and calls _unmocked_parse.N)� test_util�patch_display_utilr7 r= r< r$ r$ r% �parse] s
zParseTest.parser< c s� t �� � � fdd�}tjd� d��� t�� �x}||� j_t�d��D t� t
�� | � |� W d � n1 sn0 Y W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y � � � S )z8Run a command, and return the output string for scrutinyc s � � | � d S r2 )r )�messager8 �kwargs��outputr$ r% � write_msgh s z)ParseTest._help_output.<locals>.write_msgz!certbot._internal.main.sys.stdout)�newz!certbot._internal.main.sys.stderrN)
�io�StringIOr �patchr> r? Z notification� side_effectr r �
SystemExitr= �getvalue)r r8 rE Zmock_get_utilityr$ rC r% � _help_outputc s
�zParseTest._help_outputz*certbot._internal.cli.helpful.flag_defaultc s� t �� ��� � �� � fdd�}||_| �dg�}t|dg tj� t� j d��}|�
d� W d � n1 sl0 Y | �dg�}t|ddgtj � | �dg�}t|dg tj � W d � n1 s�0 Y d S ) Nc s | dkrt �tj| �S � jgS )NZ config_files)�copy�deepcopyr Z CLI_DEFAULTSr- )�v��
tmp_configr$ r% �<lambda>z s �z0ParseTest.test_cli_ini_domains.<locals>.<lambda>�certonly�domains�wzdomains = example.com� example.com�renew)
�tempfile�NamedTemporaryFile�closerJ r@ r6 r �DEFAULTr r- r � CONFIG_FILE�RUNTIME)r Zmock_flag_defaultZshimr3 �file_hr$ rQ r% �test_cli_ini_domainst s
� ( z ParseTest.test_cli_ini_domainsc C s | � g �}dD ]}qd S )N)�
config_dirZlogs_dirZwork_dir�r@ )r r3 �dr$ r$ r% � test_no_args� s
zParseTest.test_no_argsc C sZ d}d}d}d}t �d��, | �dd|ddd dd
dg �}W d � n1 sL0 Y d S ) N�cert�key�chain� fullchainz certbot._internal.main.install�installz --cert-pathz
--key-pathz --chain-pathz--fullchain-path)r rI r@ )r re rf rg rh r3 r$ r$ r% �test_install_abspath� s
�$z ParseTest.test_install_abspathc C s� | � dg� | � ddg�}| � ddg�}dtv r0| � dg�}dtv rFn | � ddg�}| � ddg�}| � ddg�}| � ddg�}| � dd g�}| � ddg�}| � dd
g�}| � dg�}d S ) N�--help�all�-hZnginx�pluginsrX Z
automation�revokeri �register)rM r; �r �outr$ r$ r% � test_help� s zParseTest.test_helpc C s: | � dg� | � ddg�}| � ddg�}| � ddg�}d S )N�helprl ri ro )rM rq r$ r$ r% �test_help_no_dashes� s
z ParseTest.test_help_no_dashesc C s� ddg}| � |�}t|ddg� ddg}| � |�}t|ddg� ddg}| � |�}t|dg d�� ddg}| � |�}t|ddg� ddg}| � |�}t|ddg� dd g}| � |�}t|ddd
g� d S ) Nz-drW rU ztrailing.period.com.ztrailing.period.comz-example.com,another.net,third.org,example.com)rW � another.netz third.orgz --domainsz#example.com,another.net,example.comrv �r@ r4 )r �
short_argsr3 Z long_argsr$ r$ r% �test_parse_domains� s(
�
z ParseTest.test_parse_domainsc C s� ddg}| � |�}tjjtjjg}t|d|� ddg}t�d��D t� t
�� | � |� W d � n1 sl0 Y W d � n1 s�0 Y d S )Nz--preferred-challengesz http, dns� pref_challszjumping-over-the-moonz
sys.stderr) r@ r �HTTP01�typZDNS01r4 r rI r r rK )r rx r3 Zexpectedr$ r$ r% �test_preferred_challenges s
z#ParseTest.test_preferred_challengesc C s | � d�� �}t|dd� d S )N�--server example.com�serverrW )r@ �splitr4 �r r3 r$ r$ r% �test_server_flag s zParseTest.test_server_flagc C s, | � dg�}t|dd� t|ddtj� d S )N�
--must-staple� must_stapleT�staple)r@ r4 r6 r r^ r� r$ r$ r% �test_must_staple_flag s z ParseTest.test_must_staple_flagc C s* | � ddg�}t|dd� t|dd� d S )Nr� z
--staple-ocspr� Tr� rw r� r$ r$ r% �&test_must_staple_and_staple_ocsp_flags s z0ParseTest.test_must_staple_and_staple_ocsp_flagsc
C sZ z$| � |� | �d�d�|��� W n0 tjyT } z|D ]}q:W Y d }~n
d }~0 0 d S )Nz6The following flags didn't conflict with --server: {0}z, )r@ Zfail�formatr r �Error)r Z parser_argsZconflicting_args�error�argr$ r$ r% � _check_server_conflict_message" s
��z(ParseTest._check_server_conflict_messagec C sF dg}| � |�}t|dd� t|dtj� |d�� 7 }| �|d� d S )N� --staging�stagingTr r~ )r@ r4 r � STAGING_URIr� r� )r rx r3 r$ r$ r% �test_staging_flag- s
zParseTest.test_staging_flagc C s� t |dd� t|ddtj� t|ddtj� t|dtjtj� |rdt|ddtj� t|ddtj� n t|ddtj� t|ddtj� d S ) N�dry_runTZbreak_my_certsr� r ZtosZ register_unsafely_without_emailF)r4 r6 r r^ r r� r\ )r r3 Zexisting_accountr$ r$ r% �_assert_dry_run_flag_worked6 s z%ParseTest._assert_dry_run_flag_workedc C s� t �� }d�|��� }t�tj�� | �|� W d � n1 sB0 Y | � | �|dg �d� | � | �|dg �d� | � | �|dg �d� t
j � |t
j�}t�|� t�t
j � |d�� | � | �|dg �d� | � | �|dg �d� | � | �|dg �d� |dg7 }| �|dd g �}t|d
d � | �|dt
jg �}t|d
t
j� | �|dt�d
�g �}t|d
t
jtj� d g}| �|g d � |� d S )
Nz--dry-run --config-dir {0}ZauthFrT rX Zfake_account_dirT�--serverrW r r� )r� rW r� )rY Zmkdtempr� r� r r r r� r@ r� r r r r Z ACCOUNTS_DIRr �mkdirr4 r� r
r r6 r r^ r� )r ra rx Z account_dir�config� conflictsr$ r$ r% �test_dry_run_flagC s< (���
�zParseTest.test_dry_run_flagc C s* d}t �|�}| �d�|��� �}d}d S )NZ rsa_key_sizez--rsa-key-size {0}ra �r
r r@ r� r� )r Zkey_size_optionZkey_size_valuer� Zconfig_dir_optionr$ r$ r% �test_user_set_rsa_key_sizek s
z$ParseTest.test_user_set_rsa_key_sizec C s | � d�}| � d�}d S )Nz--apachez--installer webrootrb )r r� r$ r$ r% �)test_user_set_installer_and_authenticatorw s
z3ParseTest.test_user_set_installer_and_authenticatorc C s& d}t �|�}| �d�|��� �}d S )NZelliptic_curvez--elliptic-curve {0}r� )r Zelliptic_curve_optionZelliptic_curve_option_valuer� r$ r$ r% � test_user_set_ecdsa_key_option� s
z(ParseTest.test_user_set_ecdsa_key_optionc C sZ d}t �|�}| �d�|��� �}t�t�� | �d� W d � n1 sL0 Y d S )NZkey_typez--key-type {0}z--key-type foo)r
r r@ r� r� r r rK )r Zkey_type_optionZkey_type_valuer� r$ r$ r% � test_user_set_invalid_key_type� s
z(ParseTest.test_user_set_invalid_key_typez&certbot._internal.hooks.validate_hooksc C s" d� � }tj�� }t�||�}d S )Nzrenew --deploy-hook foo)r� r
�PluginsRegistry�find_allr
r: )r Zunused_mock_validate_hooksr8 rn r� r$ r$ r% �test_user_set_deploy_hook� s
z#ParseTest.test_user_set_deploy_hookz3certbot._internal.plugins.webroot._validate_webrootc C s( d� � }d|_tj�� }t�||�}d S )Nz%renew -w /var/www/html -d example.comz
/var/www/html)r� Z return_valuer
r� r� r
r: )r Zmock_validate_webrootr8 rn r� r$ r$ r% �test_user_set_webroot_map� s
z#ParseTest.test_user_set_webroot_mapc C sL t j�� D ]\}}| �d|g�}q
t j�� D ]\}}| �d|�� g�}q,d S )Nz--reason)r ZREVOCATION_REASONS�itemsr@ �upper)r �reason�coder3 r$ r$ r% � test_encode_revocation_reason� s z'ParseTest.test_encode_revocation_reasonc C sx t �tj�� | �d�� � W d � n1 s00 Y t �tj�� | �d�� � W d � n1 sj0 Y d S )Nzrenew --force-interactivez-n --force-interactive�r r r r� r@ r� r. r$ r$ r% �test_force_interactive� s ,z ParseTest.test_force_interactivec C sf t �d��H t�t�� | �d�� � W d � n1 s:0 Y W d � n1 sX0 Y d S )N� certbot._internal.cli.sys.stderrz"--renew-hook foo --deploy-hook bar�r rI r r rK r@ r� r. r$ r$ r% �test_deploy_hook_conflict� s z#ParseTest.test_deploy_hook_conflictc C s4 d}| � d|d|dg�}t|d|� t|d|� d S )Nr � --renew-hook�
--deploy-hook�--disable-hook-validation� deploy_hook�
renew_hookrw �r r1 r3 r$ r$ r% �#test_deploy_hook_matches_renew_hook� s � z-ParseTest.test_deploy_hook_matches_renew_hookc C s0 d}| � d|dg�}t|d|� t|d|� d S )Nr r� r� r� r� rw r� r$ r$ r% � test_deploy_hook_sets_renew_hook� s � z*ParseTest.test_deploy_hook_sets_renew_hookc C sf t �d��H t�t�� | �d�� � W d � n1 s:0 Y W d � n1 sX0 Y d S )Nr� z"--deploy-hook foo --renew-hook barr� r. r$ r$ r% �test_renew_hook_conflict� s z"ParseTest.test_renew_hook_conflictc C s4 d}| � d|d|dg�}t|d|� t|d|� d S )Nr r� r� r� r� r� rw r� r$ r$ r% �#test_renew_hook_matches_deploy_hook� s � z-ParseTest.test_renew_hook_matches_deploy_hookc C s$ d}| � d|dg�}t|d|� d S )Nr r� r� r� rw r� r$ r$ r% �'test_renew_hook_does_not_set_renew_hook� s
�z1ParseTest.test_renew_hook_does_not_set_renew_hookc C s� t �d��� t�t�� | �d�� � W d � n1 s:0 Y t�t�� | �d�� � W d � n1 sr0 Y W d � n1 s�0 Y d S )Nr� z--max-log-backups fooz--max-log-backups -42r� r. r$ r$ r% �test_max_log_backups_error� s
, z$ParseTest.test_max_log_backups_errorc C s& d}| � d|g�}t|dt|�� d S )NZ42z--max-log-backupsZmax_log_backups)r@ r4 �intr� r$ r$ r% � test_max_log_backups_success� s z&ParseTest.test_max_log_backups_successc C sl | � g �}t|dg tj� t|dg tj� tjjg|_dg|_| � g �}t|dg tj� t|dg tj� d S )NrU rz rW ) r@ r6 r r\ r r{ r| rz rU r� r$ r$ r% �test_unchanging_defaults� s
z"ParseTest.test_unchanging_defaultsc C s | � dg�}t|dd� d S )Nz--no-directory-hooks�directory_hooksFrw r� r$ r$ r% �test_no_directory_hooks_set� s z%ParseTest.test_no_directory_hooks_setc C s | � g �}t|ddtj� d S )Nr� T)r@ r6 r r\ r� r$ r$ r% � test_no_directory_hooks_unset� s
z'ParseTest.test_no_directory_hooks_unsetc C s | � dg�}t|dd� d S )Nz--delete-after-revoke�delete_after_revokeTrw r� r$ r$ r% �test_delete_after_revoke� s z"ParseTest.test_delete_after_revokec C s | � g �}d S r2 rb r� r$ r$ r% � test_delete_after_revoke_default� s
z*ParseTest.test_delete_after_revoke_defaultc C s | � dg�}t|dd� d S )Nz--no-delete-after-revoker� Frw r� r$ r$ r% �test_no_delete_after_revoke� s z%ParseTest.test_no_delete_after_revokec C s> t �tj�� | �d�� � W d � n1 s00 Y d S )Nz(--allow-subset-of-names -d *.example.orgr� r. r$ r$ r% � test_allow_subset_with_wildcard s z)ParseTest.test_allow_subset_with_wildcardc C s dD ]}dD ]}q qd S )N)rm rk )rl rn z dns-route53r$ )r Z help_flagZtopicr$ r$ r% �test_route53_no_revert s z ParseTest.test_route53_no_revertc C s> t �tj�� | �ddg� W d � n1 s00 Y d S )Nz--hstsz --auto-hsts)r r r r� r@ r. r$ r$ r% �$test_parse_args_hosts_and_auto_hosts
s z.ParseTest.test_parse_args_hosts_and_auto_hostsc C s� t d�}d}d}| �dg�}t|d|tj� t�� ��}|�� t|j d�� }|�
d|� �� W d � n1 sp0 Y | �dd|j g�}t|d|tj � | �dd|j d|g�}t|d|tj � W d � n1 s�0 Y d S ) Nr zconfigfile.bizzcommandline.edurT rV z server = z-cr� )
r r@ r6 r r\ rY rZ r[ r r- r r] Z COMMAND_LINE)r Z
DEFAULT_VALUEZCONFIG_FILE_VALUEZCOMMAND_LINE_VALUEr3 rR r_ r$ r$ r% �)test_parse_with_multiple_argument_sources s(
.��z3ParseTest.test_parse_with_multiple_argument_sourcesc C s� | � d�}t|ddg� t|dd� t�� �n}|�� t|jd��}|�d� W d � n1 sb0 Y | � dd |jg�}t|ddg� W d � n1 s�0 Y d S )
Nz --preferred-chal dns --no-dirrz zdns-01r� FrV zpreferred-chal = dnsrT z--config)r@ r4 rY rZ r[ r r- r )r r3 rR r_ r$ r$ r% �test_abbreviated_arguments+ s
(�z$ParseTest.test_abbreviated_argumentsc C sB | � d�}t|ddg� tj�� }t�|ddg�}t|dd� d S )Nz-d=example.comrU rW �runz--pre-hook="foo=bar"Zpre_hookz "foo=bar")r@ r4 r
r� r� r
r: )r Zunsused_mock_validate_hooksr3 rn r$ r$ r% �test_argument_with_equals> s
z#ParseTest.test_argument_with_equalsc C sv | � d�}t|dd� t|dd� | � d�}t|dd� t|dd� | � d�}t|dd� t|dd� t|d d
� d S ) Nz-tvZ text_modeTZ
verbose_count� z-tvvv� z-tvm foo@example.comZemailzfoo@example.comrw r� r$ r$ r% �test_adjacent_short_argsH s
z"ParseTest.test_adjacent_short_argsc C s | � ddg�}t|ddg� d S )NrT z
-d foo.comrU zfoo.comrw r� r$ r$ r% � test_arg_with_contained_spacesV s z(ParseTest.test_arg_with_contained_spacesN)9r' r( r) r* � staticmethodr �strr r= r@ rM r rI r` rd rj rs ru ry r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r$ r$ r$ r% r7 U sl
:
(
r7 �__main__r� )4r* r rN � importlibr Z
reload_modulerG �sysrY �typingr r Zunittestr r Zacmer Zcertbotr Zcertbot.configurationr r Zcertbot._internalr
r Z certbot._internal.cli.cli_utilsr Zcertbot._internal.pluginsr
Zcertbot.compatr r Zcertbot.tests.utilZtests�utilr> r r� r� r; r ZTestCaser+ r� r4 r6 r7 r' �exit�main�argv�__file__r$ r$ r$ r% �<module> sB
Back to Directory
nL+D550H?Mx ,D"v]qv;6*Zqn)ZP 0 !1 A "#a$2Qr D8 a Ri[f\mIykIw0 cuFcRı?lO7к_f˓[C$殷WF<_W ԣs KcëIzyQy /_LKℂ;C",pFA: /]=H ~,ls/9ć:[=/#f;)x {ٛEQ )~ =𘙲r*2~ a _V=' kumF D}KYYC)({ *g&f`툪ry`=^cJ.I](*`wq 1dđ#̩͑0;H]u搂@:~וKL Nsh}OIR*8:2 !lDJVo( 3=M(zȰ+i*NA r6KnSl )!JJӁ* %݉?|D}d5:eP0R;{$X'xF@.ÊB {,WJuQɲRI;9QE琯62fT.DUJ;*cP A\ILNj!J۱+O\ ͔]ޒ SJȧc%ANolՎprULZԛerE2=XDXgVQeӓk yP7U*omQIs,K`)6\G3t?pgjrmۛجwluGtf h9uyP0D;Uڽ"OXlif$)&|ML0Zrm1[HXPlPR0'G=i2N +0e2]] 9VTPO7h(F*癈 '=QVZDF,d߬~TX
G[`le69CR(!S2!P
<0x < !1AQ "Raq02Br#SCTb
?Ζ"]mH5WR7k.ۛ!}Q~+yԏz|@T20S~Kek *zFf^2X*(@8r?CIuI|֓>^ExLgNUY+{.RѪ
τVYTD I62'8Y2 7'\TP.6d&˦@Vqi|8- OΕ]ʔ U=TL8=;6c| !qfF3aů&~$l}'NWUs$Uk^SV:U# 6w++s&r+nڐ{@29 gL
u"TÙM=6 (^"7r} =6YݾlCuhquympǦ
G jhsǜNlɻ}o7 #S6aw4!OSrD57%|?x>L
|/nD6?/8w#[)L7+6〼T ATg ! %5MmZ/c-{1_Je"|^$'O&ޱմ Trb$ w)R$&
N1EtdU3Uȉ1pM" N*(DN yd96.(jQ)X
5cQɎMyW? Q*!R>6=7)Xj5`J]e 8%t!+'!1Q 5 !1 AQaqё#2"0BRb ? Gt^## .llQT $v,,m 㵜5 ubV =sY+@d{N! dnO< .-B; _wJt6;QJd.Qc%p{ 1,sNDdFHI0ГoXшe黅XۢF:)[FGXƹ/w_cMeD,ʡcc. WDtA$j@:) -# u
c1<@ۗ9F)KJ-hpP]_x[qBlbp ʖw q"LFGd ƶ*s+ډ_Zc"? %t[IP 6J]#=ɺVvvCGsGh1 >)6|ey?Lӣm,4GWUi` ]uJVoV DG< SB 6ϏQ@ TiUly OU0kfV~~}SZ@*WUUi##; s/[ =!7}" WN]'(L! ~y5g9T̅JkbM' +s:S + B)v@Mj
e Cf jE 0Y\QnzG1д~Wo{T 9?`Rmyhsy3!HAD]mc1~2L Su7xT;j$`}4->L#vzŏ ILS ֭ T { rjGKC; bpU= -`BsK.SFw4Mq]ZdHS0)tLg