starred/certbot-certbot-1
1
0
Fork 0
mirror of https://github.com/certbot/certbot.git synced 2026-08-02 19:42:02 +02:00
Code Issues Packages Projects Releases Wiki Activity

add staple-ocsp test

Browse Source
This commit is contained in:
Brad Warren
2016-09-19 15:05:16 -07:00
parent fea079400f
commit dff1368765
1 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
certbot/tests/client_test.py
+17 -7
View File
@@ -332,31 +332,41 @@ class ClientTest(unittest.TestCase):
@mock.patch("certbot.client.enhancements") @mock.patch("certbot.client.enhancements")
def test_enhance_config_no_ask(self, mock_enhancements): def test_enhance_config_no_ask(self, mock_enhancements):
config = ConfigHelper(redirect=True, hsts=False, uir=False) config = ConfigHelper(redirect=True, hsts=False,
uir=False, staple=False)
self.assertRaises(errors.Error, self.client.enhance_config, self.assertRaises(errors.Error, self.client.enhance_config,
["foo.bar"], config, None) ["foo.bar"], config, None)
mock_enhancements.ask.return_value = True mock_enhancements.ask.return_value = True
installer = mock.MagicMock() installer = mock.MagicMock()
self.client.installer = installer self.client.installer = installer
installer.supported_enhancements.return_value = ["redirect", "ensure-http-header"] installer.supported_enhancements.return_value = [
"redirect", "ensure-http-header", "staple-ocsp"]
config = ConfigHelper(redirect=True, hsts=False, uir=False) config = ConfigHelper(redirect=True, hsts=False,
uir=False, staple=False)
self.client.enhance_config(["foo.bar"], config, None) self.client.enhance_config(["foo.bar"], config, None)
installer.enhance.assert_called_with("foo.bar", "redirect", None) installer.enhance.assert_called_with("foo.bar", "redirect", None)
config = ConfigHelper(redirect=False, hsts=True, uir=False) config = ConfigHelper(redirect=False, hsts=True,
uir=False, staple=False)
self.client.enhance_config(["foo.bar"], config, None) self.client.enhance_config(["foo.bar"], config, None)
installer.enhance.assert_called_with("foo.bar", "ensure-http-header", installer.enhance.assert_called_with("foo.bar", "ensure-http-header",
"Strict-Transport-Security") "Strict-Transport-Security")
config = ConfigHelper(redirect=False, hsts=False, uir=True) config = ConfigHelper(redirect=False, hsts=False,
uir=True, staple=False)
self.client.enhance_config(["foo.bar"], config, None) self.client.enhance_config(["foo.bar"], config, None)
installer.enhance.assert_called_with("foo.bar", "ensure-http-header", installer.enhance.assert_called_with("foo.bar", "ensure-http-header",
"Upgrade-Insecure-Requests") "Upgrade-Insecure-Requests")
self.assertEqual(installer.save.call_count, 3) config = ConfigHelper(redirect=False, hsts=False,
self.assertEqual(installer.restart.call_count, 3) uir=False, staple=True)
self.client.enhance_config(["foo.bar"], config, None)
installer.enhance.assert_called_with("foo.bar", "staple-ocsp", None)
self.assertEqual(installer.save.call_count, 4)
self.assertEqual(installer.restart.call_count, 4)
@mock.patch("certbot.client.enhancements") @mock.patch("certbot.client.enhancements")
def test_enhance_config_unsupported(self, mock_enhancements): def test_enhance_config_unsupported(self, mock_enhancements):