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 sd ddddddg}t d|0 d| j_d| j_tt| j}W d n1 sV0 Y d S )NZautohsts auto_hsts)nameZcli_destZ
somethingelse somethingz#certbot.plugins.enhancements._INDEXT)r Zpatchconfigr r listr Zenabled_enhancements)r _Z FAKEINDEXZenabledr r r %test_enhancement_enabled_enhancements s
.z5EnhancementTest.test_enhancement_enabled_enhancementsc C s d| j _d S )NT)r r r
r r r test_are_requested' s z"EnhancementTest.test_are_requestedc C s d| j _t| j d}d S )NTr )r r r Z Installer)r Z unsupportedr r r test_are_supported. s z"EnhancementTest.test_are_supportedc C s, d| j _ddg}d}t||| j| j d S )NTz example.comzwww.example.comlineage)r r r enabler )r Zdomainsr r r r test_enable4 s
zEnhancementTest.test_enable) __name__
__module__ __qualname____doc__r test_utilZpatch_display_utilr r r r
__classcell__r r r r r s
r __main__ )r sysZunittestr ZpytestZcertbot._internal.pluginsr Zcertbot.pluginsr Zcertbot.tests.utilZtestsutilr ZConfigTestCaser r exitmainargv__file__r r r r s 1