Hide keyboard shortcuts

Hot-keys on this page

r m x p   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

"""Module for enhancement UI.""" 

import logging 

import unittest 

 

import mock 

 

from certbot import errors 

from certbot.display import util as display_util 

 

 

class AskTest(unittest.TestCase): 

"""Test the ask method.""" 

def setUp(self): 

logging.disable(logging.CRITICAL) 

 

def tearDown(self): 

logging.disable(logging.NOTSET) 

 

@classmethod 

def _call(cls, enhancement): 

from certbot.display.enhancements import ask 

return ask(enhancement) 

 

@mock.patch("certbot.display.enhancements.util") 

def test_redirect(self, mock_util): 

mock_util().menu.return_value = (display_util.OK, 1) 

self.assertTrue(self._call("redirect")) 

 

def test_key_error(self): 

self.assertRaises(errors.Error, self._call, "unknown_enhancement") 

 

 

class RedirectTest(unittest.TestCase): 

"""Test the redirect_by_default method.""" 

@classmethod 

def _call(cls): 

from certbot.display.enhancements import redirect_by_default 

return redirect_by_default() 

 

@mock.patch("certbot.display.enhancements.util") 

def test_secure(self, mock_util): 

mock_util().menu.return_value = (display_util.OK, 1) 

self.assertTrue(self._call()) 

 

@mock.patch("certbot.display.enhancements.util") 

def test_cancel(self, mock_util): 

mock_util().menu.return_value = (display_util.CANCEL, 1) 

self.assertFalse(self._call()) 

 

@mock.patch("certbot.display.enhancements.util") 

def test_easy(self, mock_util): 

mock_util().menu.return_value = (display_util.OK, 0) 

self.assertFalse(self._call()) 

 

 

if __name__ == "__main__": 

unittest.main() # pragma: no cover