OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	05/25/2021 01:14:26 PM	rwxr-xr-x
📄 __init__.py	30 bytes	02/07/2019 09:20:29 PM	rw-r--r--
📁 __pycache__	-	05/25/2021 01:14:27 PM	rwxr-xr-x
📄 common.py	16.9 KB	02/07/2019 09:20:29 PM	rw-r--r--
📄 common_test.py	16.53 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 disco.py	9.92 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 disco_test.py	11.34 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 dns_common.py	11.7 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 dns_common_lexicon.py	5.39 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 dns_common_lexicon_test.py	651 bytes	02/07/2019 09:20:30 PM	rw-r--r--
📄 dns_common_test.py	8.25 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 dns_test_common.py	1.61 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 dns_test_common_lexicon.py	5.48 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 enhancements.py	5.58 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 enhancements_test.py	2.36 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 manual.py	10.6 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 manual_test.py	7.37 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 null.py	1.34 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 null_test.py	624 bytes	02/07/2019 09:20:30 PM	rw-r--r--
📄 selection.py	13.55 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 selection_test.py	7.76 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 standalone.py	11.36 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 standalone_test.py	9.26 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 storage.py	4.08 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 storage_test.py	5.37 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 util.py	1.7 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 util_test.py	1.61 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 webroot.py	11.9 KB	02/07/2019 09:20:30 PM	rw-r--r--
📄 webroot_test.py	11.95 KB	02/07/2019 09:20:30 PM	rw-r--r--

Editing: enhancements_test.py

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

from certbot.plugins import enhancements
from certbot.plugins import null

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(EnhancementTest, self).setUp()
        self.mockinstaller = mock.MagicMock(spec=enhancements.AutoHSTSEnhancement)

@test_util.patch_get_utility()
    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))
        self.assertEqual(len(enabled), 2)
        self.assertTrue([i for i in enabled if i["name"] == "autohsts"])
        self.assertTrue([i for i in enabled if i["name"] == "somethingelse"])

def test_are_requested(self):
        self.assertEqual(
            len([i for i in enhancements.enabled_enhancements(self.config)]), 0)
        self.assertFalse(enhancements.are_requested(self.config))
        self.config.auto_hsts = True
        self.assertEqual(
            len([i for i in enhancements.enabled_enhancements(self.config)]), 1)
        self.assertTrue(enhancements.are_requested(self.config))

def test_are_supported(self):
        self.config.auto_hsts = True
        unsupported = null.Installer(self.config, "null")
        self.assertTrue(enhancements.are_supported(self.config, self.mockinstaller))
        self.assertFalse(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)
        self.assertTrue(self.mockinstaller.enable_autohsts.called)
        self.assertEqual(self.mockinstaller.enable_autohsts.call_args[0],
                          (lineage, domains))

if __name__ == '__main__':
    unittest.main()  # pragma: no cover