← Back
Editing: enhancements_test.cpython-38.pyc
U �]p � @ sd d Z ddlZddlZddlmZ ddlmZ ddlm mZ G dd� de j �Z e dkr`e� � dS )z Tests for new style enhancements� N)� enhancements)�nullc sH e Zd ZdZ� fdd�Ze�� dd� �Zdd� Zdd � Z d d � Z � Z S ) �EnhancementTestz@Tests for new style enhancements in certbot.plugins.enhancementsc s" t t| ��� tjtjd�| _d S )N)�spec)�superr �setUp�mockZ MagicMockr ZAutoHSTSEnhancement� mockinstaller��self�� __class__� �C/usr/lib/python3/dist-packages/certbot/plugins/enhancements_test.pyr s zEnhancementTest.setUpc C s� ddd�ddd�g}t �d|��& d| j_d| j_tt�| j��}W 5 Q R X | �t |�d� | � d d � |D �� | � d d � |D �� d S ) N�autohsts� auto_hsts)�nameZcli_dest� somethingelse� somethingz#certbot.plugins.enhancements._INDEXT� c S s g | ]}|d dkr|�qS )r r r ��.0�ir r r � <listcomp>$ s zIEnhancementTest.test_enhancement_enabled_enhancements.<locals>.<listcomp>c S s g | ]}|d dkr|�qS )r r r r r r r r % s ) r Zpatch�configr r �listr �enabled_enhancements� assertEqual�len� assertTrue)r �_Z FAKEINDEX�enabledr r r �%test_enhancement_enabled_enhancements s ��� z5EnhancementTest.test_enhancement_enabled_enhancementsc C st | � tdd� t�| j�D ��d� | �t�| j�� d| j_| � tdd� t�| j�D ��d� | �t�| j�� d S )Nc S s g | ]}|�qS r r r r r r r ) s z6EnhancementTest.test_are_requested.<locals>.<listcomp>r Tc S s g | ]}|�qS r r r r r r r - s � ) r r r r r � assertFalseZ are_requestedr r r r r r �test_are_requested' s � �z"EnhancementTest.test_are_requestedc C sD d| j _t�| j d�}| �t�| j | j�� | �t�| j |�� d S )NTr ) r r r Z Installerr r Z are_supportedr r$ )r Z unsupportedr r r �test_are_supported0 s z"EnhancementTest.test_are_supportedc C sV d| j _ddg}d}t�||| j| j � | �| jjj� | �| jjj d ||f� d S )NTz example.comzwww.example.com�lineager ) r r r �enabler r Zenable_autohstsZcalledr Z call_args)r Zdomainsr' r r r � test_enable6 s �zEnhancementTest.test_enable) �__name__� __module__� __qualname__�__doc__r � test_utilZpatch_get_utilityr" r% r& r) � __classcell__r r r r r s r �__main__)r- Zunittestr Zcertbot.pluginsr r Zcertbot.tests.utilZtests�utilr. ZConfigTestCaser r* �mainr r r r �<module> s 5
Save File
Cancel