|
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
�]�� � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m m Z
ddl mZ ddlmZ ddlmZ ddlmZ e
�d�Zdd � Zd
d � ZG d d
� d
ej�ZG dd� de
j�ZG dd� de�Z G dd� de�Z G dd� de�Z e dk�re� � dS )�Tests for certbot.storage.� N)�errors)�os)�
filesystem)�ALL_FOUR� cert_512.pemc C s t D ]}t�t| |�� qdS )z9Unlink all four items associated with this RenewableCert.N)r r �unlink�getattr)� rc_object�kind� r �</usr/lib/python3/dist-packages/certbot/tests/storage_test.py�
unlink_all s r c
C s4 t D ]*}tt| |�d��}|�|� W 5 Q R X qdS )z9Put dummy data into all four files of this RenewableCert.�wN)r �openr �write)r
r �fr r r
�fill_with_sample_data s r c @ s` e Zd ZdZdd� Zdd� Ze�d�e�d�dd � ��Ze�d�d
d � �Z e�d�d d
� �Z
dS )�RelevantValuesTestz*Tests for certbot.storage.relevant_values.c C s ddi| _ d S )N�server� example.org)�values��selfr r r
�setUp' s zRelevantValuesTest.setUpc O s ddl m} |||�S )Nr )�relevant_values)�certbot.storager )r �args�kwargsr r r r
�_call* s zRelevantValuesTest._callzcertbot.cli.option_was_setz.certbot.plugins.disco.PluginsRegistry.find_allc C s6 dg|_ d|_ d| jd< | �| �| j�� �| j� d S )Nzcertbot-foo:barT�* zcertbot_foo:bar_baz)� return_valuer � assertEqualr �copy)r Z
mock_find_all�mock_option_was_setr r r
�test_namespace. s
�z!RelevantValuesTest.test_namespacec C sP d|_ d| jd< d| jd< d| jd< | j�� }d| jd< | �| �| j�|� d S ) NTZallow_subset_of_namesZapacheZ
authenticatori9 � rsa_key_sizeZthereZhello�r! r r# r" r �r r$ Zexpected_relevant_valuesr r r
�test_option_set8 s
z"RelevantValuesTest.test_option_setc C s2 d|_ | j�� }d| jd< | �| �| j�|� d S )NFi r& r' r( r r r
�test_option_unsetD s
z$RelevantValuesTest.test_option_unsetN) �__name__�
__module__� __qualname__�__doc__r r �mock�patchr% r) r* r r r r
r $ s
r c s2 e Zd ZdZ� fdd�Zd dd�Zdd� Z� ZS )
�BaseRenewableCertTestz�Base class for setting up Renewable Cert tests.
.. note:: It may be required to write out self.config for
your test. Check :class:`.cli_test.DuplicateCertTest` for an example.
c s^ ddl m} tt| ��� t�tj� | j
j dd�� tj� | j
j dd�}t�|� t�tj� | j
j d�� t �
� }tD ]&}tj� | j
j dd|d �}|||< qvttj� | j
j ddd�d �� W 5 Q R X ||d< tj� | j
j dd
�|_|�� || _ttj� | j
j dd �d ��}|�d
� W 5 Q R X t �
� | _t�d�� }d|_|�|j| j
�| _W 5 Q R X d S )Nr ��storage�liver �archiveZrenewal�.pem�README�a�example.org.confz IGNORE.THISr z This file should be ignored!z-certbot.storage.RenewableCert._check_symlinksT)�certbotr3 �superr1 r r �makedirsr �path�join�config�
config_dir� configobj� ConfigObjr r �filenamer � config_file�defaultsr/ r0 r! �
RenewableCert�test_rc)r r3 � archive_pathrD r � kind_pathZjunkZcheck�� __class__r r
r V s<
�
��
�
zBaseRenewableCertTest.setUpNc C s� t | j|�}tj�|�r"t�|� t�tj�tjjtjjddd� ||��|� t
|d��"}|� |d krp|� d�n|� W 5 Q R X |dkr�t
�|d� d S )Nr5 r z
{0}{1}.pem�wb�ascii�privkey� )r rG r r= �lexistsr �symlinkr> �pardir�formatr r �encoder �chmod)r r �ver�value�linkr r r r
�_write_out_kind| s
�� &z%BaseRenewableCertTest._write_out_kindc C s& t D ] }| �|d� | �|d� qd S )N� � )r rY �r r r r r
�_write_out_ex_kinds� s z)BaseRenewableCertTest._write_out_ex_kinds)N)r+ r, r- r. r rY r] �
__classcell__r r rJ r
r1 N s &
r1 c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
e�d�dd� �Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Ze�d'�e�d(�d)d*� ��Zd+d,� Ze�d'�e�d-�d.d/� ��Ze�d0�d1d2� �Ze � d3�e�d0�d4d5� ��Z e�d0�e�d6�d7d8� ��Z e�d0�d9d:� �Z e�d0�d;d<� �Z!e�d=�d>d?� �Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&dHdI� Z'dJdK� Z(dLdM� Z)dNS )O�RenewableCertTestsr c C sH | � | jjd� tD ].}| � t| j|�tj�| jj dd|d �� qd S )Nr r4 r6 )
r" rG � lineagenamer r r r= r> r? r@ r\ r r r
�test_initialization� s
��z&RenewableCertTests.test_initializationc C sx ddl m} tj�| jjd�}t|d��}|�d� W 5 Q R X | � t
j |j || j� t�
|� | � t
j |j d| j� dS )z�Test that the RenewableCert constructor will complain if
the renewal configuration file doesn't end in ".conf"
r r2 z broken.confr z [No closing bracket for you!ZfunN)r: r3 r r= r> r? r@ r r � assertRaisesr �CertStorageErrorrF r )r r3 Zbrokenr r r r
�test_renewal_bad_config� s �
�z*RenewableCertTests.test_renewal_bad_configc C sd ddl m} t�� }d|d< d|d< d|d< tj�| jjd �|_ |�
� | � t j
|j|j | j� d
S ) z�Test that the RenewableCert constructor will complain if
the renewal configuration file is missing a required file element.r r2 zimaginary_cert.pem�certzimaginary_chain.pem�chainzimaginary_fullchain.pem� fullchainzimaginary_config.confN)r: r3 rA rB r r= r> r? r@ rC r rb r rc rF )r r3 r? r r r
� test_renewal_incomplete_config� s �z1RenewableCertTests.test_renewal_incomplete_configc C s^ ddl m} | �� | �d| jk� t�d��}|�| jj| j � W 5 Q R X | �
|j j � d S )Nr r2 �version�certbot.storage.logger)
r: r3 r] �
assertTruerD r/ r0 rF rC r? � assertFalse�warning�called�r r3 � mock_loggerr r r
�test_no_renewal_version� s z*RenewableCertTests.test_no_renewal_versionc C s| ddl m} | �� d| jd< | j�� t�d��}|�| jj| j � W 5 Q R X | �
|j j � | �
d|j j
d d k� d S )Nr r2 z99.99.99ri rj )r: r3 r] rD r r/ r0 rF rC r? rk �inforn Z call_argsro r r r
�test_renewal_newer_version� s
z-RenewableCertTests.test_renewal_newer_versionc C s� | j j}d| j _| �| j �� � || j _| �| j �� � t| j � | �| j �� � t| j � tD ]&}t�tj �
d|d �t | j |�� q`| �| j �� � t| j � tD ]*}t�tj �
| j j
|d �t | j |�� q�| �| j �� � t| j � tD ],}t�tj �
dddd|d �t | j |�� q�| �| j �� � t| j � | �| j �� � t�| j j� t�tj �
ddddd�| j j� t| j jd��}|�d� W 5 Q R X | �| j �� � d S ) Nz
relative/path�..z17.pemr5 r zfullchain_17.pemr zwrongly-named fullchain)rG re rl � _consistentr r r r rQ r= r> r r? r@ rk r rg r r )r Zoldcertr r r r r
�test_consistent� sL
�
�
�
�
��z"RenewableCertTests.test_consistentc C s� | � dd� | �tj�| j�d�tj�| jj ddd��� t�
| jj � t� tj�| jj ddd�| jj � t
| jj d��}|�d� W 5 Q R X | �tj�| j�d�tj�| jj ddd��� d S )Nre � r5 r z
cert17.pemr )rY rk r r= �samefilerG �current_targetr> r? r@ r re rQ r r )r r r r r
�test_current_target� s* ������z&RenewableCertTests.test_current_targetc C s\ dD ]}| � d|� qt�| jj� t�tj�ddddd�| jj� | �| j� d�d� d S )N)� � �
� re rt r5 r z
cert10.pemr} )
rY r r rG re rQ r= r> r" �current_version)r rV r r r
�test_current_version s ��z'RenewableCertTests.test_current_versionc C s | � | j�d�d � d S )Nre )r" rG r r r r r
�test_no_current_version s z*RenewableCertTests.test_no_current_versionc C s� t j�dd�D ]}tD ]}| �||� qq| �| j�� d� | �| j�� d� | �dd� | �| j�� d� | �| j�� d� | �dd� | �dd� | �| j�� d� tD ]}| �|d � q�| �| j�� d � | �| j�� d
� d S ) Nr{ � r| rN � � re rg rw � ) �six�moves�ranger rY r" rG �latest_common_versionZnext_free_version�r rV r r r r
� test_latest_and_next_versions s z0RenewableCertTests.test_latest_and_next_versionsrj c C s� t �� }| j_t �� }| j_t �� | j_d|_| �| j�� d� | �|j d� | �|j
j d� d|_| �| j�� d� | �|j d� | �|j
j d� d S )NFTr r{ ) r/ ZMockrG �update_all_links_to�has_pending_deploymentr� r! r" Zensure_deployedZ
call_countrm )r rp Z mock_updateZmock_has_pendingr r r
�test_ensure_deployed* s z'RenewableCertTests.test_ensure_deployedc C s� t j�dd�D ].}tD ]$}| �||� | �|| j�|�� qq| j�dd� | j�dd� | �d| j�d�� | �d| j�d�� | �d| j�d�� | �d| j�d �� | j�dd
� | �t j
� t � | jj
��d � d S ) Nr{ r� re � rN � r| rf rg i� z
chain3000.pem)r� r� r� r rY r" rG r �_update_link_tor r= �basename�readlinkrf r� r r r
�test_update_link_to; s �z&RenewableCertTests.test_update_link_toc C s. | � dd� | �dtj�| j�dd��� d S )Nre rZ z cert8.pemr� )rY r" r r= r� rG ri r r r r
� test_versionN s �z RenewableCertTests.test_versionc C s� t j�dd�D ].}tD ]$}| �||� | �|| j�|�� qq| �| j�� d� t j�dd�D ]@}| j� |� tD ]}| �|| j�|�� qr| �| j�� d� q^d S �Nr{ r� r| )
r� r� r� r rY r" rG r r� r� r� r r r
� test_update_all_links_to_successV s z3RenewableCertTests.test_update_all_links_to_successc C sf t jfdd�}| �� t�d�� }||_| �t| jj d� W 5 Q R X t
D ]}| � | j� |�d� qHd S )Nc S s0 t j�| �}d|kr$|�d�r$t�n|| � d S )Nrg �prev)r r= r� �
startswith�
ValueError)r= � real_unlinkr� r r r
�unlink_or_raisec s zTRenewableCertTests.test_update_all_links_to_partial_failure.<locals>.unlink_or_raise�certbot.storage.os.unlinkrZ �
r r r] r/ r0 � side_effectrb r� rG r� r r" r �r r� Z mock_unlinkr r r r
�(test_update_all_links_to_partial_failureb s z;RenewableCertTests.test_update_all_links_to_partial_failurec C sf t jfdd�}| �� t�d�� }||_| �t| jj d� W 5 Q R X t
D ]}| � | j� |�d� qHd S )Nc S s" dt j�| �krt�n|| � d S )Nrg )r r= r� r� )r= r� r r r
r� t s zQRenewableCertTests.test_update_all_links_to_full_failure.<locals>.unlink_or_raiser� rZ r[ r� r� r r r
�%test_update_all_links_to_full_failures s z8RenewableCertTests.test_update_all_links_to_full_failurec C s� t j�dd�D ].}tD ]$}| �||� | �|| j�|�� qqt j�dd�D ]X}| j�|� tD ]}| �|| j�|�� q`|dk r�| � | j�
� � qL| � | j�
� � qLd S r� ) r� r� r� r rY r" rG r r� rk r� rl r� r r r
�test_has_pending_deployment� s z.RenewableCertTests.test_has_pending_deploymentc C s� | � ddt�d�� | �| j�� ddg� | � ddt�d�� | �| j�d�ddg� | � ddt�d�� | �| j�d�dgd d
� d D � � t�| jj� | � t
j | jj� d S ) Nre rZ zcert-san_512.pem� example.comzwww.example.com� r zcert-5sans_512.pemc S s g | ]}d � |��qS )z{0}.example.com)rS )�.0�cr r r
�
<listcomp>� s z1RenewableCertTests.test_names.<locals>.<listcomp>Zabcd) rY � test_util� load_vectorr" rG �namesr r re rb r rc r r r r
�
test_names� s& �� �
�z RenewableCertTests.test_nameszcertbot.storage.clizcertbot.storage.datetimec C s� t �d�}| �� | j�d� t| jjd��}|�|� W 5 Q R X | j�d� t| jjd��}|�|� W 5 Q R X tj |_ d|j
_ i | jj d< dD ]J\}}}tj�
|�}||jj_ || jj d< || jj d < | �| j�� |� q�d
S ) z<Test should_autorenew() on the basis of expiry time windows.r rZ rL r[ F�
renewalparams))�@*�Tz2 monthsT)r� z1 weekT)r� z4 daysF)r� �2 daysF)�@�Iz7 yearsT)r� z11 years 2 monthsT)r� z8 hoursF)r� r� F)r� z40 daysF)r� z9 monthsF)� ��Tz 0 secondsT)r� z
10 secondsT)r� z
10 minutesT)r� z10 weeksT)r� z 10 monthsT)r� z10 yearsT)r� z 99 monthsTZdeploy_before_expiryZrenew_before_expiryN)r� r� r] rG r� r re r �datetime� timedelta�
set_by_clir! �
configurationZutcfromtimestampZutcnowr" �should_autorenew) r Z
mock_datetime�mock_cliZ test_certr Z current_time�interval�resultZsometimer r r
� test_time_interval_judgments� s"
z/RenewableCertTests.test_time_interval_judgmentsc C s` i | j jd< | �| j �� � d| j jd d< | �| j �� � d| j jd d< | �| j �� � d S )Nr� �True� autorenew�False)rG r� rk Zautorenewal_is_enabledrl r r r r
�test_autorenewal_is_enabled� s z.RenewableCertTests.test_autorenewal_is_enabledz*certbot.storage.RenewableCert.ocsp_revokedc C sn d|j _ddi| jjd< | �| j�� � d| jjd d< tD ]}| �|d� q<d|_| �| j�� � d|_dS ) zTTest should_autorenew on the basis of reasons other than
expiry time window.Fr� r� r� r� rZ TN) r� r! rG r� rl r� r rY rk )r Z mock_ocspr� r r r r
�test_should_autorenew� s z(RenewableCertTests.test_should_autorenewz certbot.storage.relevant_valuesc C s* dd� |_ tj�dd�D ]}tD ]}| �||� q q| j�d� | �d| j� ddd d| j
�� t | j� dd���}| �|�
� d � W 5 Q R X t | j� d
d���}| �|�
� d � W 5 Q R X t | j� d d���}| �|�
� d
� W 5 Q R X | �tj�| j� dd��� | �tj�| j� dd��� | �d| j� ddd d| j
�� | �d| j� ddd d| j
�� dD ]F}| �tj�| j� d|��� | �dtj�t�| j� d|���� �q`tD ]<}| �| j�|�ttj�dd��� | �| j�|�d� �q�| j�d� | �d| j� ddd d| j
�� tD ]<}| �| j�|�ttj�dd��� | �| j�|�d� �qt | j� d d���}| �|�
� d� W 5 Q R X tj�| j
j| jj�d }t |d ��}|�d � W 5 Q R X | �d| j� dd d d!| j
�� | �tj� | j� dd��� | �tj�| j� dd��� | �tj� |�� d S )"Nc S s | S �Nr ��xr r r
�<lambda>� � z8RenewableCertTests.test_save_successor.<locals>.<lambda>r{ r� r� s new cert� new chainre znew certrf z new chainrg znew certnew chainrN r� s agains newer chainr� s hellos other chain)r� r� r� z privkey3.pem� s lasts attemptr} Z lastattemptz .conf.newr z)We previously crashed while writing me :(s with� as key) r� r� r� r� r rY rG r� r" �save_successorr? r ri �readrl r r= �islinkrk r� r� �available_versions�listr r> �renewal_configs_dirr` r �exists)r �mock_rvrV r r �iZtemp_config_filer r r
�test_save_successor� s�
�� �� �� � " ��� �� �� z&RenewableCertTests.test_save_successorz:Group/everybody permissions are not maintained on Windows.c C s dd� |_ tD ]}| �|d� q| j�d� | �t�| j�dd�d�� t� | j�dd�d� | j�
ddd d| j � | �t�| j�dd �d�� | j�
d dd
d| j � | �t�| j�dd �d �� t� | j�dd �d
� | j�
d dd
d| j � | �t�| j�dd�d�� d S )Nc S s | S r� r r� r r r
r� 7 r� zMRenewableCertTests.test_save_successor_maintains_group_mode.<locals>.<lambda>r{ rN rO i$ � newcertr� r� � new_privkeyr� i� � � ) r� r rY rG r� rk r �
check_moderi rU r� r? )r r� r r r r
�(test_save_successor_maintains_group_mode2 s
z;RenewableCertTests.test_save_successor_maintains_group_modez8certbot.storage.filesystem.copy_ownership_and_apply_modec C st dd� |_ tD ]}| �|d� q| j�d� | j�ddd d| j� | �|j� | j�dddd| j� | � |j� d S )Nc S s | S r� r r� r r r
r� M r� zFRenewableCertTests.test_save_successor_maintains_gid.<locals>.<lambda>r{ r� r� r� r� )
r� r rY rG r� r� r? rl rn rk )r Zmock_ownershipr� r r r r
�!test_save_successor_maintains_gidH s
z4RenewableCertTests.test_save_successor_maintains_gidc C s� dd� |_ ddlm} |j�dddd| j�}| �|�� � | �tj �
tj � | jj d ��� | �tj �
tj � | jj
d
��� | �tj �
tj � | jj
dd
��� | �t�|jd �� t|jd ��}| �|�� d
� W 5 Q R X |j�dddd| j�}| �tj �
tj � | jj d��� | �tj �
tj � | jj
dd
��� t�tj � | jj
d�� | �tj|jjdddd| j� t�tj � | jjd�� | �tj|jjdddd| j� |j�dddd| j�}dS ) z$Test for new_lineage() class method.c S s | S r� r r� r r r
r� [ r� z5RenewableCertTests.test_new_lineage.<locals>.<lambda>r r2 �the-lineage.coms certs privkeys chain�the-lineage.com.confr7 rO �rbs certchain� cert2� privkey2� chain2zthe-lineage.com-0001.confzthe-lineage.com-0001zthe-lineage.com-0002s cert3s privkey3s chain3zother-example.coms cert4s privkey4s chain4N)r� r: r3 rF � new_lineager? rk ru r r= r� r> r� �live_dirr r� Zkey_pathr rg r" r� �mkdirrb r rc �default_archive_dir)r r� r3 r� r r r r
�test_new_lineageV s�
� � � � � � �
� � � �z#RenewableCertTests.test_new_lineagec C s� dd� |_ ddlm} t�| jj� t�| jj� t�| jj� |j �
dddd| j� | � t j
�t j
�| jjd ��� | � t j
�t j
�| jjdd
��� | � t j
�t j
�| jjdd ��� d S )
z9Test that directories can be created if they don't exist.c S s | S r� r r� r r r
r� � r� zFRenewableCertTests.test_new_lineage_nonexistent_dirs.<locals>.<lambda>r r2 r� r� r� r� r� z privkey.pemz privkey1.pemN)r� r: r3 �shutilZrmtreer? r� r� r� rF r� rk r r= r� r> )r r� r3 r r r
�!test_new_lineage_nonexistent_dirs� s8
�
�� � �z4RenewableCertTests.test_new_lineage_nonexistent_dirsz(certbot.storage.util.unique_lineage_namec C s4 ddl m} d|_| �tj|jjdddd| j� d S )Nr r2 )Z this_does_not_end_with_dot_confZyikesr� re rN rf ) r: r3 r! rb r rc rF r� r? )r Zmock_ulnr3 r r r
� test_invalid_config_filename� s �z/RenewableCertTests.test_invalid_config_filenamec C s� | � tj| jjd� | � tj| jjd� | � tj| jjdd� | � tj| jjd� | � tj| jjd� | � tj| jj dd� d S )NZelephantrw )
rb r rc rG ry r ri r� Znewest_available_versionr� r r r r
�
test_bad_kind� s@ � � � � � �z RenewableCertTests.test_bad_kindc C s | � | j�� � d S r� )rl rG Z ocsp_revokedr r r r
�test_ocsp_revoked� s z$RenewableCertTests.test_ocsp_revokedc
) C s� ddl m} tj�t�dddddd��}tj�t�dd d
dd d ��}t�d
��t�dd dddd��}|df||df||df||df|t�d� |df|t�d� |df|t�d� |df|t�d� |df|t�d� |df|t�d� |df|t�d� |df|t�d � |df|t�d � |df|t�d � |d f|t�d � |d f|t�d � |d f|t�d � |d f|t�d!� |d"f|t�d#� |d"f|t�d#� i}|�� D ]&\}}|\}} | � |�
|| �|� �q�d S )$Nr r2 i� r[ r~ �; � i� r} r� � � z
Asia/Shanghaii� � � � �) � z17 daysrw Z30� z7 weeks�1 z1 monthz1 yearin im z 1 year 1 dayio z 1 year-1 dayil z4 yearsi� ) r: r3 �pytzZUTCZfromutcr� �timezoner� �itemsr" Zadd_time_interval)
r r3 Ztime_1Ztime_2Ztime_3ZintendedZ
parametersZexceptedZ base_timer� r r r
�test_add_time_interval� sd
� ��z)RenewableCertTests.test_add_time_intervalc C s� i | j jd< | j jd }| �| j jd� d|d< | �| j jd� d|d< | �| j jd� d|d< | �| j jd� d|d< | �| j jd� d S ) Nr� Fz6https://acme-staging-v02.api.letsencrypt.org/directoryr Tz)https://staging.someotherca.com/directoryz.https://acme-v01.api.letsencrypt.org/directoryz.https://acme-v02.api.letsencrypt.org/directory)rG r� r" Z is_test_cert)r Zrpr r r
�test_is_test_cert� s z$RenewableCertTests.test_is_test_certc C sZ ddl m} | �tj|j| jj| j� t �
d| jt d � | �tj|j| jj| j� d S )Nr r2 Zmissing) r: r3 rb r rc rF rD rC r? r rQ r �r r3 r r r
�test_missing_cert� s � �z$RenewableCertTests.test_missing_certc
C s t j�| jjd�}t j�| jjd�}t|d��}|�d� W 5 Q R X t�|d� i }t D ] }d||< qXd}dd i}d
d l
m } |� |||||� t|d ��}|�
� } W 5 Q R X | �d
| k� | �d| k� | �d| k� | �d�t
j�| k� | �t�t �|�j�t�t �|�j�� d S )Nz sample-filezsample-file.newr zM[renewalparams]
useful = value # A useful value
useless = value # Not needed
i� Z somewhereZ the_archiveZusefulZ new_valuer r2 �rzuseful = new_valuezA useful valueZuselessz
version = {0})r r= r> r? r@ r r r rU r r: r3 Zwrite_renewal_configr� rk rS � __version__r" �stat�S_IMODE�lstat�st_mode)
r ZtempZtemp2r �targetr� � archive_dirZ
relevant_datar3 Zcontentr r r
�test_write_renewal_config� s*
�z,RenewableCertTests.test_write_renewal_configc C s� ddl m} tj�| jjdd�}tD ]L}| j| }|d }tj�||�}t |d��
� t� tj�| jj|�|� q$| � t
j|j| jj| j� |j| jj| jdd� d S ) Nr r2 r5 r z1.pemr8 T)Zupdate_symlinks)r: r3 r r= r> r? r@ r rD r �closerQ rb r rc rF rC )r r3 Zarchive_dir_pathr Z live_pathr� rH r r r
�test_update_symlinks s
��z'RenewableCertTests.test_update_symlinksN)*r+ r, r- r. ra rd rh rq rs rv rz r� r� r� r/ r0 r� r� r� r� r� r� r� r� r� r� r� r� r� Zskip_on_windowsr� r� r� r� r� r� r� r� 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 � Z
S )�DeleteFilesTestz&Tests for certbot.storage.delete_filesc
s� t t| ��� tD ]4}tj�| jjdd|d �}t |d�� W 5 Q R X q| j
� � | � tj�
tj�| jjd��� | � tj�
tj�| jjd��� | � tj�
tj�| jjdd��� d S )Nr4 r r6 r8 r9 r5 )r; r r r r r= r> r? r@ r rD r rk r� r� r� )r r rI rJ r r
r - s* �
� � �zDeleteFilesTest.setUpc C s4 ddl m} t�d�� |�| jd� W 5 Q R X d S )Nr r2 rj r )r: r3 r/ r0 Z delete_filesr? r� r r r
r = s zDeleteFilesTest._callc C sn | � � | �tj�tj�| jjd��� | �tj�tj�| jjd��� | �tj�tj�| jj dd��� d S )Nr9 r r5 )
r rl r r= r� r> r? r� r� r@ r r r r
�test_delete_all_filesB s � � �z%DeleteFilesTest.test_delete_all_filesc C sx t | jjd��}|�d� W 5 Q R X | �tj| j� | �t j
� t j
� | j
jd��� | �t j
� t j
� | j
jd��� d S )Nr8 Zasdfasfasdfasdfr r9 )r rD rC r rb r rc r rk r r= r� r> r? r� rl r� )r rD r r r
�test_bad_renewal_configL s � �z'DeleteFilesTest.test_bad_renewal_configc C sX t �| jj� | �tj| j� | �t j �
t j � | j j
d��� | �t j �
| jj�� d S )Nr )r �removerD rC rb r rc r rk r= r� r> r? r� rl r r r r
�test_no_renewal_configV s �z&DeleteFilesTest.test_no_renewal_configc C s~ t �t j�| jjdd�� | �� | �t j�| j j
�� | �t j�t j�| jjd��� | �t j�t j�| jj dd��� d S )Nr zcert.pemr5 � r r r= r> r? r� r rl r� rD rC r@ r r r r
�test_no_cert_file] s
� � �z!DeleteFilesTest.test_no_cert_filec C s~ t �t j�| jjdd�� | �� | �t j�| j j
�� | �t j�t j�| jjd��� | �t j�t j�| jj dd��� d S )Nr r7 r5 r r r r r
�test_no_readme_fileg s
� � �z#DeleteFilesTest.test_no_readme_filec C s� t tj�| jjdd�d�� W 5 Q R X | �� | �tj�| j j
�� | � tj�tj�| jjd��� | �tj�tj�| jj dd��� d S )Nr Z
other_filer8 r5 )
r r r= r> r? r� r rl r� rD rC rk r@ r r r r
�test_livedir_not_emptyq s& ��
� �z&DeleteFilesTest.test_livedir_not_emptyc C sr t j�| jjdd�}t �|� | �� | �t j�| j j
�� | �t j�t j�| jj d��� | �t j�|�� d S )Nr5 r ) r r= r> r? r@ �rmdirr rl r� rD rC r� )r r r r r
�test_no_archive| s
�z DeleteFilesTest.test_no_archive)r+ r, r- r. r r r r r
r r
r r r^ r r rJ r
r + s
r c s8 e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Z� ZS )
�CertPathForCertNameTestz0Test for certbot.storage.cert_path_for_cert_namec sJ t t| ��� | j�� | �� tj�| j j
ddd�| _ | j df| j _ d S )Nr4 r z
fullchain.pemr� )
r; r r rD r r] r r= r> r? r@ rg Z cert_pathr rJ r r
r � s
�z CertPathForCertNameTest.setUpc C s ddl m} |||�S )Nr )�cert_path_for_cert_name)r r )r Z
cli_configZcertnamer r r r
r � s z CertPathForCertNameTest._callc C s | � | �| jd�| jdf� d S )Nr rg )r" r r? rg r r r r
�test_simple_cert_name� s z-CertPathForCertNameTest.test_simple_cert_namec C s | � tj| j| jd� d S )Nzfake-example.org)rb r rc r r? r r r r
�test_no_such_cert_name� s z.CertPathForCertNameTest.test_no_such_cert_name) r+ r, r- r. r r r r r^ r r rJ r
r � s
r �__main__)!r. r� r� r� ZunittestrA r/ r� r� r: Zcertbot.tests.utilZtests�utilr� r Zcertbot.compatr r r r Z load_certZCERTr r ZTestCaser ZConfigTestCaser1 r_ r r r+ �mainr r r r
�<module> s:
*@ "Z