Close stdin before waiting to allow --select-1 to work

iterfzf/__init__.py

@@ -106,16 +106,16 @@ def iterfzf(
             if e.errno != errno.EPIPE and errno.EPIPE != 32:
                 raise
             break
-    if proc is None or proc.wait() not in [0, 1]:
-        if print_query:
-            return None, None
-        else:
-            return None
     try:
         stdin.close()
     except IOError as e:
         if e.errno != errno.EPIPE and errno.EPIPE != 32:
             raise
+    if proc is None or proc.wait() not in [0, 1]:
+        if print_query:
+            return None, None
+        else:
+            return None
     stdout = proc.stdout
     decode = (lambda b: b) if byte else (lambda t: t.decode(encoding))
     output = [decode(ln.strip(b'\r\n\0')) for ln in iter(stdout.readline, b'')]

iterfzf/test_iterfzf.py

@@ -0,0 +1,26 @@
+import unittest
+import iterfzf
+
+flavors = [
+    "Chocolate", "Chocolate Chip", "Vanilla", "Strawberry", "Blueberry",
+    "Rocky Road"
+]
+
+
+class IterFzfTest(unittest.TestCase):
+
+    def test_no_query(self):
+        choice = iterfzf.iterfzf(flavors, executable="fzf")
+        self.assertEqual("Chocolate", choice)
+
+    def test_select_one(self):
+        choice = iterfzf.iterfzf(
+            flavors, query="Vani", __extra__=["-1"], executable="fzf"
+        )
+        self.assertEqual("Vanilla", choice)
+
+    def test_select_one_ambiguous(self):
+        choice = iterfzf.iterfzf(
+            flavors, query="Choc", __extra__=["-1"], executable="fzf"
+        )
+        self.assertTrue(choice.rfind('Chocolate') == 0)