|
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
�]K � @ s| d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddl m
m Z G dd� de j
�Zed krxe�� dS )
z$Tests for renewal updater interfaces� N)�
interfaces)�main)�updater)� enhancementsc s� e Zd ZdZ� fdd�Ze�d�e�d�e�d�e�� dd� ����Z d d
� Z
e�d �d d
� �Z e�d �dd� �Z e�d�dd� �Z
dd� Ze�d�dd� �Zdd� Ze�d�dd� �Zdd� Z� ZS ) �RenewUpdaterTestz@Tests for interfaces.RenewDeployer and interfaces.GenericUpdaterc sN t t| ��� tjtjd�| _t�� | j_tjtj d�| _
tjt j d�| _
d S )N)�spec)�superr �setUp�mock� MagicMockr ZGenericUpdater�generic_updater�restartZ
RenewDeployer�renew_deployerr �AutoHSTSEnhancement�
mockinstaller��self�� __class__� �A/usr/lib/python3/dist-packages/certbot/tests/renewupdater_test.pyr s
zRenewUpdaterTest.setUpz certbot.main._get_and_save_certz5certbot.plugins.selection.choose_configurator_pluginsz2certbot.plugins.selection.get_unprepared_installerc C s� t �� |_| j}|d f|_||_t �d�� t�| jd t �� � W 5 Q R X | �|j j
� |j � � |j � � t
�| jt �� d � | �|j jd� | �|j j
� d S )Nz certbot.main._init_le_client� )r
r � return_valuer �patchr Z
renew_cert�config�
assertTruer
�calledZ
reset_mockZgeneric_updatesr �run_generic_updaters� assertEqual�
call_count� assertFalse)r �_� mock_getiZ mock_selectZ mock_getsaveZmock_generic_updaterr r r �test_server_updates s
z$RenewUpdaterTest.test_server_updatesc C s4 t �� }| j}t�| j||� | �|j�|�� d S �N) r
r r r �run_renewal_deployerr r Z renew_deployZ called_with)r ZlineageZ
mock_deployerr r r �test_renew_deployer- s z$RenewUpdaterTest.test_renew_deployerz certbot.updater.logger.debugc C s> d| j _t�| j d d � | �|j� | �|jd d d� d S )NTr z"Skipping updaters in dry-run mode.)r �dry_runr r r r r � call_args�r Zmock_logr r r �test_updater_skip_dry_run3 s �z*RenewUpdaterTest.test_updater_skip_dry_runc C s> d| j _t�| j d d � | �|j� | �|jd d d� d S )NTr z*Skipping renewal deployer in dry-run mode.)r r' r r% r r r r( r) r r r �test_deployer_skip_dry_run; s �z+RenewUpdaterTest.test_deployer_skip_dry_runc C sB | j |_t�| jt�� d � | �| j jj � | �
| j jj d� d S )Nr ) r r r r r r
r r �update_autohstsr r r �r r" r r r �test_enhancement_updatesC s z)RenewUpdaterTest.test_enhancement_updatesc C s* t �| jt�� | j� | �| jjj� d S r$ ) r r% r r
r r r �deploy_autohstsr r r r r �test_enhancement_deployerJ s �z*RenewUpdaterTest.test_enhancement_deployerc C s8 d| j _| j|_t�| j t�� d � | �| jj j
� d S �NT) r �disable_renew_updatesr r r r r
r r r, r r- r r r �#test_enhancement_updates_not_calledO s z4RenewUpdaterTest.test_enhancement_updates_not_calledc C s2 d| j _t�| j t�� | j� | �| jjj � d S r1 )
r r2 r r% r
r r r r/ r r r r r �$test_enhancement_deployer_not_calledV s
�z5RenewUpdaterTest.test_enhancement_deployer_not_calledc C s\ dt jd ddd�g}| j|_t�d|�� t�| jt� � d � W 5 Q R X | �
| jj j � d S �NZTestr/ Zenable_autohsts)�name�classZupdater_functionZdeployer_functionZenable_functionz#certbot.plugins.enhancements._INDEX)
r r r r r
r r r r r r r, r )r r" � FAKEINDEXr r r �test_enhancement_no_updater\ s �� z,RenewUpdaterTest.test_enhancement_no_updaterc C sV dt jdd dd�g}t�d|�� t�| jt�� | j� W 5 Q R X | � | jj
j � d S r5 ) r r r
r r r% r r r r r/ r )r r8 r r r � test_enhancement_no_deployerl s �� �z-RenewUpdaterTest.test_enhancement_no_deployer)�__name__�
__module__� __qualname__�__doc__r r
r � test_utilZpatch_get_utilityr# r& r* r+ r. r0 r3 r4 r9 r: �
__classcell__r r r r r s*
r �__main__)r> Zunittestr
Zcertbotr r r Zcertbot.pluginsr Zcertbot.tests.utilZtests�utilr? ZConfigTestCaser r; r r r r �<module> s n