← Back
Editing: storage_test.cpython-38.pyc
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
Save File
Cancel