summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--test/io/console/test_io_console.rb5
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