JFIF $
d d 7
Viewing File: /usr/lib/python3.9/site-packages/certbot/_internal/tests/__pycache__/storage_test.cpython-39.pyc
a
}|�gm� � @ sB d Z ddlZddlZddlZddlZddlZddlmZ ddlZddlZddl Z ddl
Z
ddl
m Z ddl
m Z ddl
mZ ddlmZ ddlmZ ddlm mZ e�d �Zd
d � Zd d
� ZG dd� dej�ZG dd� dej �Z G dd� de �Z G dd� de �Z G dd� de �Z e!dk�r>e�"e�#ej$dd� e%g �� dS )�$Tests for certbot._internal.storage.� N)�mock)�
configuration)�errors)�ALL_FOUR)�
filesystem)�os� 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 �H/usr/lib/python3.9/site-packages/certbot/_internal/tests/storage_test.py�
unlink_all s r c C sH t D ]>}tt| |�d��}|�|� W d � q1 s80 Y 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 @ sR e Zd ZdZdd� Zdd� Ze�d�dd� �Zd d
� Z d d � Z
d
d� Z dd� Z dS )�RelevantValuesTestz4Tests for certbot._internal.storage.relevant_values.c C s& ddd�| _ t�� | _t�� | j_d S )N� example.orgZrsa)�server�key_type)�valuesr Z MagicMock� mock_config� set_by_user��selfr r r �setUp) s
zRelevantValuesTest.setUpc C s ddl m} || jj_|| j�S )Nr ��relevant_values)�certbot._internal.storager! r �to_dict� return_value)r r r! r r r �_call. s
zRelevantValuesTest._callz8certbot._internal.plugins.disco.PluginsRegistry.find_allc C s: dg|_ d| jj_ d| jd< | �| j�� �| jks6J �d S )Nzcertbot-foo:barT�* zcertbot_foo:bar_baz)r$ r r r r% �copy)r Z
mock_find_allr r r �test_namespace3 s
z!RelevantValuesTest.test_namespacec C sT d| j j_d| jd< d| jd< d| jd< | j�� }d| jd< | �| j�|ksPJ �d S ) NT�allow_subset_of_names�apache�
authenticatori9 � rsa_key_sizeZthereZhello�r r r$ r r' r% �r �expected_relevant_valuesr r r �test_option_set; s
z"RelevantValuesTest.test_option_setc C s6 d| j j_| j�� }d| jd< | �| j�|ks2J �d S )NFi r, r- r. r r r �test_option_unsetF s
z$RelevantValuesTest.test_option_unsetc s| d� � fdd�| j _| j�� }d | j� < | �| j�|ks<J �d| j� < | �| j�|ksZJ �d| j� < | �| j�|ksxJ �d S )NZmanual_public_ip_logging_okc s | � kr dS dS )NFTr )�v�Zdeprected_optionr r �<lambda>P � z9RelevantValuesTest.test_deprecated_item.<locals>.<lambda>TF)r r r r' r% r. r r3 r �test_deprecated_itemN s
z'RelevantValuesTest.test_deprecated_itemc C sr ddl m} ddlm} ddlm} ddlm} |j�� }|� |g d��}|j
d dd d
d �}||�|ksnJ �d S ) Nr r )�disco)�cli)� constants)z--allow-subset-of-namesz--authenticatorr* r ZecdsaTr* )r r r) r+ ) r" r! Zcertbot._internal.pluginsr7 �certbot._internalr8 r9 ZPluginsRegistryZfind_allZprepare_and_parse_argsZ CLI_DEFAULTS)r r! r7 r8 r9 ZPLUGINS� namespacer/ r r r �test_with_real_parserZ s
�z(RelevantValuesTest.test_with_real_parserN)
�__name__�
__module__� __qualname__�__doc__r r% r �patchr( r0 r1 r6 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} t� �� t�tj�| j j
dd�� tj�| j j
dd�}t�|� t�tj�| j j
d�� t � � }t
D ]&}tj�| j j
dd|d �}|||< qrttj�| j j
ddd�d �� W d � n1 s�0 Y ||d< tj�| j j
dd
�|_|�� || _ttj�| j j
dd �d ��}|�d
� W d � n1 �s>0 Y t � � | _t�d��(}d|_|�|j| j �| _W d � n1 �s�0 Y d S )Nr ��storage�liver �archiveZrenewal�.pem�README�a�example.org.confz IGNORE.THISr z This file should be ignored!z7certbot._internal.storage.RenewableCert._check_symlinksT)r: rD �superr r �makedirsr �path�join�config�
config_dir� configobj� ConfigObjr r �filenamer � config_file�defaultsr rA r$ �
RenewableCert�test_rc)r rD Z archive_pathrT r
� kind_pathZjunk�check�� __class__r r r w s<
�
�� � *
zBaseRenewableCertTest.setUpNc C s� t | j|�}tj�|�r"t�|� t�tj�tjjtjjddd� ||��|� t
|d��,}|� |d u rp|� d�n|� W d � n1 s�0 Y |dkr�t
�|d� d S )NrF r z
{0}{1}.pem�wb�ascii�privkey� )r rW r rM �lexistsr
�symlinkrN �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 ri �r r
r r r �_write_out_ex_kinds� s z)BaseRenewableCertTest._write_out_ex_kinds)N)r= r> r? r@ r ri rm �
__classcell__r r rZ r rB o s &
rB 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&� Zej�ejd'�e�d(�d)d*� ��Zd+d,� Z ej�ejd'�e�d-�d.d/� ��Z e�d0�d1d2� �Z e � 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&e�dB�dCdD� �Z'dEdF� Z(dGdH� Z)dIdJ� Z*dKdL� Z+dMdN� Z,dOdP� Z-dQS )R�RenewableCertTestsr c C sH | j jdksJ �tD ].}t| j |�tj�| jjdd|d �ksJ �qd S )Nr rE rG ) rW � lineagenamer r r rM rN rO rP rl r r r �test_initialization� s
�z&RenewableCertTests.test_initializationc C s� ddl m} tj�| jjd�}t|d��}|�d� W d � n1 sH0 Y t �
t j �� |�
|| j� W d � n1 s�0 Y t�|� t �
t j �� |�
d| j� W d � n1 s�0 Y dS )z�Test that the RenewableCert constructor will complain if
the renewal configuration file doesn't end in ".conf"
r rC z broken.confr z [No closing bracket for you!NZfun)r: rD r rM rN rO rP r r �pytest�raisesr �CertStorageErrorrV r
)r rD �brokenr r r r �test_renewal_bad_config� s (,
z*RenewableCertTests.test_renewal_bad_configc C s� ddl m} t�� }d|d< d|d< d|d< tj�| jjd �|_ |�
� t � t
j�� |�|j | j� W d
� n1 sz0 Y d
S ) z�Test that the RenewableCert constructor will complain if
the renewal configuration file is missing a required file element.r rC zimaginary_cert.pem�certzimaginary_chain.pem�chainzimaginary_fullchain.pem� fullchainzimaginary_config.confN)r: rD rQ rR r rM rN rO rP rS r rr rs r rt rV )r rD rO r r r � test_renewal_incomplete_config� s z1RenewableCertTests.test_renewal_incomplete_configc C sr ddl m} | �� d| jvs"J �t�d��"}|�| jj| j� W d � n1 sT0 Y |j j
du snJ �d S )Nr rC �version� certbot._internal.storage.loggerF) r: rD rm rT r rA rV rS rO �warning�called�r rD � mock_loggerr r r �test_no_renewal_version� s 0z*RenewableCertTests.test_no_renewal_versionc C s� ddl m} | �� d| jd< | j�� t�d��"}|�| jj| j � W d � n1 sZ0 Y |j
j spJ �d|j
j d d v s�J �d S )Nr rC z99.99.99r{ r| )
r: rD rm rT r r rA rV rS rO �infor~ � call_argsr r r r �test_renewal_newer_version� s
0 z-RenewableCertTests.test_renewal_newer_versionc C s� | j j}d| j _| j �� r J �|| j _| j �� r4J �t| j � | j �� rLJ �t| j � tD ]&}t�tj� d|d �t
| j |�� qZ| j �� r�J �t| j � tD ]*}t�tj� | j j |d �t
| j |�� q�| j �� r�J �t| j � tD ],}t�tj� dddd|d �t
| j |�� q�| j �� �r$J �t| j � | j �� �s>J �t�
| j j� t�tj� ddddd�| j j� t| j jd��}|�d� W d � n1 �s�0 Y | j �� �r�J �d S ) Nz
relative/path�..z17.pemrF r zfullchain_17.pemr zwrongly-named fullchain)rW rw � _consistentr r r r ra rM rN r rO rP r
ry r r )r Zoldcertr
r r r r �test_consistent� sL
�
�
�
�
��*z"RenewableCertTests.test_consistentc C s� | � dd� tj�| j�d�tj�| jjddd��s8J �t� | jj
� t� tj�| jjddd�| jj
� t | jj
d��}|�
d� W d � n1 s�0 Y tj�| j�d�tj�| jjddd��s�J �d S )Nrw � rF r z
cert17.pemr )ri r rM �samefilerW �current_targetrN rO rP r
rw ra 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�dksXJ �d S )N)� � �
� rw r� rF r z
cert10.pemr� ) ri r r
rW rw ra rM rN �current_version)r rf r r r �test_current_version' s ��z'RenewableCertTests.test_current_versionc C s | j �d�d u sJ �d S )Nrw )rW r� r r r r �test_no_current_version/ s z*RenewableCertTests.test_no_current_versionc C s� t dd�D ]}tD ]}| �||� qq
| j�� dks8J �| j�� dksJJ �| �dd� | j�� dkshJ �| j�� dkszJ �| �dd� | �dd� | j�� dks�J �tD ]}| �|d � q�| j�� d ks�J �| j�� d
ks�J �d S ) Nr� � r� r^ � � rw ry r� � )�ranger ri rW �latest_common_versionZnext_free_version�r rf r
r r r � test_latest_and_next_versions2 s z0RenewableCertTests.test_latest_and_next_versionsr| c C s� t �� }| j_t �� }| j_t �� | j_d|_| j�� du sDJ �|jdksRJ �|j jdksbJ �d|_| j�� du szJ �|jdks�J �|j jdks�J �d S )NFTr r� )
r ZMockrW �update_all_links_to�has_pending_deploymentr� r$ Zensure_deployed�
call_countr} )r r� Z mock_updateZmock_has_pendingr r r �test_ensure_deployedI s z'RenewableCertTests.test_ensure_deployedc C s� t dd�D ].}tD ]$}| �||� || j�|�ksJ �qq
| j�dd� | j�dd� d| j�d�ksjJ �d| j�d�ks~J �d| j�d�ks�J �d| j�d �ks�J �| j�dd
� tj�t �
| jj ��d ks�J �d S ) Nr� r� rw � r^ � r� rx ry i� z
chain3000.pem) r� r ri rW r� �_update_link_tor rM �basenamer �readlinkrx r� r r r �test_update_link_toZ s �z&RenewableCertTests.test_update_link_toc C s. | � dd� dtj�| j�dd��ks*J �d S )Nrw rj z cert8.pemr� )ri r rM r� rW r{ r r r r � test_versionm s �z RenewableCertTests.test_versionc C s� t dd�D ].}tD ]$}| �||� || j�|�ksJ �qq
| j�� dksLJ �t dd�D ]@}| j�|� tD ]}|| j�|�ksjJ �qj| j�� dksVJ �qVd S �Nr� r� r� )r� r ri rW r� r� r� r� r r r � test_update_all_links_to_successu s z3RenewableCertTests.test_update_all_links_to_successc C s� t jfdd�}| �� t�d��L}||_t�t�� | j �
d� W d � n1 sT0 Y W d � n1 sr0 Y t D ]}| j � |�dks�J �q�d S )Nc S s. t j�| �}d|v r"|�d�r"t�|| � d S )Nry �prev)r rM r� �
startswith�
ValueError)rM � real_unlinkr� r r r �unlink_or_raise� s zTRenewableCertTests.test_update_all_links_to_partial_failure.<locals>.unlink_or_raise�#certbot._internal.storage.os.unlinkrj �
r r
rm r rA � side_effectrr rs r� rW r� r r� �r r� � mock_unlinkr
r r r �(test_update_all_links_to_partial_failure� s Hz;RenewableCertTests.test_update_all_links_to_partial_failurec C s� t jfdd�}| �� t�d��L}||_t�t�� | j �
d� W d � n1 sT0 Y W d � n1 sr0 Y t D ]}| j � |�dks�J �q�d S )Nc S s dt j�| �v rt�|| � d S )Nry )r rM r� r� )rM r� r r r r� � s zQRenewableCertTests.test_update_all_links_to_full_failure.<locals>.unlink_or_raiser� rj rk r� r� r r r �%test_update_all_links_to_full_failure� s Hz8RenewableCertTests.test_update_all_links_to_full_failurec C s� t dd�D ].}tD ]$}| �||� || j�|�ksJ �qq
t dd�D ]T}| j�|� tD ]}|| j�|�ksXJ �qX|dk r�| j�� s�J �qD| j�� rDJ �qDd S r� )r� r ri rW r� r� r� r� r r r �test_has_pending_deployment� s z.RenewableCertTests.test_has_pending_deploymentc C sr | � ddt�d�� | j�� ddgks*J �t�| jj� t� t
j �� | j�� W d � n1 sd0 Y d S )Nrw rj zcert-san_512.pem� example.comzwww.example.com) ri � test_util� load_vectorrW �namesr r
rw rr rs r rt r r r r �
test_names� s �z RenewableCertTests.test_namesr z"certbot._internal.storage.datetimec C s t �d�}| �� | j�d� t| jjd��}|�|� W d � n1 sL0 Y | j�d� t| jjd��}|�|� W d � n1 s�0 Y tj |_ d|_
i | jj d< dD ]B\}}}tj� |t
j�}||jj_
|| jj d < | j�� |ks�J �q�dS )
z<Test should_autorenew() on the basis of expiry time windows.r rj r\ Nrk 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 monthsTZrenew_before_expiry)r� r� rm rW r� r rw r �datetime� timedeltar$ r Z
fromtimestamp�pytz�UTCZnow�should_autorenew) r Z
mock_datetime�mock_set_by_userZ test_certr Z current_time�interval�resultZsometimer r r � test_time_interval_judgments� s
( (
z/RenewableCertTests.test_time_interval_judgmentsc C sZ i | j jd< | j �� sJ �d| j jd d< | j �� s8J �d| j jd d< | j �� rVJ �d S )Nr� �True� autorenew�False)rW r Zautorenewal_is_enabledr r r r �test_autorenewal_is_enabled� s z.RenewableCertTests.test_autorenewal_is_enabledz4certbot._internal.storage.RenewableCert.ocsp_revokedc C sh d|_ ddi| jjd< | j�� r$J �d| jjd d< tD ]}| �|d� q8d|_ | j�� s^J �d|_ dS ) zTTest should_autorenew on the basis of reasons other than
expiry time window.Fr� r� r� r� rj TN)r$ rW r r� r ri )r Z mock_ocspr� r
r r r �test_should_autorenew� s z(RenewableCertTests.test_should_autorenewz)certbot._internal.storage.relevant_valuesc
C s� dd� |_ tdd�D ]}tD ]}| �||� q q| j�d� d| j�ddd d| j�ksZJ �t| j� dd��� }|�
� d ks~J �W d � n1 s�0 Y t| j� d
d��� }|�
� d ks�J �W d � n1 s�0 Y t| j� d d���"}|�
� d
k�sJ �W d � n1 �s0 Y t j �
| j� dd���r@J �t j �
| j� dd���s\J �d| j�ddd d| j�k�s|J �d| j�ddd d| j�k�s�J �dD ]H}t j �
| j� d|���s�J �dt j �t�| j� d|���k�s�J ��q�tD ]<}| j�|�ttdd��k�sJ �| j�|�dk�s�J ��q�| j�d� d| j�ddd d| j�k�sXJ �tD ]<}| j�|�ttdd��k�s�J �| j�|�dk�s\J ��q\t| j� d d���"}|�
� dk�s�J �W d � n1 �s�0 Y t j �| jj| jj�d }t|d ��}|�d � W d � n1 �s&0 Y d| j�dd d d!| j�k�sPJ �t j �| j� dd���slJ �t j �
| j� dd���r�J �t j �|��r�J �d S )"Nc S s | � � S �N�r# ��xr r r r4 r5 z8RenewableCertTests.test_save_successor.<locals>.<lambda>r� r� r� s new cert� new chainrw znew certrx z new chainry znew certnew chainr^ 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 ri rW r� �save_successorrO r r{ �readr rM �islinkr� r r� �available_versions�listr� rN �renewal_configs_dirrp r �exists)r �mock_rvrf r
r �iZtemp_config_filer r r �test_save_successor sl
�
..2 � � � �
�
2 �� *� 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�sFJ �t�| j�dd�d� | j� ddd d| j
� t�| j�dd �d�s�J �| j� d dd
d| j
� t�| j�dd �d �s�J �t�| j�dd �d
� | j� d dd
d| j
� t�| j�dd�d��sJ �d S )Nc S s | � � S r� r� r� r r r r4 B r5 zMRenewableCertTests.test_save_successor_maintains_group_mode.<locals>.<lambda>r� r^ r_ i$ � newcertr� r� � new_privkeyr� i� � � ) r� r ri rW r� r �
check_moder{ re r� rO )r r� r
r r r �(test_save_successor_maintains_group_mode= s
z;RenewableCertTests.test_save_successor_maintains_group_modezBcertbot._internal.storage.filesystem.copy_ownership_and_apply_modec C st dd� |_ tD ]}| �|d� q| j�d� | j�ddd d| j� |jdu sPJ �| j�dddd| j� |jspJ �d S ) Nc S s | � � S r� r� r� r r r r4 X r5 zFRenewableCertTests.test_save_successor_maintains_gid.<locals>.<lambda>r� r� r� Fr� r� )r� r ri rW r� r� rO r~ )r Zmock_ownershipr� r
r r r �!test_save_successor_maintains_gidS s
z4RenewableCertTests.test_save_successor_maintains_gidc C s� dd� |_ ddlm} |j�dddd| j�}|�� s8J �tj� tj�
| jj d ��sVJ �tj� tj�
| jj d
��stJ �tj� tj�
| jj dd
��s�J �t
�|jd �s�J �t|jd �� }|�� d
ks�J �W d � n1 s�0 Y |j�dddd| j�}tj� tj�
| jj d���sJ �tj� tj�
| jj dd
���s:J �t
�tj�
| jjd�� |j�dddd| j�}tj� tj�
| jj dd
���s�J �t
�|jd ��s�J �t�tj�
| jj d�tj�
| jj d�� t�tj��& |j�dddd| j� W d � n1 �s 0 Y t�tj�
| jj d�tj�
| jj d �� t�tj��& |j�d d d d | j� W d � n1 �sl0 Y |j�dddd| j�}dS ) z$Test for new_lineage() class method.c S s | � � S r� r� r� r r r r4 f r5 z5RenewableCertTests.test_new_lineage.<locals>.<lambda>r rC �the-lineage.coms certs privkeys chain�the-lineage.com.confrH r_ �rbs certchainN� cert2� privkey2� chain2zthe-lineage.com-0001.confzthe-lineage.com-0001zthe-lineage.com-0002s cert3s privkey3s chain3zthe-lineage.com-0003s cert4s privkey4s chain4zother-example.coms cert5s privkey5s chain5)r� r: rD rV � new_lineagerO r� r rM r� rN r� �live_dirr r� Zkey_pathr ry r� �mkdir�default_archive_dir�shutil�copytreerr rs r rt )r r� rD r� r r r r �test_new_lineagea sh
� �
�
�
. � �
� �
� �
�$�
�$ �z#RenewableCertTests.test_new_lineagec C s� dd� |_ ddlm} t�| jj� t�| jj� t�| jj� |j �
dddd| j� t j �
t j �| jjd ��stJ �t j �
t j �| jjdd
��s�J �t j �
t j �| jjdd ��s�J �d S )
z9Test that directories can be created if they don't exist.c S s | � � S r� r� r� r r r r4 � r5 zFRenewableCertTests.test_new_lineage_nonexistent_dirs.<locals>.<lambda>r rC r� r� r� r� r� z privkey.pemz privkey1.pemN)r� r: rD r� �rmtreerO r� r� r� rV r� r rM r� rN )r r� rD r r r �!test_new_lineage_nonexistent_dirs� s&
���
�
�z4RenewableCertTests.test_new_lineage_nonexistent_dirsz2certbot._internal.storage.util.unique_lineage_namec C sX ddl m} d|_t�tj��& |j�dddd| j � W d � n1 sJ0 Y d S )Nr rC )Z this_does_not_end_with_dot_confZyikesr� rw r^ rx )
r: rD r$ rr rs r rt rV r� rO )r Zmock_ulnrD r r r � test_invalid_config_filename� s
�z/RenewableCertTests.test_invalid_config_filenamec C s\ t �tj�� | j�d� W d � n1 s.0 Y t �tj�� | j�d� W d � n1 sf0 Y t �tj�� | j�dd� W d � n1 s�0 Y t �tj�� | j�d� W d � n1 s�0 Y t �tj�� | j� d� W d � n1 �s0 Y t �tj�� | j�
dd� W d � n1 �sN0 Y d S )NZelephantr� ) rr rs r rt rW r� r� r{ r� Znewest_available_versionr� r r r r �
test_bad_kind� s **,*,z RenewableCertTests.test_bad_kindz4certbot.ocsp.RevocationChecker.ocsp_revoked_by_pathsc C sP t D ]}| �|d� q| j�� }| j�d|�}| j�d|�}d|_| j�|�sRJ �|jd d |kshJ �|jd d |ks~J �d|_| j�|�r�J �|jd d |ks�J �|jd d |ks�J �t|_ t
� d�� }| j�|�r�J �W d � n1 s�0 Y |jd d |k�sJ �|jd d |k�s0J �|jd d }d|v �sLJ �d S ) Nr� rw rx Tr Fz(certbot._internal.storage.logger.warningz-An error occurred determining the OCSP status) r ri rW r� r{ r$ Z ocsp_revokedr� r� r� r rA )r Z mock_checkerr
r{ Zexpected_cert_pathZexpected_chain_path�loggerZlog_msgr r r �test_ocsp_revoked� s(
.z$RenewableCertTests.test_ocsp_revokedc
) C s� ddl m} tjddddddtjd�}tjd d
d dd d
tjd�}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 � |d f|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$� |d#f|t�d$� i}|�� D ](\}}|\}} |� || �|k�s�J ��q�d S )%Nr rC i� rk r� �; � )Ztzinfoi� r� r� � � z
Asia/Shanghaii� � � � �) � z17 daysr� Z30� z7 weeks�1 z1 monthz1 yearin im z 1 year 1 dayio z 1 year-1 dayil z4 yearsi� )
r: rD r� r� r� �timezoneZfromutcr� �itemsZadd_time_interval)
r rD Ztime_1Ztime_2Ztime_3Zintended�
parametersZexceptedZ base_timer� r r r �test_add_time_interval� s>
��
�z)RenewableCertTests.test_add_time_intervalc C sD i | j jd< | j jd u s J �| j jd }d|d< | j jdks@J �d S )Nr� zhttps://acme.example/dirr )rW r r �r Zrpr r r � test_server s
z RenewableCertTests.test_serverc C s� i | j jd< | j jd }| j jdu s(J �d|d< | j jdu s@J �d|d< | j jdu sXJ �d|d< | j jdu spJ �d|d< | j jdu s�J �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)rW r Z is_test_certr r r r �test_is_test_cert s z$RenewableCertTests.test_is_test_certc C s� ddl m} t�tj��" |�| jj| j � W d � n1 s@0 Y t
� d| jt d � t�tj��" |�| jj| j � W d � n1 s�0 Y d S )Nr rC �missing)
r: rD rr rs r rt rV rT rS rO r ra r �r rD r r r �test_missing_cert s 0z$RenewableCertTests.test_missing_certc
C s> t j�| jjd�}t j�| jjd�}t|d��}|�d� W d � n1 sN0 Y t�|d� i }t D ] }d||< qld}dd i}d
d l
m } |� |||||� t|d ��}|�
� } W d � n1 s�0 Y d
| v s�J �d| v s�J �d| vs�J �d�tj�| v �sJ �t�t �|�j�t�t �|�j�k�s: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 rC �rzuseful = new_valuezA useful valueZuselessz
version = {0})r rM rN rO rP r r r re r r: rD Zwrite_renewal_configr� rc �certbot� __version__�stat�S_IMODE�lstat�st_mode)
r ZtempZtemp2r �targetr� � archive_dirZ
relevant_datarD Zcontentr r r �test_write_renewal_config( s* (
& �z,RenewableCertTests.test_write_renewal_configc C s� t D ]}| �|d� qt�d��"}| j�� |�� W d � n1 sH0 Y t D ] }tdd�D ]}| �||� qdqVt�d��J}| j�� |jdt t � ks�J �d|j
d d d v s�J �W d � n1 s�0 Y d S )Nr� zcertbot.compat.os.unlinkr� r� z1.pemr ) r ri r rA rW �truncateZassert_not_calledr� r� �lenZcall_args_list)r r
r� r� r r r �
test_truncateI s
&
z RenewableCertTests.test_truncateN).r= r> r? r@ rq rv rz r� r� r� r� r� r� r� r rA r� r� r� r� r� r� r� r� �objectr ZNamespaceConfigr� r� r� r� r� Zskip_on_windowsr� r� r� r� r� r� r� r r r r r r r r r r ro � sd
*
+
9
5
*
!ro c s` e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z � Z
S )�DeleteFilesTestz0Tests for certbot._internal.storage.delete_filesc s� t � �� tD ]H}tj�| jjdd|d �}t|d�� W d � q1 sL0 Y q| j �
� tj� tj�| jj d��s�J �tj� tj�| jj
d��s�J �tj� tj�| jjdd��s�J �d S )NrE r rG rI rJ rF )rK r r r rM rN rO rP r rT r r� r� r� )r r
rX rZ r r r \ s"
�
�
�
�zDeleteFilesTest.setUpc C sH ddl m} t�d�� |�| jd� W d � n1 s:0 Y d S )Nr rC r| r )r: rD r rA Z delete_filesrO r r r r r% l s zDeleteFilesTest._callc C sh | � � tj�tj�| jjd��r&J �tj�tj�| jjd��rDJ �tj�tj�| jjdd��rdJ �d S )NrJ r rF ) r% r rM r� rN rO r� r� rP r r r r �test_delete_all_filesq s �
�
�z%DeleteFilesTest.test_delete_all_filesc C s� t | jjd��}|�d� W d � n1 s.0 Y t�tj�� | �� W d � n1 sb0 Y t j
� t j
� | j
jd��s�J �t j
� t j
� | j
jd��r�J �d S )NrI Zasdfasfasdfasdfr rJ )r rT rS r rr rs r rt r% r rM r� rN rO r� r� )r rT r r r �test_bad_renewal_config{ s (& �
�z'DeleteFilesTest.test_bad_renewal_configc C sx t �| jj� t�tj�� | �� W d � n1 s80 Y t j �
t j � | j j
d��s`J �t j �
| jj�rtJ �d S �Nr )r �removerT rS rr rs r rt r% rM r� rN rO r� r r r r �test_no_renewal_config� s & �
z&DeleteFilesTest.test_no_renewal_configc C sx t �t j�| jjdd�� | �� t j�| jj �r6J �t j�t j�| jjd��rTJ �t j�t j�| jj
dd��rtJ �d S )Nr zcert.pemrF � r r rM rN rO r� r% r� rT rS rP r r r r �test_no_cert_file� s
� �
�z!DeleteFilesTest.test_no_cert_filec C sx t �t j�| jjdd�� | �� t j�| jj �r6J �t j�t j�| jjd��rTJ �t j�t j�| jj
dd��rtJ �d S )Nr rH rF r! r r r r �test_no_readme_file� s
� �
�z#DeleteFilesTest.test_no_readme_filec C s� t tj�| jjdd�d�� W d � n1 s00 Y | �� tj�| jj �rVJ �tj�tj�| jjd��stJ �tj�tj�| jj
dd��r�J �d S )Nr Z
other_filerI rF ) r r rM rN rO r� r% r� rT rS rP r r r r �test_livedir_not_empty� s
�� �
�z&DeleteFilesTest.test_livedir_not_emptyc C sl t j�| jjdd�}t �|� | �� t j�| jj �r:J �t j�t j�| jj
d��rXJ �t j�|�rhJ �d S )NrF r ) r rM rN rO rP �rmdirr% r� rT rS 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& rn r r rZ r r Z s
r c s8 e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Z� ZS )
�CertPathForCertNameTestz:Test for certbot._internal.storage.cert_path_for_cert_namec sB t � �� | j�� | �� tj�| jj ddd�| _
| j
| j_ d S )NrE r z
fullchain.pem) rK r rT r rm r rM rN rO rP ry Z cert_pathr rZ 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�| jksJ �d S r )r% rO ry r r r r �test_simple_cert_name� s z-CertPathForCertNameTest.test_simple_cert_namec C s> t �tj�� | �| jd� W d � n1 s00 Y d S )Nzfake-example.org)rr rs r rt r% rO 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* rn r r rZ r r' � s
r' �__main__r� )&r@ r� r� r �sysZunittestr rQ rr r� r r r r" r Zcertbot.compatr r Zcertbot.tests.utilZtests�utilr� Z load_certZCERTr r ZTestCaser ZConfigTestCaserB ro r r' r= �exit�main�argv�__file__r r r r �<module> s>
I@ 0\
Back to Directory
nL+D550H?Mx ,D"v]qv;6*Zqn)ZP 0 !1 A "#a$2Qr D8 a Ri[f\mIykIw0 cuFcRı?lO7к_f˓[C$殷WF<_W ԣs KcëIzyQy /_LKℂ;C",pFA: /]=H ~,ls/9ć:[=/#f;)x {ٛEQ )~ =𘙲r*2~ a _V=' kumF D}KYYC)({ *g&f`툪ry`=^cJ.I](*`wq 1dđ#̩͑0;H]u搂@:~וKL Nsh}OIR*8:2 !lDJVo( 3=M(zȰ+i*NA r6KnSl )!JJӁ* %݉?|D}d5:eP0R;{$X'xF@.ÊB {,WJuQɲRI;9QE琯62fT.DUJ;*cP A\ILNj!J۱+O\ ͔]ޒ SJȧc%ANolՎprULZԛerE2=XDXgVQeӓk yP7U*omQIs,K`)6\G3t?pgjrmۛجwluGtf h9uyP0D;Uڽ"OXlif$)&|ML0Zrm1[HXPlPR0'G=i2N +0e2]] 9VTPO7h(F*癈 '=QVZDF,d߬~TX
G[`le69CR(!S2!P
<0x < !1AQ "Raq02Br#SCTb
?Ζ"]mH5WR7k.ۛ!}Q~+yԏz|@T20S~Kek *zFf^2X*(@8r?CIuI|֓>^ExLgNUY+{.RѪ
τVYTD I62'8Y2 7'\TP.6d&˦@Vqi|8- OΕ]ʔ U=TL8=;6c| !qfF3aů&~$l}'NWUs$Uk^SV:U# 6w++s&r+nڐ{@29 gL
u"TÙM=6 (^"7r} =6YݾlCuhquympǦ
G jhsǜNlɻ}o7 #S6aw4!OSrD57%|?x>L
|/nD6?/8w#[)L7+6〼T ATg ! %5MmZ/c-{1_Je"|^$'O&ޱմ Trb$ w)R$&
N1EtdU3Uȉ1pM" N*(DN yd96.(jQ)X
5cQɎMyW? Q*!R>6=7)Xj5`J]e 8%t!+'!1Q 5 !1 AQaqё#2"0BRb ? Gt^## .llQT $v,,m 㵜5 ubV =sY+@d{N! dnO< .-B; _wJt6;QJd.Qc%p{ 1,sNDdFHI0ГoXшe黅XۢF:)[FGXƹ/w_cMeD,ʡcc. WDtA$j@:) -# u
c1<@ۗ9F)KJ-hpP]_x[qBlbp ʖw q"LFGd ƶ*s+ډ_Zc"? %t[IP 6J]#=ɺVvvCGsGh1 >)6|ey?Lӣm,4GWUi` ]uJVoV DG< SB 6ϏQ@ TiUly OU0kfV~~}SZ@*WUUi##; s/[ =!7}" WN]'(L! ~y5g9T̅JkbM' +s:S + B)v@Mj
e Cf jE 0Y\QnzG1д~Wo{T 9?`Rmyhsy3!HAD]mc1~2L Su7xT;j$`}4->L#vzŏ ILS ֭ T { rjGKC; bpU= -`BsK.SFw4Mq]ZdHS0)tLg