From 111ca657d86e0caf31dbc81f83037a021fd22131 Mon Sep 17 00:00:00 2001
From: Amjad Mashaal <me@amjad.io>
Date: Mon, 6 Jun 2016 21:17:02 +0200
Subject: [PATCH] Adding testing

---
 certbot/tests/cli_test.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/certbot/tests/cli_test.py b/certbot/tests/cli_test.py
index 4ae69e69d..06aba4ab3 100644
--- a/certbot/tests/cli_test.py
+++ b/certbot/tests/cli_test.py
@@ -882,6 +882,15 @@ class CLITest(unittest.TestCase):  # pylint: disable=too-many-public-methods
         mock_sys.exit.assert_called_with(''.join(
             traceback.format_exception_only(KeyboardInterrupt, interrupt)))
 
+        # Test dialog errors
+
+        import dialog
+        exception = dialog.error(message="test message")
+        main._handle_exception(
+                dialog.DialogError, exc_value=exception, trace=None, config=None)
+        error_msg = mock_sys.exit.call_args_list[-1][0][0]
+        self.assertTrue("test message" in error_msg)
+
     def test_read_file(self):
         rel_test_path = os.path.relpath(os.path.join(self.tmp_dir, 'foo'))
         self.assertRaises(