/usr/lib/python3.9/site-packages/certbot/_internal/tests/__pycache__
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 �|�\}}|t j�|�ks�J �||ks�J �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�abspath)�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 jdkrBt�d�dks J �t�d�dks.J �t�d�dksxJ �n6t�d�dksTJ �t�d�d ksfJ �t�d�d ksxJ �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 �r! r% r% r& �test_default_directories@ s z(FlagDefaultTest.test_default_directoriesN)r( r) r* r+ r3 r% r% r% r& r, = s r, )�attr�valuec C s$ t | |�|ksJ �| �|�s J �d S �N)�getattr� set_by_user)� namespacer4 r5 r% r% r& � assert_set_by_user_with_valueK s r: )r4 r5 �sourcec C s( t | |�|ksJ �| j| |ks$J �d S r6 )r7 Zargument_sources)r9 r4 r5 r; r% r% r& �assert_value_and_sourceP s r< 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�r> 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_utilr= rC rB r% r% r& �parse] s zParseTest.parserB 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 r6 )r )�messager> �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 �patchrD rE Z notification� side_effectr r � SystemExitrC �getvalue)r! r> rK Zmock_get_utilityr% rI 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_DEFAULTSr1 )�v�� tmp_configr% r& �<lambda>z s �z0ParseTest.test_cli_ini_domains.<locals>.<lambda>�certonly�domains�wzdomains = example.com� example.com�renew) �tempfile�NamedTemporaryFile�closerP rF r< r �DEFAULTr r1 r � CONFIG_FILE�RUNTIME)r! Zmock_flag_defaultZshimr9 �file_hr% rW r& �test_cli_ini_domainst s � ( z ParseTest.test_cli_ini_domainsc C s> | � g �}dD ]*}t||�t�|�ks*J �|�|�rJ �qd S )N)r. r0 r/ )rF r7 r r r8 )r! r9 �dr% r% r& � test_no_args� s zParseTest.test_no_argsc C s� d}d}d}d}t �d��, | �dd|ddd dd dg �}W d � n1 sL0 Y |jtj�|�kslJ �|jtj�|�ks�J �|jtj�|�ks�J �|j tj�|�ks�J �d S ) N�cert�key�chain� fullchainz certbot._internal.main.install�install� --cert-path� --key-pathz --chain-pathz--fullchain-path) r rO rF Z cert_pathr r r Zkey_pathZ chain_pathZfullchain_path)r! ri rj rk rl r9 r% r% r& �test_install_abspath� s �$z ParseTest.test_install_abspathc C sZ | � dg� | � ddg�}d|v s&J �d|v s2J �d|v s>J �d|vsJJ �d|vsVJ �d|vsbJ �d |vsnJ �| � d d g�}d tv r�d |v s�J �d|vs�J �d |vs�J �| � d g�}d tv r�d|v s�J �n d|v s�J �| � ddg�}d|vs�J �d|v s�J �d|v �s J �| � d dg�}d|v �s&J �| � d dg�}d|v �sBJ �| � d dg�}d|v�s^J �| � ddg�}d|v �szJ �d|v �s�J �| � ddg�}d|v �s�J �d|v �s�J �d|v �s�J �d|v �s�J �d |v �s�J �| � d d g�}d|v�s�J �d|v�sJ �| � d g�}tj|v �s"J �tjd d � |v �s:J �d|v�sHJ �d|v�sVJ �d S ) N�--help�all�--configurator� how a certificate is deployed�--webroot-path�--text�%s�{0}� --renew-hook�-hZnginxz --nginx-ctlz --checkpointszUse the Nginx pluginz (the certbot nginx plugin is not�pluginsz --preparez"plugins" subcommandr^ z--keepZ automation�revokerm rn ro �--reason�--delete-after-revoke�--no-delete-after-revoke�register�d )rS rA r Z SHORT_USAGEZCOMMAND_OVERVIEW�r! �outr% r% r& � test_help� sZ zParseTest.test_helpc C s� | � dg� | � ddg�}d|v s&J �d|v s2J �d|v s>J �d|vsJJ �d|vsVJ �d|vsbJ �| � dd g�}d |v s|J �d |v s�J �| � dd g�}d |v s�J �d |v s�J �d S ) N�helprr rs rt ru rv rw rx rm rn ro r| �rS r� 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-dr] r[ ztrailing.period.com.ztrailing.period.comz-example.com,another.net,third.org,example.com)r] � another.netz third.orgz --domainsz#example.com,another.net,example.comr� �rF r: )r! � short_argsr9 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) rF r �HTTP01�typZDNS01r: r rO r r rQ )r! r� r9 Zexpectedr% r% r& �test_preferred_challenges s z#ParseTest.test_preferred_challengesc C s | � d�� �}t|dd� d S )N�--server example.com�serverr] )rF �splitr: �r! r9 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)rF r: r< r rd 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� r� 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 nP tjyt } z6dt|�v sFJ �|D ]}|t|�v sJJ �qJW Y d }~n d }~0 0 d S )Nz6The following flags didn't conflict with --server: {0}z, �--server)rF Zfail�formatr r �Error�str)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� )rF r: r � STAGING_URIr� r� )r! r� r9 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)r: r< r rd r r� rb )r! r9 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}ZauthFrZ r^ Zfake_account_dirTr� r] r� r� )r� r] r� )r_ Zmkdtempr� r� r r r r� rF r� r r r r Z ACCOUNTS_DIRr �mkdirr: r� r r r< r rd r� )r! r. r� Z account_dir�config� conflictsr% r% r& �test_dry_run_flagC s< (��� �zParseTest.test_dry_run_flagc C sT d}t �|�}| �d�|��� �}|�|�s0J �d}|�|�rBJ �|�d�rPJ �d S )NZ rsa_key_sizez--rsa-key-size {0}r. � authenticator�r r rF r� r� r8 )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 sP | � d�}|�d�sJ �|�d�s&J �| � d�}|�d�s>J �|�d�rLJ �d S )Nz--apacheZ installerr� z--installer webroot)rF r8 )r! r� r% r% r& �)test_user_set_installer_and_authenticatorw s z3ParseTest.test_user_set_installer_and_authenticatorc C s4 d}t �|�}| �d�|��� �}|�|�s0J �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 sh d}t �|�}| �d�|��� �}|�|�s0J �t�t�� | �d� W d � n1 sZ0 Y d S )NZkey_typez--key-type {0}z--key-type foo) r r rF r� r� r8 r r rQ )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 s0 d� � }tj�� }t�||�}|�d�s,J �d S )Nzrenew --deploy-hook foo� renew_hook)r� r �PluginsRegistry�find_allr r@ r8 )r! Zunused_mock_validate_hooksr> r{ r� r% r% r& �test_user_set_deploy_hook� s z#ParseTest.test_user_set_deploy_hookz3certbot._internal.plugins.webroot._validate_webrootc C s6 d� � }d|_tj�� }t�||�}|�d�s2J �d S )Nz%renew -w /var/www/html -d example.comz /var/www/htmlZ webroot_map)r� Z return_valuer r� r� r r@ r8 )r! Zmock_validate_webrootr> r{ r� r% r% r& �test_user_set_webroot_map� s z#ParseTest.test_user_set_webroot_mapc C sh t j�� D ]$\}}| �d|g�}|j|ks J �q t j�� D ](\}}| �d|�� g�}|j|ks:J �q:d S )Nr} )r ZREVOCATION_REASONS�itemsrF �reason�upper)r! r� �coder9 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� rF r� r2 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 rO r r rQ rF r� r2 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 ry � --deploy-hook�--disable-hook-validation� deploy_hookr� r� �r! r5 r9 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� r� 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� r2 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� ry r� r� r� r� r� r% r% r& �#test_renew_hook_matches_deploy_hook� s � z-ParseTest.test_renew_hook_matches_deploy_hookc C s2 d}| � d|dg�}|jd u s"J �t|d|� d S )Nr ry r� r� )rF 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 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� r2 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)rF r: �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 )Nr[ r� r] ) rF r< r rb r r� r� r� r[ 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_hooksFr� 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)rF r< r rb r� r% r% r& � test_no_directory_hooks_unset� s z'ParseTest.test_no_directory_hooks_unsetc C s | � dg�}t|dd� d S )Nr~ �delete_after_revokeTr� r� r% r% r& �test_delete_after_revoke� s z"ParseTest.test_delete_after_revokec C s* | � g �}|jd u sJ �|�d�r&J �d S )Nr� )rF r� r8 r� r% r% r& � test_delete_after_revoke_default� s z*ParseTest.test_delete_after_revoke_defaultc C s | � dg�}t|dd� d S )Nr r� Fr� 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� r2 r% r% r& � test_allow_subset_with_wildcard s z)ParseTest.test_allow_subset_with_wildcardc C s. dD ]$}dD ]}d| � ||g�vs J �q qd S )N)rz rq )rr r{ z dns-route53zcertbot-route53:authr� )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� rF r2 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.edurZ r\ z server = z-cr� ) r rF r< r rb r_ r` ra r r1 r rc Z COMMAND_LINE)r! Z DEFAULT_VALUEZCONFIG_FILE_VALUEZCOMMAND_LINE_VALUEr9 rX re 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-dirr� zdns-01r� Fr\ zpreferred-chal = dnsrZ z--config)rF r: r_ r` ra r r1 r )r! r9 rX re 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.comr[ r] �runz--pre-hook="foo=bar"Zpre_hookz "foo=bar")rF r: r r� r� r r@ )r! Zunsused_mock_validate_hooksr9 r{ 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.comr� r� r% r% r& �test_adjacent_short_argsH s z"ParseTest.test_adjacent_short_argsc C s | � ddg�}t|ddg� d S )NrZ z -d foo.comr[ zfoo.comr� r� r% r% r& � test_arg_with_contained_spacesV s z(ParseTest.test_arg_with_contained_spacesN)9r( r) r* r+ � staticmethodr r� r rC rF rS r rO rf rh rp 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% r% r% r& r= U sl : ( r= �__main__r� )4r+ r rT � importlibr Z reload_modulerM �sysr_ �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�utilrD r r� r� rA r ZTestCaser, r� r: r< r= r( �exit�main�argv�__file__r% r% r% r& �<module> sB
.
Edit
..
Edit
__init__.cpython-39.opt-1.pyc
Edit
__init__.cpython-39.pyc
Edit
account_test.cpython-39.opt-1.pyc
Edit
account_test.cpython-39.pyc
Edit
auth_handler_test.cpython-39.opt-1.pyc
Edit
auth_handler_test.cpython-39.pyc
Edit
cert_manager_test.cpython-39.opt-1.pyc
Edit
cert_manager_test.cpython-39.pyc
Edit
cli_test.cpython-39.opt-1.pyc
Edit
cli_test.cpython-39.pyc
Edit
client_test.cpython-39.opt-1.pyc
Edit
client_test.cpython-39.pyc
Edit
configuration_test.cpython-39.opt-1.pyc
Edit
configuration_test.cpython-39.pyc
Edit
crypto_util_test.cpython-39.opt-1.pyc
Edit
crypto_util_test.cpython-39.pyc
Edit
eff_test.cpython-39.opt-1.pyc
Edit
eff_test.cpython-39.pyc
Edit
error_handler_test.cpython-39.opt-1.pyc
Edit
error_handler_test.cpython-39.pyc
Edit
errors_test.cpython-39.opt-1.pyc
Edit
errors_test.cpython-39.pyc
Edit
helpful_test.cpython-39.opt-1.pyc
Edit
helpful_test.cpython-39.pyc
Edit
hook_test.cpython-39.opt-1.pyc
Edit
hook_test.cpython-39.pyc
Edit
lock_test.cpython-39.opt-1.pyc
Edit
lock_test.cpython-39.pyc
Edit
log_test.cpython-39.opt-1.pyc
Edit
log_test.cpython-39.pyc
Edit
main_test.cpython-39.opt-1.pyc
Edit
main_test.cpython-39.pyc
Edit
ocsp_test.cpython-39.opt-1.pyc
Edit
ocsp_test.cpython-39.pyc
Edit
renewal_test.cpython-39.opt-1.pyc
Edit
renewal_test.cpython-39.pyc
Edit
renewupdater_test.cpython-39.opt-1.pyc
Edit
renewupdater_test.cpython-39.pyc
Edit
reverter_test.cpython-39.opt-1.pyc
Edit
reverter_test.cpython-39.pyc
Edit
storage_test.cpython-39.opt-1.pyc
Edit
storage_test.cpython-39.pyc
Edit
util_test.cpython-39.opt-1.pyc
Edit
util_test.cpython-39.pyc
Edit