Your IP : 216.73.216.172
U
e�_�; � @ sH d Z ddlZddlZddlZddlZddlZddlZddlm Z ddl
m Z ddl
m Z ddl
m
Z
ddlmZ ddlmZ dd lmZ dd
lmZ ej�e�d ��Ze je�e jd d
�d�ded�Z G dd� dej �Z G dd� dej �Z G dd� dej �Z!G dd� dej �Z"G dd� dej �Z#G dd � d ej$�Z%e&d k�rDe�'� dS ) z!Tests for certbot.plugins.common.� N)�
challenges)� achallenges)� crypto_util)�errors)�os)�
filesystem)� acme_util)�utilzrsa512_key.pems token1)�token�pendingzencryption-example.demo)ZchallbZdomainZ account_keyc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �NamespaceFunctionsTestz7Tests for certbot.plugins.common.*_namespace functions.c C s ddl m} | �d|d�� d S )Nr )�option_namespacezfoo-�foo)�certbot.plugins.commonr
� assertEqual)�selfr
� r �=/usr/lib/python3/dist-packages/certbot/plugins/common_test.py�test_option_namespace s z,NamespaceFunctionsTest.test_option_namespacec C s ddl m} | �d|d�� d S )Nr ��dest_namespaceZfoo_r �r r r �r r r r r �test_dest_namespace! s z*NamespaceFunctionsTest.test_dest_namespacec C s ddl m} | �d|d�� d S )Nr r Zfoo_bar_�foo-barr 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 @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� 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 @ s e Zd Zedd� �ZdS )z$PluginTest.setUp.<locals>.MockPluginc S s |dddd d� d S )Nr �different_to_foo_bar� ��dest�x�yr )�cls�addr r r �add_parser_arguments1 s z9PluginTest.setUp.<locals>.MockPlugin.add_parser_argumentsN)r r r � classmethodr* r r r r �
MockPlugin0 s r, �mock��config�name)r r! �
plugin_clsr- � MagicMockr/ �plugin)r r! r, r r r �setUp- s
zPluginTest.setUpc C s& | � d| jj� | � | j| jj� d S )Nr- )r r3 r0 r/ �r r r r � test_init9 s zPluginTest.test_initc C s | � d| jj� d S )Nzmock-)r r3 r
r5 r r r r = s z PluginTest.test_option_namespacec C s | � d| j�d�� d S )Nz mock-foo_bar�foo_bar)r r3 Z option_namer5 r r r �test_option_name@ s zPluginTest.test_option_namec C s | � d| jj� d S )NZmock_)r r3 r r5 r r r r C s z PluginTest.test_dest_namespacec C s, | � d| j�d�� | � d| j�d�� d S )N� mock_foo_barr r7 )r r3 r% r5 r r r � test_destF s zPluginTest.test_destc C s | � | jj| j�d�� d S )Nr )r r/ r9 r3 �confr5 r r r � test_confJ s zPluginTest.test_confc C s. t �� }| j�|d� |jjdddd d� d S )Nr- z--mock-foo-barr" r# r$ )r- r2 r1 Zinject_parser_options� add_argument�assert_called_once_with)r �parserr r r �test_inject_parser_optionsM s �z%PluginTest.test_inject_parser_optionsN) r r r r r4 r6 r r8 r r: 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 dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Z� ZS )"�
InstallerTestz+Tests for certbot.plugins.common.Installer.c sF t t| ��� t�| jj� ddlm} || jdd�| _ | j j
| _
d S )Nr )� InstallerrB r. ) �superrA r4 r �mkdirr/ Z
config_dirr rB � installer�reverter)r rB �� __class__r r r4 Y s �zInstallerTest.setUpc C s t d�}d}| �d||� d S )N�zfoo.barzbaz.qux�foo bar baz qux�add_to_checkpoint)�set�_test_wrapped_method)r �files�
save_notesr r r �test_add_to_real_checkpointb s z)InstallerTest.test_add_to_real_checkpointc C s | � d� d S )NF�� _test_add_to_checkpoint_commonr5 r r r � test_add_to_real_checkpoint2g s z*InstallerTest.test_add_to_real_checkpoint2c C s | � d� d S )NTrQ r5 r r r � test_add_to_temporary_checkpointj s z.InstallerTest.test_add_to_temporary_checkpointc C s@ t d�}d}tj| jj|d�}|r(d}nd}| �||||� d S )NrI rJ )� temporaryZadd_to_temp_checkpointrK )rL � functools�partialrE rK �_test_adapted_method)r rU rN rO �installer_func�reverter_func_namer r r rR m s
�z,InstallerTest._test_add_to_checkpoint_commonc C s | � dd� d S )NZfinalize_checkpointr �rM r5 r r r �test_finalize_checkpoint{ s z&InstallerTest.test_finalize_checkpointc C s | � d� d S )NZrecovery_routiner[ r5 r r r �test_recovery_routine~ s z#InstallerTest.test_recovery_routinec C s | � d� d S )NZrevert_temporary_configr[ r5 r r r � test_revert_temporary_config� s z*InstallerTest.test_revert_temporary_configc C s | � dd� d S )NZrollback_checkpoints�* r[ r5 r r r �test_rollback_checkpoints� s z'InstallerTest.test_rollback_checkpointsc C s | � d� d S )N�view_config_changesr[ r5 r r r �test_view_config_changes� s z&InstallerTest.test_view_config_changesc C s2 t �� � t jdddd� | j�� W 5 Q R X d S )N�errorz.*view_config_changesz .*common$)�module)�warnings�catch_warnings�filterwarningsrE ra r5 r r r �+test_view_config_changes_warning_supression� s
z9InstallerTest.test_view_config_changes_warning_supressionc O s$ t | j|�}| j||f|�|� 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)�getattrrE rX )r r0 �args�kwargsrY r r r rM � s z"InstallerTest._test_wrapped_methodc O sT t j�| j|��:}|||� |j||� tj|_| jtj |f|�|� W 5 Q R X 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�objectrF r> r Z
ReverterErrorZ side_effect� assertRaisesZ PluginError)r rY rZ Z passed_argsZ
passed_kwargsZ
reverter_funcr r r rX � s
���z"InstallerTest._test_adapted_methodc C s$ | j �� | �tj�| j j�� d S �N)rE Zinstall_ssl_dhparams�
assertTruer �path�isfileZ ssl_dhparamsr5 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.constantsrt r � sha256sum)r rt r r r �_current_ssl_dhparams_hash� s z(InstallerTest._current_ssl_dhparams_hashc C s$ ddl m} | �| �� |kd� d S )Nr )�ALL_SSL_DHPARAMS_HASHESzwConstants.ALL_SSL_DHPARAMS_HASHES must be appended with the sha256 hash of self.config.ssl_dhparams when it is updated.)ru rx rp rw )r rx r r r �$test_current_file_hash_in_all_hashes� s �z2InstallerTest.test_current_file_hash_in_all_hashes)r r r r r4 rP rS rT rR r\ r] r^ r` rb rh rM rX rs rw ry �
__classcell__r r rG r rA V s"
rA c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�AddrTestz-Tests for certbot.client.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:*�192.168.1.1:80� [fe00::1]� [fe00::1]:*� [fe00::1]:80z [fe00::1]:5z [fe00:1:2:3:4:5:6:7:8:9]:8080) r r} �
fromstring�addr1�addr2�addr3�addr4�addr5�addr6�addr7�addr8�r r} r r r r4 � s zAddrTest.setUpc C s$ | � | j�� d� | � | j�� d� | � | j�� d� | � | j�� d� | � | j�� d� | � | j�� d� | � | j�� d� | � | j�� d� | � | j�� d� | � | j�� d� | � | j�� d� | � | j�� d� | � | j� � d� | � | j� � d� | � | j
�� d� | � | j � � d� d S ) Nr~ � �*Z80r� zfe00:0:0:0:0:0:0:1�5zfe00:1:2:3:4:5:6:7) r r� Zget_addrZget_portr� r� r� r� r� Zget_ipv6_explodedr� r� r5 r r r �test_fromstring� s, � � �zAddrTest.test_fromstringc C sp | � t| j�d� | � t| j�d� | � t| j�d� | � t| j�d� | � t| j�d� | � t| j�d� d S )Nr~ r r� r� r� r� )r �strr� r� r� r� r� r� r5 r r r �test_str� s zAddrTest.test_strc C s� | � t| j�d��d� | � t| j�d��d� | � t| j�d��d� | � t| j�d��d� | � t| j�d��d� | � t| j�d��d � d S )
NZ443z192.168.1.1:443r� r~ r� r z
[fe00::1]:443r� r� )r r� r� � get_addr_objr� r� r� r5 r r r �test_get_addr_obj� s zAddrTest.test_get_addr_objc C s� | � | j| j�d�� | �| j| j� | �| jdk� | � | j| j�d�� | �| j| j� | �| jdk� ddlm } | � | j|�
d�� | � | j|�
d�� d S )Nr� i
r r| z
[fe00:0:0::1]z[fe00:0::0:0:1]) r r� r� r� �assertNotEqual� assertFalser� r� r r} r� r� r r r �test_eq� s zAddrTest.test_eqc
C s� ddl m} t| j| jg�}|�d�}|�d�}t||g�}| �||� t| j| jg�}|�d�}|�d�}t||g�} | �|| � d S )Nr r| r~ r r� r� ) r r} rL r� r� r� r r� r� )
r r} Zset_aZaddr1bZaddr2bZset_bZset_cZaddr4bZaddr5bZset_dr r r �test_set_inclusion s
zAddrTest.test_set_inclusionN)
r r r r r4 r� r� r� r� r� r r r r r{ � s
r{ c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �ChallengePerformerTestz4Tests for certbot.plugins.common.ChallengePerformer.c C s" t �� }ddlm} ||�| _d S )Nr )�ChallengePerformer)r- r2 r r� � performer)r Z configuratorr� r r r r4 s z ChallengePerformerTest.setUpc C s8 | j �td� | �dt| j j�� | �dg| j j� d S )Nr r# )r� Z add_chall�ACHALLr �lenZachalls�indicesr5 r r r �test_add_chall s z%ChallengePerformerTest.test_add_challc C s | � t| jj� d S ro )rn �NotImplementedErrorr� Zperformr5 r r r � test_perform$ s z#ChallengePerformerTest.test_performN)r r r r r4 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 t| ��� dg| _tj�| jd�| _tj�| jd�| _ tj�| jd�| _
tj�| jd�| _ | j | j
fD ]6}t |d��}|�
|� W 5 Q R X | j�t�|�� qjd S )NZ
someotherhashzoptions-ssl-dest.confz.options-ssl-conf.txtzoptions-ssl-old.confzoptions-ssl-src.conf�w)rC r� r4 �hashesr rq �joinZtempdir� dest_path� hash_path�old_path� source_path�open�write�appendr rv )r rq �frG r r r4 + 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 ro )r rv r� r5 r r r �_current_file_hash> s z3InstallVersionControlledFileTest._current_file_hashc C s0 | � tj�| j�� | �t�| j�| �� � d S ro ) rp r rq rr r� r r rv r� r5 r r r �_assert_current_fileA s �z5InstallVersionControlledFileTest._assert_current_filec C s( | � tj�| j�� | �� | �� d S ro )r� r rq rr r� r� r� r5 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 ro )�shutilZcopyfiler� r� r� r� r5 r r r �!test_prev_file_updates_to_currentQ s zBInstallVersionControlledFileTest.test_prev_file_updates_to_currentc C s� | � � t| jd��}|�d� W 5 Q R X t�d�� }| � � | �|jj� W 5 Q R X | � t
j � | j�� | �
t�| j�| �� � | �t�| j�| �� � d S )N�a� a new line for the wrong hash
� certbot.plugins.common.logger)r� r� r� r� r- rl r� �warning�calledrp r rq rr r r rv r� r� r� )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 5 Q R X t | jd��}|�d� W 5 Q R X t�d��&}| �� | �|jj d d d� W 5 Q R X | �t
� | j �| �
� � t�d�� }| �� | �|jj� W 5 Q R X d S )Nr� r� r� Zhashofanoldversionr� r zh%s has been manually modified; updated file saved to %s. We recommend updating %s for security purposes.)r� r� r� r� r- rl r� r r� Z call_argsr rv r� r� r� 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 r4 r� r� r� r� r� r� r� r� rz r r rG r r� ( s
r� �__main__)(r rV r� Zunittestre ZjosepyZjoser- Zacmer Zcertbotr r r Zcertbot.compatr r Z
certbot.testsr r Z test_utilZJWKRSA�loadZ load_vectorZAUTH_KEYZ"KeyAuthorizationAnnotatedChallengeZchall_to_challbZHTTP01r� ZTestCaser r ZConfigTestCaserA r{ r� ZTempDirTestCaser� r �mainr r r r �<module> s< � �,oQL