SWAMPy

✎ CARVE FLESH

EDITING: enhancements_test.py

"""Tests for new style enhancements"""
import sys
import unittest
from unittest import mock

import pytest

from certbot._internal.plugins import null
from certbot.plugins import enhancements
import certbot.tests.util as test_util

class EnhancementTest(test_util.ConfigTestCase):
    """Tests for new style enhancements in certbot.plugins.enhancements"""

def setUp(self):
        super().setUp()
        self.mockinstaller = mock.MagicMock(spec=enhancements.AutoHSTSEnhancement)

@test_util.patch_display_util()
    def test_enhancement_enabled_enhancements(self, _):
        FAKEINDEX = [
            {
                "name": "autohsts",
                "cli_dest": "auto_hsts",
            },
            {
                "name": "somethingelse",
                "cli_dest": "something",
            }
        ]
        with mock.patch("certbot.plugins.enhancements._INDEX", FAKEINDEX):
            self.config.auto_hsts = True
            self.config.something = True
            enabled = list(enhancements.enabled_enhancements(self.config))
        assert len(enabled) == 2
        assert [i for i in enabled if i["name"] == "autohsts"]
        assert [i for i in enabled if i["name"] == "somethingelse"]

def test_are_requested(self):
        assert len(list(enhancements.enabled_enhancements(self.config))) == 0
        assert not enhancements.are_requested(self.config)
        self.config.auto_hsts = True
        assert len(list(enhancements.enabled_enhancements(self.config))) == 1
        assert enhancements.are_requested(self.config)

def test_are_supported(self):
        self.config.auto_hsts = True
        unsupported = null.Installer(self.config, "null")
        assert enhancements.are_supported(self.config, self.mockinstaller)
        assert not enhancements.are_supported(self.config, unsupported)

def test_enable(self):
        self.config.auto_hsts = True
        domains = ["example.com", "www.example.com"]
        lineage = "lineage"
        enhancements.enable(lineage, domains, self.mockinstaller, self.config)
        assert self.mockinstaller.enable_autohsts.called
        assert self.mockinstaller.enable_autohsts.call_args[0] == \
                          (lineage, domains)

if __name__ == '__main__':
    sys.exit(pytest.main(sys.argv[1:] + [__file__]))  # pragma: no cover

CANCEL

Name	Type	Size	Modified	Actions
↩ ..	DIR	—	—
📁 __pycache__/	DIR	—	2026-03-10 17:37
📄 __init__.py	PY	28 B	2024-02-08 19:45	EDIT
📄 common_test.py	PY	14.1 KB	2024-02-08 19:45	EDIT
📄 disco_test.py	PY	11.2 KB	2024-02-08 19:45	EDIT
📄 dns_common_test.py	PY	8.3 KB	2024-02-08 19:45	EDIT
📄 enhancements_test.py	PY	2.3 KB	2024-02-08 19:45	EDIT
📄 manual_test.py	PY	7.3 KB	2024-02-08 19:45	EDIT
📄 null_test.py	PY	674 B	2024-02-08 19:45	EDIT
📄 selection_test.py	PY	9.7 KB	2024-02-08 19:45	EDIT
📄 standalone_test.py	PY	6.7 KB	2024-02-08 19:45	EDIT
📄 storage_test.py	PY	5.4 KB	2024-02-08 19:45	EDIT
📄 util_test.py	PY	1.5 KB	2024-02-08 19:45	EDIT
📄 webroot_test.py	PY	15 KB	2024-02-08 19:45	EDIT