Your IP : 216.73.216.172
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