mirror of
https://github.com/certbot/certbot.git
synced 2026-01-23 07:20:55 +03:00
59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
"""Test letsencrypt_apache.display_ops."""
|
|
import sys
|
|
import unittest
|
|
|
|
import mock
|
|
import zope.component
|
|
|
|
from letsencrypt_apache.tests import util
|
|
|
|
from letsencrypt.display import util as display_util
|
|
|
|
|
|
class SelectVhostTest(unittest.TestCase):
|
|
"""Tests for letsencrypt_apache.display_ops.select_vhost."""
|
|
|
|
def setUp(self):
|
|
zope.component.provideUtility(display_util.FileDisplay(sys.stdout))
|
|
self.base_dir = "/example_path"
|
|
self.vhosts = util.get_vh_truth(
|
|
self.base_dir, "debian_apache_2_4/two_vhost_80")
|
|
|
|
@classmethod
|
|
def _call(cls, vhosts):
|
|
from letsencrypt_apache.display_ops import select_vhost
|
|
return select_vhost("example.com", vhosts)
|
|
|
|
@mock.patch("letsencrypt_apache.display_ops.zope.component.getUtility")
|
|
def test_successful_choice(self, mock_util):
|
|
mock_util().menu.return_value = (display_util.OK, 3)
|
|
self.assertEqual(self.vhosts[3], self._call(self.vhosts))
|
|
|
|
@mock.patch("letsencrypt_apache.display_ops.zope.component.getUtility")
|
|
def test_more_info_cancel(self, mock_util):
|
|
mock_util().menu.side_effect = [
|
|
(display_util.HELP, 1),
|
|
(display_util.HELP, 0),
|
|
(display_util.CANCEL, -1),
|
|
]
|
|
|
|
self.assertEqual(None, self._call(self.vhosts))
|
|
self.assertEqual(mock_util().notification.call_count, 2)
|
|
|
|
def test_no_vhosts(self):
|
|
self.assertEqual(self._call([]), None)
|
|
|
|
@mock.patch("letsencrypt_apache.display_ops.display_util")
|
|
@mock.patch("letsencrypt_apache.display_ops.zope.component.getUtility")
|
|
@mock.patch("letsencrypt_apache.display_ops.logging")
|
|
def test_small_display(self, mock_logging, mock_util, mock_display_util):
|
|
mock_display_util.WIDTH = 20
|
|
mock_util().menu.return_value = (display_util.OK, 0)
|
|
self._call(self.vhosts)
|
|
|
|
self.assertTrue(mock_logging.is_called)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main() # pragma: no cover
|