a
}|g @ s d Z ddlZddlmZ ddlZddlmZ ddlmZ ddl m
m Z G dd de j
Zedkreeejd d eg dS )
z Tests for new style enhancements N)mock)null) enhancementsc sH e Zd ZdZ fddZe dd Zdd Zdd Z d
d Z
Z S ) EnhancementTestz@Tests for new style enhancements in certbot.plugins.enhancementsc s t tjtjd| _d S )N)spec)supersetUpr Z MagicMockr ZAutoHSTSEnhancement
mockinstallerself __class__ U/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/enhancements_test.pyr s
zEnhancementTest.setUpc C s ddddddg}t d|0 d| j_d| j_tt| j}W d n1 sV0 Y t|dkspJ d d
|D sJ d d
|D sJ d S ) Nautohsts auto_hsts)nameZcli_dest
somethingelse somethingz#certbot.plugins.enhancements._INDEXT c S s g | ]}|d dkr|qS )r r r .0ir r r
$ zIEnhancementTest.test_enhancement_enabled_enhancements..c S s g | ]}|d dkr|qS )r r r r r r r r % r ) r Zpatchconfigr r listr enabled_enhancementslen)r _Z FAKEINDEXZenabledr r r %test_enhancement_enabled_enhancements s
.z5EnhancementTest.test_enhancement_enabled_enhancementsc C sd t tt| jdks J t| jr,J d| j_t tt| jdksPJ t| js`J d S )Nr T )r r r r r Z
are_requestedr r
r r r test_are_requested' s
z"EnhancementTest.test_are_requestedc C s@ d| j _t| j d}t| j | js*J t| j |r s 1