support filenos as args for readline wrapper

Author: RonnyPfannschmidt

pyrepl/readline.py

@@ -182,9 +182,9 @@ class _ReadlineWrapper(object):
     startup_hook = None
     config = ReadlineConfig()
 
-    def __init__(self):
-        self.f_in = os.dup(0)
-        self.f_out = os.dup(1)
+    def __init__(self, f_in=None, f_out=None):
+        self.f_in = f_in if f_in is not None else os.dup(0)
+        self.f_out = f_out if f_out is not None else os.dup(1)
 
     def get_reader(self):
         if self.reader is None:

testing/test_readline.py

@@ -4,11 +4,12 @@
 
 
 def test_raw_input():
-    readline_wrapper = _ReadlineWrapper()
     master, slave = pty.openpty()
-    readline_wrapper.f_in = slave
+    readline_wrapper = _ReadlineWrapper(slave, slave)
     os.write(master, b'input\n')
-    result = readline_wrapper.raw_input('prompt:')
+
+    result = readline_wrapper.get_reader().readline()
+    #result = readline_wrapper.raw_input('prompt:')
     assert result == 'input'
     # A bytes string on python2, a unicode string on python3.
     assert isinstance(result, str)