🔐 Sid Gifari File Manager Pro
v8.0.5 | 2026-06-14 15:44:52 | PHP 8.1.34
📂
/ (Root)
/
usr
/
lib
/
python3.9
/
site-packages
/
certbot
/
_internal
/
tests
/
__pycache__
📍 /usr/lib/python3.9/site-packages/certbot/_internal/tests/__pycache__
🔄 Refresh
✏️
Editing: cert_manager_test.cpython-39.pyc
Read Only
a }|�gs � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlZddlZddl m Z ddl m Z ddl m Z ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ G d d � d ej�ZG d d� dej�ZG dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z G dd� dej�Z!G dd � d ej�Z"G d d � d ej�Z#G d d � d ej$�Z%e&d!k�r�e�'e�(ej)d"d� e*g �� dS )#�)Tests for certbot._internal.cert_manager.� N)�mock)� configuration)�errors)�ALL_FOUR)� storage_test)� filesystem)�os)�utilc s( e Zd ZdZ� fdd�Zdd� Z� ZS )�BaseCertManagerTestz2Base class for setting up Cert Manager tests. c s� t � �� d� j_t�� jj� d tj� � jj d�d�� _ � fdd�� j D �� _ t tj� � jjd�d��}|�d� W d � n1 s�0 Y d S ) NFZspecialarchive)� example.org� other.comc s i | ]}|� � |� j| ��qS � )�_set_up_config�domains)�.0�domain��selfr �M/usr/lib/python3.9/site-packages/certbot/_internal/tests/cert_manager_test.py� <dictcomp>% s �z-BaseCertManagerTest.setUp.<locals>.<dictcomp>z IGNORE.THIS�wz This file should be ignored!)�super�setUp�config�quietr �makedirs�renewal_configs_dirr �path�join� config_dirr Z config_files�open�write)r Zjunk�� __class__r r r s � �zBaseCertManagerTest.setUpc C s� t �tj�| jj|�� t�� }|d ur<t �|� ||d< nt �tj�| jj |�� t D ] }tj�| jj||d �||< qXtj�| jj |d �|_ |� � |S )N� archive_dirz.pemz.conf)r r r r r r Zlive_dir� configobjZ ConfigObjZdefault_archive_dirr r �filenamer" )r r Zcustom_archive� config_file�kindr r r r . s � �z"BaseCertManagerTest._set_up_config)�__name__� __module__� __qualname__�__doc__r r � __classcell__r r r# r r s r c @ s� e Zd ZdZdd� Ze�� e�d�e�d�e�d�dd� ����Z e�� e�d�e�d�d d � ���Z e�� e�d�e�d�d d � ���Z e�� e�d�e�d�d d� ���Z e�� e�d�e�d�dd� ���Z e�� e�d�e�d�dd� ���ZdS )� DeleteTestz4Tests for certbot._internal.cert_manager.delete c C s ddl m} |�| j� d S �Nr �� cert_manager)�certbot._internalr2 �deleter )r r2 r r r �_callH s zDeleteTest._callzcertbot.display.util.notify�3certbot._internal.cert_manager.lineage_for_certnamez&certbot._internal.storage.delete_filesc C s> | j |_d|� j_d| j_| �� |�| jd� |�d� dS )� Test deleteTr z6Deleted all files relating to certificate example.org.N)�test_rc� return_value�yesnor �certnamer5 �assert_called_once_with)r �mock_delete_files�mock_lineage_for_certnameZ mock_notify� mock_utilr r r �test_delete_from_config_yesL s �z&DeleteTest.test_delete_from_config_yesc C s4 | j |_d|� j_d| j_| �� |jdks0J �dS )r7 Fr r N)r8 r9 r: r r; r5 � call_count�r r= r> r? r r r �test_delete_from_config_no\ s z%DeleteTest.test_delete_from_config_noc C s> | j |_tjdgf|� j_d|� j_| �� |�| jd� dS )r7 r TN) r8 r9 � display_util�OK� checklistr: r5 r< r rB r r r �"test_delete_interactive_single_yesh s z-DeleteTest.test_delete_interactive_single_yesc C s> | j |_tjdgf|� j_d|� j_| �� |jdks:J �dS )r7 r Fr N�r8 r9 rD rE rF r: r5 rA rB r r r �!test_delete_interactive_single_not s z,DeleteTest.test_delete_interactive_single_noc C s\ | j |_tjddgf|� j_d|� j_| �� |�| jd� |�| jd� |j dksXJ �dS )r7 r � other.orgT� N) r8 r9 rD rE rF r: r5 Zassert_any_callr rA rB r r r �$test_delete_interactive_multiple_yes� s z/DeleteTest.test_delete_interactive_multiple_yesc C s@ | j |_tjddgf|� j_d|� j_| �� |jdks<J �dS )r7 r rJ Fr NrH rB r r r �#test_delete_interactive_multiple_no� s z.DeleteTest.test_delete_interactive_multiple_noN)r* r+ r, r- r5 � test_util�patch_display_utilr �patchr@ rC rG rI rL rM r r r r r/ D s6 r/ c @ s� e Zd ZdZdd� Ze�d�e�� dd� ��Z e�d�e�� dd� ��Z e�d �e�d�e�� e�d �e�d �d d � �����Z e�d�e�� dd� ��Z e�d�e�d�dd� ��Z dS )�CertificatesTestz:Tests for certbot._internal.cert_manager.certificates c O s ddl m} ||i |��S )Nr )� certificates)� certbot._internal.cert_managerrR )r �args�kwargsrR r r r � _certificates� s z CertificatesTest._certificatesz%certbot._internal.cert_manager.loggerc C s& | � | j� |jjsJ �|js"J �d S �N)rV r �warning�called�r � mock_utility� mock_loggerr r r � test_certificates_parse_fail� s z-CertificatesTest.test_certificates_parse_failc C s4 d| j _| �| j � |jjdu s$J �|jjs0J �d S )NTF)r r rV Z notificationrY rX rZ r r r �test_certificates_quiet� s z(CertificatesTest.test_certificates_quietz)certbot.crypto_util.verify_renewable_cert�'certbot._internal.storage.RenewableCertz5certbot._internal.cert_manager._report_human_readablec C sJ d |_ d|_ | �| j� |jjdu s(J �|js2J �|js<J �|jsFJ �d S )N� F)r9 rV r rX rY )r Z mock_report�mock_renewable_certr[ r\ Z mock_verifierr r r � test_certificates_parse_success� s z0CertificatesTest.test_certificates_parse_successc C s| t �� }t�tjtj�|d�tj�|d�tj�|d�dd��}t � |j � | � |� |j jdu sdJ �|jsnJ �t�|� d S )Nr ZworkZlogsF)r Zwork_dirZlogs_dirr )�tempfileZmkdtempr ZNamespaceConfigr � MagicMockr r r r r r rV rX rY �shutil�rmtree)r r[ r\ Z empty_tempdirZ empty_configr r r �test_certificates_no_files� s � z+CertificatesTest.test_certificates_no_filesz(certbot.crypto_util.get_serial_from_certzBcertbot._internal.cert_manager.ocsp.RevocationChecker.ocsp_revokedc s� d |_ d|_ dd l}dd l}ddlm� |j�|j�}tjdd�}||_ ddg|j _ d|_ |g�tjd d d��� ��fd d �}|� }d |v s�J �| j |j d d �7 _ |� }d|v p�d|v du s�J �d|v s�J �d|vs�J �| j |j dd�7 _ |� }d|v �sJ �d|v�sJ �d|v �s"J �d|v�s0J �| j |j d d�7 _ |� }d|v �sZJ �d|v �shJ �d|v�svJ �d|_ d|_ |� }d|v �s�J �tjdd�}||_ ddg|j _ d|_ �� |� |� }tt�d|��d k�s�J �dg�_|� }tt�d|��dk�s J �dg�_|� }tt�d|��d k�s4J �d�_|� }tt�d|��dk�sZJ �d �_|� }tt�d|��dk�s�J �d S ) Ni��Ir r1 Znameone�� lineagenameZnametwoF)r; ri c s � � ���S rW )Z_report_human_readabler �r2 Z mock_configZ parsed_certsr r �<lambda>� � z=CertificatesTest.test_report_human_readable.<locals>.<lambda>zINVALID: EXPIREDrK )Zhoursz1 hourz 2 hour(s)TZVALID�INVALID� )Zdaysz1 dayZunderz3 dayszINVALID: TEST_CERT, REVOKEDZ indescribablez thrice.namedzINVALID:Zhorror)r9 �datetime�pytzr3 r2 ZnowZUTCr rd Z target_expiry�namesZ is_test_certZ timedelta�append�len�re�findallr r; ) r Z mock_revokedZ mock_serialro rp Zexpiry�certZ get_report�outr rj r �test_report_human_readable� sl z+CertificatesTest.test_report_human_readableN)r* r+ r, r- rV r rP rN rO r] r^ rb rg rx r r r r rQ � s( rQ c @ s6 e Zd ZdZe�d�e�d�e�d�dd� ���ZdS )�SearchLineagesTestz:Tests for certbot._internal.cert_manager._search_lineages.� certbot.util.make_or_verify_dir�,certbot._internal.storage.renewal_conf_filesr_ c C sF dg|_ tj|_ddlm} |�| jdd� d�dks8J �|jsBJ �d S )NZbadfiler r1 c S s | S rW r ��xr r r rk * rl z<SearchLineagesTest.test_cert_storage_error.<locals>.<lambda>�check) r9 r �CertStorageError� side_effectr3 r2 Z_search_lineagesr rY )r ra �mock_renewal_conf_files�mock_make_or_verify_dirr2 r r r �test_cert_storage_error s z*SearchLineagesTest.test_cert_storage_errorN)r* r+ r, r- r rP r� r r r r ry s ry c @ sn e Zd ZdZe�d�e�d�e�d�dd� ���Ze�d�e�d�dd� ��Ze�d�e�d�d d � ��Zd S ) �LineageForCertnameTestz=Tests for certbot._internal.cert_manager.lineage_for_certnamerz �3certbot._internal.storage.renewal_file_for_certnamer_ c C sH d|_ tjdd�}||_ ddlm} |�| jd�|ks:J �|jsDJ �d S )N� somefile.conf� example.comrh r r1 )r9 r �Mockr3 r2 �lineage_for_certnamer rY )r ra �mock_renewal_conf_filer� � mock_matchr2 r r r �test_found_match1 s z'LineageForCertnameTest.test_found_matchc C s6 d|_ ddlm} |�| jd�d u s(J �|js2J �d S )Nzother.com.confr r1 r� )r9 r3 r2 r� r rY �r r� r� r2 r r r � test_no_match= s z$LineageForCertnameTest.test_no_matchc C s: t �� |_ddlm} |�| jd�d u s,J �|js6J �d S )Nr r1 r� )r r r� r3 r2 r� r rY r� r r r �test_no_renewal_fileE s z+LineageForCertnameTest.test_no_renewal_fileN) r* r+ r, r- r rP r� r� r� r r r r r� . s r� c @ sR e Zd ZdZe�d�e�d�e�d�dd� ���Ze�d�e�d�dd� ��Zd S ) �DomainsForCertnameTestz=Tests for certbot._internal.cert_manager.domains_for_certnamerz r� r_ c C sX d|_ tjdd�}ddg}||j_ ||_ ddlm} |�| jd�|ksJJ �|jsTJ �d S )Nr� r� rh r r r1 ) r9 r r� rq r3 r2 �domains_for_certnamer rY )r ra r� r� r� r r2 r r r r� Q s �z'DomainsForCertnameTest.test_found_matchc C s6 d|_ ddlm} |�| jd�d u s(J �|js2J �d S )Nr� r r1 r )r9 r3 r2 r� r rY r� r r r r� ` s z$DomainsForCertnameTest.test_no_matchN)r* r+ r, r- r rP r� r� r r r r r� N s r� c s� e Zd ZdZ� fdd�Zdd� Ze�d�e� � dd� ��Z e� � d d � �Z e� � e�d �d d � ��Z e� � e�d�dd� ��Z e� � e�d�dd� ��Ze� � e�d�dd� ��Z� ZS )�RenameLineageTestz7Tests for certbot._internal.cert_manager.rename_lineagec s t � �� d| j_d| j_d S )Nr Zafter)r r r r; � new_certnamer r# r r r l s zRenameLineageTest.setUpc O s ddl m} |j|i |��S r0 )r3 r2 Zrename_lineage)r rT rU r2 r r r r5 q s zRenameLineageTest._callr{ c C s� d | j _d| j _g |_t�tj�� | �| j � W d � n1 sD0 Y dg|_|� }t j df|j _t�tj�� | �| j � W d � n1 s�0 Y t j df|j _t�tj�� | �| j � W d � n1 s�0 Y d S )N�twozone.confr ���) r r; r� r9 �pytest�raisesr �Errorr5 rD �CANCEL�menurE )r �mock_get_utilityr� � util_mockr r r �test_no_certnameu s **z"RenameLineageTest.test_no_certnamec C s� d| j _d | j _|� }tjdf|j_t�t j �� | � | j � W d � n1 sR0 Y tj d f|j_t�t j �� | � | j � W d � n1 s�0 Y d S )N�one�name) r r; r� rD r� �inputr9 r� r� r r� r5 rE )r r� r� r r r �test_no_new_certname� s *z&RenameLineageTest.test_no_new_certnamer6 c C sR d| j _d| j _d |_t�tj�� | �| j � W d � n1 sD0 Y d S )Nr� r� ) r r; r� r9 r� r� r �ConfigurationErrorr5 )r r> �unused_get_utilityr r r �test_no_existing_certname� s z+RenameLineageTest.test_no_existing_certnamez7certbot._internal.storage.RenewableCert._check_symlinksc C sR d|_ | �| j� ddlm} |�| j| jj�}|d us<J �|j| jjksNJ �d S �NTr r1 )r9 r5 r r3 r2 r� r� ri )r � mock_checkr� r2 �updated_lineager r r �test_rename_cert� s z"RenameLineageTest.test_rename_certc C sn d|_ d | j_|� }tjdf|j_ | �| j� ddlm} |� | j| jj �}|d usXJ �|j | jj ksjJ �d S r� ) r9 r r; rD rE r� r5 r3 r2 r� r� ri )r r� r� r� r2 r� r r r �%test_rename_cert_interactive_certname� s z7RenameLineageTest.test_rename_cert_interactive_certnamec C s� d|_ d| j_t�tj�� | �| j� W d � n1 s<0 Y d�t j j �| j_t�tj�� | �| j� W d � n1 s�0 Y d S )NTr z one{0}two) r9 r r� r� r� r r� r5 �formatr r �sep)r r� r� r r r �!test_rename_cert_bad_new_certname� s *z3RenameLineageTest.test_rename_cert_bad_new_certname)r* r+ r, r- r r5 r rP rN rO r� r� r� r� r� r� r. r r r# r r� i s( r� c s2 e Zd ZdZ� fdd�Ze�d�dd� �Z� ZS )�DuplicativeCertsTestz!Test to avoid duplicate lineages.c s t � �� | j�� | �� d S rW )r r r( r" �_write_out_ex_kindsr r# r r r � s zDuplicativeCertsTest.setUprz c C s� ddl m} t�d�}t| jjd��}|�|� W d � n1 sD0 Y || jddg�}|dksjJ �|| jdd g�}|d j j � d �s�J �|d d u s�J �|| jg d ��}|d d u s�J �|d j j � d �s�J �|| jdd g�}|dks�J �d S )Nr )�find_duplicative_certszcert-san_512.pem�wbzwow.netz hooray.org)NNr� �www.example.comzexample.org.confrn )r� r� � something.newr� ) rS r� rN Z load_vectorr! r8 rv r" r Z configfiler' �endswith)r Zunused_makedirr� Z test_cert�f�resultr r r �test_find_duplicative_names� s, ( � � � �z0DuplicativeCertsTest.test_find_duplicative_names) r* r+ r, r- r r rP r� r. r r r# r r� � s r� c s� e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd d � Ze � d �d d� �Z e � d �dd� �Z e � d �dd� �Z e � d �dd� �Zdd� Z� ZS )�CertPathToLineageTestz=Tests for certbot._internal.cert_manager.cert_path_to_lineagec sB t � �� | j�� | �� tj�| jj ddd�| _ | j | j_ d S �N�liver � fullchain.pem� r r r( r" r� r r r r r � fullchain� cert_pathr r# r r r � s �zCertPathToLineageTest.setUpc C s ddl m} ||�S )Nr )�cert_path_to_lineage)rS r� )r � cli_configr� r r r r5 � s zCertPathToLineageTest._callc C s ddl m} |||�S )Nr ��_archive_files)rS r� )r r� Zfiletyper� r r r r� � s z$CertPathToLineageTest._archive_filesc C s d| � | j�ksJ �d S �Nr )r5 r r r r r �test_basic_match s z&CertPathToLineageTest.test_basic_matchc C sX | j }tj�| j jddd�|_t�tj �� | � |� W d � n1 sJ0 Y d S )Nr� Z SailorMoonr� ) r r r r r r� r� r� r r� r5 )r Zbad_test_configr r r �test_no_match_exists s �z*CertPathToLineageTest.test_no_match_existsz2certbot._internal.cert_manager._acceptable_matchesc C s. dd� g|_ | j| j_d| �| j�ks*J �d S )Nc S s | j S rW )�fullchain_pathr| r r r rk rl z>CertPathToLineageTest.test_options_fullchain.<locals>.<lambda>r )r9 r� r r� r5 �r �mock_acceptable_matchesr r r �test_options_fullchain s z,CertPathToLineageTest.test_options_fullchainc C sB dd� g|_ tj�| jjddd�}|| j_d| �| j�ks>J �d S )Nc S s | j S rW )r� r| r r r rk rl z>CertPathToLineageTest.test_options_cert_path.<locals>.<lambda>r� r zcert.pem)r9 r r r r r r� r5 )r r� Ztest_cert_pathr r r �test_options_cert_path s �z,CertPathToLineageTest.test_options_cert_pathc sB t j�� jjddd�� j_� fdd�g|_d� �� j�ks>J �d S )N�archiver z cert11.pemc s � � | d�S )Nrv r� r| r r r rk rl zACertPathToLineageTest.test_options_archive_cert.<locals>.<lambda>�r r r r r r� r9 r5 r� r r r �test_options_archive_cert s �z/CertPathToLineageTest.test_options_archive_certc sB t j�� jjddd�� j_� fdd�g|_d� �� j�ks>J �d S )Nr� r zfullchain11.pemc s � � | d�S )Nr� r� r| r r r rk $ s zFCertPathToLineageTest.test_options_archive_fullchain.<locals>.<lambda>r� r� r r r � test_options_archive_fullchain s �z4CertPathToLineageTest.test_options_archive_fullchainc C s" | j | j_d| �| j�ks J �d S r� )r� r r� r5 r r r r �test_only_path( s z$CertPathToLineageTest.test_only_path)r* r+ r, r- r r5 r� r� r� r rP r� r� r� r� r� r. r r r# r r� � s r� c sT e Zd ZdZ� fdd�Zdd� Zdd� Ze�d�d d � �Z e�d�d d � �Z � Z S ) �MatchAndCheckOverlapszfTests for certbot._internal.cert_manager.match_and_check_overlaps w/o overlapping archive dirs.c sB t � �� | j�� | �� tj�| jj ddd�| _ | j | j_ d S r� r� r r# r r r 1 s �zMatchAndCheckOverlaps.setUpc C s ddl m} |||||�S )Nr )�match_and_check_overlaps)rS r� )r r� Zacceptable_matchesZ match_funcZrv_funcr� r r r r5 9 s zMatchAndCheckOverlaps._callc s: ddl m} dg� �� j|� � fdd�dd� �ks6J �d S )Nr )�_acceptable_matchesr c s � j jS rW )r r� r| r r r rk @ rl z8MatchAndCheckOverlaps.test_basic_match.<locals>.<lambda>c S s | j S rW rh r| r r r rk @ rl )rS r� r5 r )r r� r r r r� = s �z&MatchAndCheckOverlaps.test_basic_matchz/certbot._internal.cert_manager._search_lineagesc C sH g |_ t�tj��" | �| jd d d � W d � n1 s:0 Y d S rW )r9 r� r� r r� r5 r �r Zmock_search_lineagesr r r �test_no_matchesB s z%MatchAndCheckOverlaps.test_no_matchesc C sL ddg|_ t�tj��" | �| jd d d � W d � n1 s>0 Y d S )NZspiderZdance)r9 r� r� r ZOverlappingMatchFoundr5 r r� r r r �test_too_many_matchesH s z+MatchAndCheckOverlaps.test_too_many_matches) r* r+ r, r- r r5 r� r rP r� r� r. r r r# r r� - s r� c @ s� e Zd ZdZdd� Ze�d�e�d�dd� ��Ze�d�e�d�dd � ��Ze�d�e�d�d d � ��Z e�d�e�d�d d � ��Z e�d�e�d�dd� ��Z e�d�e�d�dd� ��Z dS )�GetCertnameTestr c C s4 t �� }|�� | _| �|j� t�� | _d | j_ d S rW ) rN rO �startr� Z addCleanup�stopr rd r r; )r Zget_utility_patchr r r r R s zGetCertnameTest.setUpr{ z2certbot._internal.storage.lineagename_for_filenamec C sl dg|_ d|_ ddlm} d}tjdf| �� j_ |j| jddd�dgksLJ �|| �� jj d d v shJ �d S ) N�example.com.confr� r r1 zWhich certificate would you�verbF��allow_multiple� r9 r3 r2 rD rE r� r� � get_certnamesr � call_args�r � mock_name� mock_filesr2 �promptr r r �test_get_certnamesY s ��z"GetCertnameTest.test_get_certnamesc C sn dg|_ d|_ ddlm} d}tjdf| �� j_ |j| jdd|d�dgksNJ �| �� jj d d |ksjJ �d S ) Nr� r� r r1 � custom promptr� F�r� Z custom_promptr� r� r r r � test_get_certnames_custom_prompte s ���z0GetCertnameTest.test_get_certnames_custom_promptc C sn dg|_ d|_ ddlm} tjdf| �� j_ t�t j ��" |j | j ddd� W d � n1 s`0 Y d S )Nr� r� r r1 �erroring_anywayFr� ) r9 r3 r2 rD r� r� r� r� r� r r� r� r �r r� r� r2 r r r � test_get_certnames_user_aborts s z-GetCertnameTest.test_get_certnames_user_abortc C sn dg|_ d|_ ddlm} d}tjdgf| �� j_ |j| jddd�dgksNJ �|| �� jj d d v sjJ �d S ) Nr� r� r r1 z Which certificate(s) would your� Tr� � r9 r3 r2 rD rE r� rF r� r r� r� r r r �!test_get_certnames_allow_multiple} s � ��z1GetCertnameTest.test_get_certnames_allow_multiplec C sp dg|_ d|_ ddlm} d}tjdgf| �� j_ |j| jdd|d�dgksPJ �| �� jj d d |kslJ �d S ) Nr� r� r r1 r� r� Tr� r� r� r r r �/test_get_certnames_allow_multiple_custom_prompt� s � ���z?GetCertnameTest.test_get_certnames_allow_multiple_custom_promptc C sn dg|_ d|_ ddlm} tjg f| �� j_ t�t j ��" |j | j ddd� W d � n1 s`0 Y d S )Nr� r� r r1 r� Tr� ) r9 r3 r2 rD r� r� rF r� r� r r� r� r r� r r r �,test_get_certnames_allow_multiple_user_abort� s z<GetCertnameTest.test_get_certnames_allow_multiple_user_abortN) r* r+ r, r- r r rP r� r� r� r� r� r� r r r r r� O s( r� �__main__rn )+r- rt re �sysrc Zunittestr r& r� Zcertbotr r Zcertbot._internal.storager Zcertbot._internal.testsr Zcertbot.compatr r Zcertbot.displayr rD Z certbot.testsrN ZConfigTestCaser ZBaseRenewableCertTestr/ rQ ry r� r� r� r� r� r� ZTestCaser� r* �exit�main�argv�__file__r r r r �<module> s< ,W ^&@"U
💾 Save Changes
❌ Cancel