/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/__pycache__
a }|�gn8 � @ sz d Z ddlZddlZddlZddlZddlmZ ddlZddlZddl m Z ddl m Z ddl m Z ddl mZ ddl mZ dd lmZ dd lmZ dd lmZ dd lmZ ej�e�d ��Ze je� e j dd�e j �ded�Z G dd� dej �Z!G dd� dej �Z"G dd� dej#�Z$G dd� dej �Z%G dd� dej �Z&G d d � d ej'�Z(e)d k�rve�*e�+ej,d d� e-g �� dS ) z!Tests for certbot.plugins.common.� N)�mock)� challenges)�messages)� achallenges)� crypto_util)�errors)� filesystem)�os)� acme_util)�utilzrsa512_key.pems token1)�tokenzencryption-example.demo)Zchallb�domainZ account_keyc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �NamespaceFunctionsTestz7Tests for certbot.plugins.common.*_namespace functions.c C s ddl m} d S )Nr )�option_namespace)�certbot.plugins.commonr )�selfr � r �O/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/common_test.py�test_option_namespace s z,NamespaceFunctionsTest.test_option_namespacec C s ddl m} d S �Nr )�dest_namespace�r r �r r r r r �test_dest_namespace# s z*NamespaceFunctionsTest.test_dest_namespacec C s ddl m} d S r r r r r r � test_dest_namespace_with_dashes' s z6NamespaceFunctionsTest.test_dest_namespace_with_dashesN)�__name__� __module__� __qualname__�__doc__r r r r r r r r s r c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dS )� PluginTestz'Test for certbot.plugins.common.Plugin.c C s@ ddl m} G dd� d|�}|| _t�� | _|| jdd�| _d S )Nr )�Pluginc @ s4 e Zd Zdd�dd�Zed�dd�Zedd� �ZdS ) z$PluginTest.setUp.<locals>.MockPluginN)�returnc S s d S �Nr �r r r r �prepare3 s z,PluginTest.setUp.<locals>.MockPlugin.preparec S s dS )N�infor r# r r r � more_info6 s z.PluginTest.setUp.<locals>.MockPlugin.more_infoc S s |dddd d� d S )Nzfoo-bar�different_to_foo_bar� ��dest�x�yr )�cls�addr r r �add_parser_arguments9 s z9PluginTest.setUp.<locals>.MockPlugin.add_parser_arguments)r r r r$ �strr& � classmethodr/ r r r r � MockPlugin2 s r2 r ��config�name)r r � plugin_clsr � MagicMockr4 Zplugin)r r r2 r r r �setUp/ s zPluginTest.setUpc C s d S r" r r# r r r � test_initA s zPluginTest.test_initc C s d S r" r r# r r r r E s z PluginTest.test_option_namespacec C s d S r" r r# r r r �test_option_nameH s zPluginTest.test_option_namec C s d S r" r r# r r r r K s z PluginTest.test_dest_namespacec C s d S r" r r# r r r � test_destN s zPluginTest.test_destc C s d S r" r r# r r r � test_confR s zPluginTest.test_confc C s. t �� }| j�|d� |jjdddd d� d S )Nr z--mock-foo-barr' r( r) )r r7 r6 Zinject_parser_options� add_argument�assert_called_once_with)r �parserr r r �test_inject_parser_optionsU s �z%PluginTest.test_inject_parser_optionsc C s d S r" r r# r r r �test_fallback_auth_hint] s z"PluginTest.test_fallback_auth_hintN) r r r r r8 r9 r r: r r; r< r@ rA r r r r r , s r c s� e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Z� ZS ) � InstallerTestz+Tests for certbot.plugins.common.Installer.c sB t � �� t�| jj� ddlm} || jdd�| _| jj | _ d S )Nr )�DummyInstallerZ Installerr3 ) �superr8 r �mkdirr4 Z config_dirZcertbot.tests.utilrC � installer�reverter)r rC �� __class__r r r8 h s �zInstallerTest.setUpc C s ddh}d}| � d||� d S )N�foo.bar�baz.qux�foo bar baz qux�add_to_checkpoint��_test_wrapped_method)r �files� save_notesr r r �test_add_to_real_checkpointq s z)InstallerTest.test_add_to_real_checkpointc C s | � d� d S )NF�� _test_add_to_checkpoint_commonr# r r r � test_add_to_real_checkpoint2v s z*InstallerTest.test_add_to_real_checkpoint2c C s | � d� d S )NTrS r# r r r � test_add_to_temporary_checkpointy s z.InstallerTest.test_add_to_temporary_checkpointc C s@ ddh}d}t j| jj|d�}|r(d}nd}| �||||� d S )NrJ rK rL )� temporaryZadd_to_temp_checkpointrM )� functools�partialrF rM �_test_adapted_method)r rW rP rQ �installer_func�reverter_func_namer r r rT | s �z,InstallerTest._test_add_to_checkpoint_commonc C s | � dd� d S )NZfinalize_checkpointZfoorN r# r r r �test_finalize_checkpoint� s z&InstallerTest.test_finalize_checkpointc C s | � d� d S )NZrecovery_routinerN r# r r r �test_recovery_routine� s z#InstallerTest.test_recovery_routinec C s | � d� d S )NZrevert_temporary_configrN r# r r r � test_revert_temporary_config� s z*InstallerTest.test_revert_temporary_configc C s | � dd� d S )NZrollback_checkpoints�* rN r# r r r �test_rollback_checkpoints� s z'InstallerTest.test_rollback_checkpointsc O s* t | j|�}| j||g|�R i |�� dS )z�Test a wrapped reverter method. :param str name: name of the method to test :param tuple args: position arguments to method :param dict kwargs: keyword arguments to method N)�getattrrF rZ )r r5 �args�kwargsr[ r r r rO � s z"InstallerTest._test_wrapped_methodc O s� t j�| j|��p}||i |�� |j|i |�� tj|_t� tj �� ||i |�� W d � n1 sh0 Y W d � n1 s�0 Y dS )a� Test an adapted reverter method :param callable installer_func: installer method to test :param str reverter_func_name: name of the method on the reverter that should be called :param tuple passed_args: positional arguments passed from installer method to the reverter method :param dict passed_kargs: keyword arguments passed from installer method to the reverter method N) r �patch�objectrG r> r Z ReverterErrorZ side_effect�pytest�raisesZ PluginError)r r[ r\ Z passed_argsZ passed_kwargsZ reverter_funcr r r rZ � s z"InstallerTest._test_adapted_methodc C s | j �� d S r" )rF Zinstall_ssl_dhparamsr# r r r �test_install_ssl_dhparams� s z'InstallerTest.test_install_ssl_dhparamsc C s ddl m} t�|�S )Nr )�SSL_DHPARAMS_SRC)�certbot._internal.constantsrj r � sha256sum)r rj r r r �_current_ssl_dhparams_hash� s z(InstallerTest._current_ssl_dhparams_hashc C s ddl m} d S )Nr )�ALL_SSL_DHPARAMS_HASHES)rk rn )r rn r r r �$test_current_file_hash_in_all_hashes� s z2InstallerTest.test_current_file_hash_in_all_hashes)r r r r r8 rR rU rV rT r] r^ r_ ra rO rZ ri rm ro � __classcell__r r rH r rB e s rB c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�AddrTestz&Tests for certbot.plugins.common.Addr.c C sp ddl m} |�d�| _|�d�| _|�d�| _|�d�| _|�d�| _|�d�| _|�d �| _ |�d �| _ d S ) Nr ��Addr� 192.168.1.1� 192.168.1.1:*z192.168.1.1:80� [fe00::1]� [fe00::1]:*z [fe00::1]:80z [fe00::1]:5z [fe00:1:2:3:4:5:6:7:8:9]:8080) r rs � fromstring�addr1�addr2Zaddr3�addr4�addr5Zaddr6Zaddr7Zaddr8�r rs r r r r8 � s zAddrTest.setUpc C s d S r" r r# r r r �test_fromstring� s zAddrTest.test_fromstringc C s d S r" r r# r r r �test_str� s zAddrTest.test_strc C s d S r" r r# r r r �test_get_addr_obj� s zAddrTest.test_get_addr_objc C s ddl m} d S )Nr rr )r rs r} r r r �test_eq� s zAddrTest.test_eqc C s` ddl m} | j| jh}|�d�}|�d�}||h}| j| jh}|�d�}|�d�}||h} d S )Nr rr rt ru rv rw )r rs ry rz rx r{ r| ) r rs Zset_aZaddr1bZaddr2bZset_bZset_cZaddr4bZaddr5bZset_dr r r �test_set_inclusion s zAddrTest.test_set_inclusionN) r r r r r8 r~ r r� r� r� r r r r rq � s rq c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �ChallengePerformerTestz4Tests for certbot.plugins.common.ChallengePerformer.c C s" t �� }ddlm} ||�| _d S )Nr )�ChallengePerformer)r r7 r r� � performer)r Z configuratorr� r r r r8 s z ChallengePerformerTest.setUpc C s | j �td� d S )Nr )r� Z add_chall�ACHALLr# r r r �test_add_chall s z%ChallengePerformerTest.test_add_challc C s8 t �t�� | j�� W d � n1 s*0 Y d S r" )rg rh �NotImplementedErrorr� Zperformr# r r r � test_perform# s z#ChallengePerformerTest.test_performN)r r r r r8 r� r� r r r r r� s r� c s` e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z � Z S )� InstallVersionControlledFileTestzATests for certbot.plugins.common.install_version_controlled_file.c s� t � �� dg| _tj�| jd�| _tj�| jd�| _tj�| jd�| _ tj�| jd�| _ | j | j fD ]J}t |d��}|� |� W d � n1 s�0 Y | j� t�|�� qfd S )NZ someotherhashzoptions-ssl-dest.confz.options-ssl-conf.txtzoptions-ssl-old.confzoptions-ssl-src.conf�w)rD r8 �hashesr �path�joinZtempdir� dest_path� hash_path�old_path� source_path�open�write�appendr rl )r r� �frH r r r8 + s (z&InstallVersionControlledFileTest.setUpc C s& ddl m} || j| j| j| j� d S )Nr )� install_version_controlled_file)r r� r� r� r� r� )r r� r r r �_call7 s �z&InstallVersionControlledFileTest._callc C s t �| j�S r" )r rl r� r# r r r �_current_file_hash> s z3InstallVersionControlledFileTest._current_file_hashc C s d S r" r r# r r r �_assert_current_fileA s z5InstallVersionControlledFileTest._assert_current_filec C s | � � | �� d S r" )r� r� r# r r r � test_no_fileF s z-InstallVersionControlledFileTest.test_no_filec C s" t d�D ]}| �� | �� qd S )N� )�ranger� r� )r �_r r r �test_current_fileK s z2InstallVersionControlledFileTest.test_current_filec C s$ t �| j| j� | �� | �� d S r" )�shutil�copyfiler� r� r� r� r# r r r �!test_prev_file_updates_to_currentQ s zBInstallVersionControlledFileTest.test_prev_file_updates_to_currentc C st | � � t| jd��}|�d� W d � n1 s40 Y t�d��}| � � W d � n1 sf0 Y d S )N�a� a new line for the wrong hash � certbot.plugins.common.logger)r� r� r� r� r re )r � mod_ssl_conf� mock_loggerr r r �3test_manually_modified_current_file_does_not_updateV s ( zTInstallVersionControlledFileTest.test_manually_modified_current_file_does_not_updatec C s� t | jd��}|�d� W d � n1 s,0 Y t | jd��}|�d� W d � n1 sb0 Y t�d��}| �� W d � n1 s�0 Y t�d��}| �� W d � n1 s�0 Y d S )Nr� r� r� Zhashofanoldversionr� )r� r� r� r� r re r� )r r� r� r� r r r �&test_manually_modified_past_file_warnsc s (( zGInstallVersionControlledFileTest.test_manually_modified_past_file_warns)r r r r r8 r� r� r� r� r� r� r� r� rp r r rH r r� ( s r� �__main__r( ).r rX r� �sysZunittestr ZjosepyZjoserg Zacmer r Zcertbotr r r Zcertbot.compatr r Z certbot.testsr r Z test_utilZJWKRSA�loadZ load_vectorZAUTH_KEYZ"KeyAuthorizationAnnotatedChallengeZchall_to_challbZHTTP01ZSTATUS_PENDINGr� ZTestCaser r ZConfigTestCaserB rq r� ZTempDirTestCaser� r �exit�main�argv�__file__r r r r �<module> s> ��9_QL
.
Edit
..
Edit
__init__.cpython-39.opt-1.pyc
Edit
__init__.cpython-39.pyc
Edit
common_test.cpython-39.opt-1.pyc
Edit
common_test.cpython-39.pyc
Edit
disco_test.cpython-39.opt-1.pyc
Edit
disco_test.cpython-39.pyc
Edit
dns_common_test.cpython-39.opt-1.pyc
Edit
dns_common_test.cpython-39.pyc
Edit
enhancements_test.cpython-39.opt-1.pyc
Edit
enhancements_test.cpython-39.pyc
Edit
manual_test.cpython-39.opt-1.pyc
Edit
manual_test.cpython-39.pyc
Edit
null_test.cpython-39.opt-1.pyc
Edit
null_test.cpython-39.pyc
Edit
selection_test.cpython-39.opt-1.pyc
Edit
selection_test.cpython-39.pyc
Edit
standalone_test.cpython-39.opt-1.pyc
Edit
standalone_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
webroot_test.cpython-39.opt-1.pyc
Edit
webroot_test.cpython-39.pyc
Edit