diff options
-rw-r--r-- | test/io/console/test_io_console.rb | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/test/io/console/test_io_console.rb b/test/io/console/test_io_console.rb index b8a1cf95b5..237a41a35f 100644 --- a/test/io/console/test_io_console.rb +++ b/test/io/console/test_io_console.rb @@ -249,12 +249,13 @@ class TestIO_Console < Test::Unit::TestCase s.close if s end - def run_pty(src) + def run_pty(src, n = 1) r, w, pid = PTY.spawn(EnvUtil.rubybin, "-rio/console", "-e", src) rescue RuntimeError skip $! else - result = r.readlines.map(&:chomp) + result = [] + n.times {result << r.gets.chomp} Process.wait(pid) if block_given? yield result |