From 9d70c4acfbc78fda0a2073f2d157a8e8bd9de84b Mon Sep 17 00:00:00 2001
From: Brad Warren <bmw@eff.org>
Date: Wed, 23 Mar 2016 15:11:36 -0700
Subject: [PATCH] Add directory_select method to FileDisplay

---
 letsencrypt/display/util.py            | 14 ++++++++++++++
 letsencrypt/tests/display/util_test.py |  9 +++++++++
 2 files changed, 23 insertions(+)

diff --git a/letsencrypt/display/util.py b/letsencrypt/display/util.py
index 05e280118..c94b0c921 100644
--- a/letsencrypt/display/util.py
+++ b/letsencrypt/display/util.py
@@ -7,6 +7,7 @@ import zope.interface
 
 from letsencrypt import interfaces
 from letsencrypt import errors
+from letsencrypt.display import completer
 
 WIDTH = 72
 HEIGHT = 20
@@ -339,6 +340,19 @@ class FileDisplay(object):
             else:
                 return code, []
 
+    def directory_select(self, message, **unused_kwargs):
+        """Display a directory selection screen.
+
+        :param str message: prompt to give the user
+
+        :returns: tuple of the form (`code`, `string`) where
+            `code` - int display exit code
+            `string` - input entered by the user
+
+        """
+        with completer.Completer():
+            return self.input(message)
+
     def _scrub_checklist_input(self, indices, tags):
         # pylint: disable=no-self-use
         """Validate input and transform indices to appropriate tags.
diff --git a/letsencrypt/tests/display/util_test.py b/letsencrypt/tests/display/util_test.py
index adec265dc..32783b3bd 100644
--- a/letsencrypt/tests/display/util_test.py
+++ b/letsencrypt/tests/display/util_test.py
@@ -232,6 +232,15 @@ class FileOutputDisplayTest(unittest.TestCase):
                 self.displayer._scrub_checklist_input(list_, TAGS))
             self.assertEqual(set_tags, exp[i])
 
+    @mock.patch("letsencrypt.display.util.FileDisplay.input")
+    def test_directory_select(self, mock_input):
+        message = "msg"
+        result = (display_util.OK, "/var/www/html",)
+        mock_input.return_value = result
+
+        self.assertEqual(self.displayer.directory_select(message), result)
+        mock_input.assert_called_once_with(message)
+
     def test_scrub_checklist_input_invalid(self):
         # pylint: disable=protected-access
         indices = [