|
Server : Apache/2.4.41 (Ubuntu) System : Linux vmi1525618.contaboserver.net 5.4.0-105-generic #119-Ubuntu SMP Mon Mar 7 18:49:24 UTC 2022 x86_64 User : www-data ( 33) PHP Version : 8.2.12 Disable Function : NONE Directory : /lib/python3/dist-packages/certbot/tests/__pycache__/ |
Upload File : |
U
�]U � @ s2 d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m m
Z ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd l mZ ej�� ZG d d
� d
e�Z G dd� dej �Z G dd� dej �Z G dd� dej �Z G dd� dej �Z!dd� Z"e#dk�r.e�$� dS )zTests for certbot.cli.� N)�
reload_module)�
challenges)�cli)� constants)�errors)�os)�
filesystem)�disco)�TempDirTestCasec @ s e Zd ZdZdd� ZdS )� TestReadFilezTest cli.read_filec C s� t �� }z�t �| j� t j�t j�| jd��}| �tj t
j |� d}t |d��}|�
|� W 5 Q R X t
� |�\}}| �|t j�|�� | �||� W 5 t �|� X d S )N�foos bar
�wb)r �getcwd�chdirZtempdir�path�relpath�join� assertRaises�argparseZArgumentTypeErrorr Z read_file�open�write� assertEqual�abspath)�selfZcurr_dirZ
rel_test_pathZ
test_contents�fr �contents� r �8/usr/lib/python3/dist-packages/certbot/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 jdkrB| �t�d�d� | �t�d�d� | �t�d�d� n6| �t�d�d� | �t�d�d � | �t�d�d
� d S ) N�nt�
config_dirz/etc/letsencrypt�work_dirz/var/lib/letsencrypt�logs_dirz/var/log/letsencryptz
C:\CertbotzC:\Certbot\libzC:\Certbot\log)r �namer r � flag_default�r r r r �test_default_directories4 s
z(FlagDefaultTest.test_default_directoriesN)r r r! r" r+ r r r r r# 1 s r# c @ sJ e Zd ZdZdd� Zedd� �Zedd� �Zdd � Ze �
d
�d d � �Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z 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+dMS )N� ParseTestz Test the cli args entrypointc C s t t� d S �N�r r r* r r r �setUpC s zParseTest.setUpc O s t jtf| �|�S )z)Get result of cli.prepare_and_parse_args.)r Zprepare_and_parse_args�PLUGINS��args�kwargsr r r �_unmocked_parseF s zParseTest._unmocked_parsec
O s, t �� � tj| |�W 5 Q R � S Q R X dS )z:Mocks zope.component.getUtility and calls _unmocked_parse.N)� test_util�patch_get_utilityr, r4 r1 r r r �parseK s
zParseTest.parsec s| t �� � � fdd�}tjd� d��L t�� �8}||� j_t�d�� | �t | j
|� � W 5 Q R X W 5 Q R X W 5 Q R X � � � S )z8Run a command, and return the output string for scrutinyc s � � | � d S r- )r )�messager2 r3 ��outputr r � write_msgV s z)ParseTest._help_output.<locals>.write_msgzcertbot.main.sys.stdout)�newzcertbot.main.sys.stderr) �six�StringIO�mock�patchr5 r6 Z notification� side_effectr �
SystemExitr4 �getvalue)r r2 r; Zmock_get_utilityr r9 r � _help_outputQ s
0zParseTest._help_outputzcertbot.cli.flag_defaultc
s� t �� ��� � �� � fdd�}||_| �dg�}| �|jg � t� jd��}|� d� W 5 Q R X | �dg�}| �|jdg� | �dg�}| �|jg � W 5 Q R X d S )Nc s | dkrt �tj| �S � jgS )NZ config_files)�copy�deepcopyr Z CLI_DEFAULTSr( )�v�Z
tmp_configr r �<lambda>g s �z0ParseTest.test_cli_ini_domains.<locals>.<lambda>�certonly�wzdomains = example.com� example.com�renew)
�tempfileZNamedTemporaryFile�closerA r7 r �domainsr r( r )r Zmock_flag_defaultZshim� namespaceZfile_hr rH r �test_cli_ini_domainsa s
� z ParseTest.test_cli_ini_domainsc C s0 | � g �}dD ] }| �t||�t�|�� qd S )N)r% r' r& )r7 r �getattrr r) )r rQ �dr r r � test_no_argsv s
zParseTest.test_no_argsc C s� d}d}d}d}t �d��" | �dd|ddd dd
dg �}W 5 Q R X | �|jtj�|�� | �|jtj�|�� | �|j tj�|�� | �|j
tj�|�� d S ) N�cert�key�chain� fullchainzcertbot.main.install�install� --cert-path�
--key-pathz --chain-pathz--fullchain-path) r? r@ r7 r Z cert_pathr r r Zkey_pathZ
chain_pathZfullchain_path)r rV rW rX rY rQ r r r �test_install_abspath{ s"
�z ParseTest.test_install_abspathc C s� | � dg� | � ddg�}| �d|k� | �d|k� | �d|k� | �d|k� | �d|k� | �d|k� | �d |k� | �d
|k� | � d d g�}d tkr�| �d
|k� | �d|k� | �d|k� | � d g�}| �d|k� d tkr�| �d|k� n| �d|k� | � ddg�}| �d|k� | �d|k� | �d|k� | � d dg�}| �d|k� | � d dg�}| �d|k� | � d dg�}| �d|k� | � ddg�}| �d|k� | �d|k� | � ddg�}| �d|k� | �d|k� | �d |k� | �d |k� | �d |k� | � d d g�}| �d|k� | �d|k� | � d g�}| �tj|k� | �tjd d � |k� | �d|k� | �d |k� d S )!N�--help�all�--configurator� how a certificate is deployed�--webroot-path�--text�--dialog�%s�{0}� --renew-hook�-hZnginxz --nginx-ctlz
--checkpointszletsencrypt-autozUse the Nginx pluginz (the certbot nginx plugin is not�pluginsz --preparez"plugins" subcommandrM z--keepZ
automation�revokerZ r[ r\ �--reason�--delete-after-revoke�--no-delete-after-revokeZconfig_changes�d )rD �
assertTruer0 r Z SHORT_USAGEZCOMMAND_OVERVIEW�r �outr r r � test_help� s^ zParseTest.test_helpc C s� | � dg� | � ddg�}| �d|k� | �d|k� | �d|k� | �d|k� | �d|k� | �d|k� | �d |k� | � dd
g�}| �d |k� | �d |k� | � dd
g�}| �d |k� | �d |k� d S )N�helpr_ r` ra rb rc rd re rf rZ r[ r\ rj )rD ro rp r r r �test_help_no_dashes� s z ParseTest.test_help_no_dashesc C s� ddg}| � |�}| �|jdg� ddg}| � |�}| �|jdg� ddg}| � |�}| �|jdddg� ddg}| � |�}| �|jdg� ddg}| � |�}| �|jdg� dd g}| � |�}| �|jddg� d S )
Nz-drL ztrailing.period.com.ztrailing.period.comz-example.com,another.net,third.org,example.comz another.netz third.orgz --domainsz#example.com,another.net,example.com)r7 r rP )r �
short_argsrQ Z long_argsr r r �test_parse_domains� s(
�
z ParseTest.test_parse_domainsc C sb ddg}| � |�}tjjtjjg}| �|j|� ddg}t�d�� | � t
| j |� W 5 Q R X d S )Nz--preferred-challengesz http, dnszjumping-over-the-moonz
sys.stderr) r7 r �HTTP01�typZDNS01r � pref_challsr? r@ r rB )r ru rQ Zexpectedr r r �test_preferred_challenges� s
z#ParseTest.test_preferred_challengesc C s | � d�� �}| �|jd� d S )N�--server example.comrL )r7 �splitr �server�r rQ r r r �test_server_flag s zParseTest.test_server_flagc C s, dg}| � |�}| �|j� | �|j� d S )Nz
--must-staple)r7 ro Z must_stapleZstaple�r ru rQ r r r �test_must_staple_flag s
z ParseTest.test_must_staple_flagc
C s| z$| � |� | �d�d�|��� W nR tjk
rv } z2| �dt|�k� |D ]}| �|t|�k� qNW 5 d }~X Y nX d S )Nz6The following flags didn't conflict with --server: {0}z, �--server)r7 Zfail�formatr r �Errorro �str)r Z parser_argsZconflicting_args�error�argr r r � _check_server_conflict_message s
��z(ParseTest._check_server_conflict_messagec C sH dg}| � |�}| �|j� | �|jtj� |d�� 7 }| �|d� d S )N� --stagingr{ ) r7 ro �stagingr r} r � STAGING_URIr| r� r� r r r �test_staging_flag s
zParseTest.test_staging_flagc C sn | � |j� | � |j� | � |j� | �|jtj� |rR| � |j� | � |j � n| �
|j� | �
|j � d S r- ) ro Zdry_runZbreak_my_certsr� r r} r r� ZtosZ register_unsafely_without_email� assertFalse)r rQ Zexisting_accountr r r �_assert_dry_run_flag_worked s z%ParseTest._assert_dry_run_flag_workedc C sp t �� }d�|��� }| �tj| j|� | �| �|dg �d� | �| �|dg �d� | �| �|dg �d� t j
� |t j
�}t�|� t�t j
� |d�� | �| �|dg �d� | �| �|dg �d� | �| �|dg �d� |dg7 }| �| �|dd g �jd � | �| �|dt jg �jt j� | �| �|dt�d
�g �jt j� d g}| �|dd d g |� d S ) Nz--dry-run --config-dir {0}ZauthFrJ rM Zfake_account_dirTr� rL r} r� )rN Zmkdtempr� r| r r r� r7 r� r r r r Z ACCOUNTS_DIRr �mkdirr r} r� r r) r� )r r% ru Z account_dirZ conflictsr r r �test_dry_run_flag- sF � � �
�� ��zParseTest.test_dry_run_flagc C s~ d}t �|�}| �d�|��� � | �t �||�� | �t �dd�� d}| �t �|t �|��� | �t �dt �d��� d S )NZ rsa_key_sizez--rsa-key-size {0}Z
no_verify_sslTr% Z
authenticator)r r) r7 r� r| ro Zoption_was_setr� )r Zkey_size_optionZkey_size_valueZconfig_dir_optionr r r �test_option_was_setT s
� �z ParseTest.test_option_was_setc C sh t j�� D ]$\}}| �d|g�}| �|j|� q
t j�� D ](\}}| �d|�� g�}| �|j|� q:d S )Nrk )r ZREVOCATION_REASONS�itemsr7 r �reason�upper)r r� �coderQ r r r � test_encode_revocation_reasonb s z'ParseTest.test_encode_revocation_reasonc C s0 | � tj| jd�� � | � tj| jd�� � d S )Nzrenew --force-interactivez-n --force-interactive�r r r� r7 r| r* r r r �test_force_interactivej s � �z ParseTest.test_force_interactivec C s. t �d�� | �t| jd�� � W 5 Q R X d S )N�certbot.cli.sys.stderrz"--renew-hook foo --deploy-hook bar�r? r@ r rB r7 r| r* r r r �test_deploy_hook_conflictp s
�z#ParseTest.test_deploy_hook_conflictc C s8 d}| � d|d|dg�}| �|j|� | �|j|� d S )Nr rg �
--deploy-hook�--disable-hook-validation�r7 r Z deploy_hook�
renew_hook�r �valuerQ r r r �#test_deploy_hook_matches_renew_hooku s �z-ParseTest.test_deploy_hook_matches_renew_hookc C s4 d}| � d|dg�}| �|j|� | �|j|� d S )Nr r� r� r� r� r r r � test_deploy_hook_sets_renew_hook} s �z*ParseTest.test_deploy_hook_sets_renew_hookc C s. t �d�� | �t| jd�� � W 5 Q R X 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 s8 d}| � d|d|dg�}| �|j|� | �|j|� d S )Nr r� rg r� r� r� r r r �#test_renew_hook_matches_deploy_hook� s �z-ParseTest.test_renew_hook_matches_deploy_hookc C s4 d}| � d|dg�}| �|jd � | �|j|� d S )Nr rg r� r� r� r r r �'test_renew_hook_does_not_set_renew_hook� s �z1ParseTest.test_renew_hook_does_not_set_renew_hookc C sB t �d��. | �t| jd�� � | �t| jd�� � W 5 Q R X 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�}| �|jt|�� d S )NZ42z--max-log-backups)r7 r Zmax_log_backups�intr� r r r � test_max_log_backups_success� s z&ParseTest.test_max_log_backups_successc C sd | � g �}| �|jg � | �|jg � tjjg|_dg|_| � g �}| �|jg � | �|jg � d S )NrL )r7 r rP ry r rw rx r~ r r r �test_unchanging_defaults� s
z"ParseTest.test_unchanging_defaultsc C s | � | �dg�j� d S )Nz--no-directory-hooks)r� r7 �directory_hooksr* r r r �test_no_directory_hooks_set� s z%ParseTest.test_no_directory_hooks_setc C s | � | �g �j� d S r- )ro r7 r� r* r r r � test_no_directory_hooks_unset� s z'ParseTest.test_no_directory_hooks_unsetc C s | � dg�}| �|j� d S )Nrl )r7 ro �delete_after_revoker~ r r r �test_delete_after_revoke� s z"ParseTest.test_delete_after_revokec C s | � g �}| �|jd � d S r- )r7 r r� r~ r r r � test_delete_after_revoke_default� s
z*ParseTest.test_delete_after_revoke_defaultc C s | � dg�}| �|j� d S )Nrm )r7 r� r� r~ r r r �test_no_delete_after_revoke� s z%ParseTest.test_no_delete_after_revokec C s | � tj| jd�� � 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 s0 dD ]&}dD ] }| � d| �||g�k� q qd S )N)rh r^ )r_ ri z dns-route53zcertbot-route53:auth)r� rD )r Z help_flagZtopicr r r �test_route53_no_revert� s z ParseTest.test_route53_no_revertc C s | � dg�}| �|j� d S )Nz--no-permissions-check)r7 ro Zno_permissions_checkr~ r r r �"test_no_permissions_check_accepted� s z,ParseTest.test_no_permissions_check_acceptedN),r r r! r" r/ � staticmethodr4 r7 rD r? r@ rR rU r] rr rt rv rz 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, ? sR
<
' r, c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � DefaultTestz Tests for certbot.cli._Default.c C s t �� | _t �� | _d S r- )r Z_Default�default1�default2r* r r r r/ � s
zDefaultTest.setUpc C s | � | j� | � | j� d S r- )r� r� r� r* r r r � test_boolean� s zDefaultTest.test_booleanc C s | � | j| j� d S r- )r r� r� r* r r r �
test_equality� s zDefaultTest.test_equalityc C s | � t| j�t| j�� d S r- )r �hashr� r� r* r r r � test_hash� s zDefaultTest.test_hashN)r r r! r" r/ r� r� r� r r r r r� � s
r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� SetByCliTestz3Tests for certbot.set_by_cli and related functions.c C s t t� d S r- r. r* r r r r/ � s zSetByCliTest.setUpc C s | � tdd�� d�� d S )Nr� z--deploy-hook foorM )ro �_call_set_by_clir| r* r r r �test_deploy_hook� s
�z SetByCliTest.test_deploy_hookc C s" d� � }d}| �td||�� d S )Nz -w /var/www/html -d example.comrM Z webroot_map)r| ro r� )r r2 �verbr r r �test_webroot_map� s z SetByCliTest.test_webroot_mapc C s$ t �dd� t �dd� | �� d S )N�manual_public_ip_logging_ok�manual_auth_hook�manual�r Zreport_config_interaction�&_test_report_config_interaction_commonr* r r r �"test_report_config_interaction_str� s
� z/SetByCliTest.test_report_config_interaction_strc C s$ t �dd� t �dd� | �� d S )N)r� )r� )r� r� r* r r r �'test_report_config_interaction_iterable� s
� z4SetByCliTest.test_report_config_interaction_iterablec C sh dg}d}dD ]}| � t|||�� qdtj_ddg}dD ]}| � t|||�� q:| �td||�� dS ) a Tests implied interaction between manual flags.
--manual implies --manual-auth-hook which implies
--manual-public-ip-logging-ok. These interactions don't actually
exist in the client, but are used here for testing purposes.
z--manualrM )r� r� r� Nz--manual-auth-hookZcommand)r� r� r� )ro r� r �
set_by_cliZdetectorr� )r r2 r� rG r r r r� s z3SetByCliTest._test_report_config_interaction_commonN)
r r r! r" r/ r� r� r� r� r� r r r r r� � s r� c C sX t �d��D}t�� �0 ||_||_t�| �W 5 Q R � W 5 Q R � S Q R X W 5 Q R X d S )Nzcertbot.cli.helpful_parser)r? r@ r5 r6 r2 r� r r� )�varr2 r� Z mock_parserr r r r� s
r� �__main__)%r" r rE rN Zunittestr? r= Z six.movesr Zacmer Zcertbot.tests.utilZtests�utilr5 Zcertbotr r r Zcertbot.compatr r Zcertbot.pluginsr r
ZPluginsRegistryZfind_allr0 r ZTestCaser# r, r� r� r� r �mainr r r r �<module> s8
6