begin require 'io/console' require 'test/unit' require 'pty' rescue LoadError end class TestIO_Console < Test::Unit::TestCase Bug6116 = '[ruby-dev:45309]' def test_raw helper {|m, s| s.print "abc\n" assert_equal("abc\r\n", m.gets) assert_send([s, :echo?]) s.raw { assert_not_send([s, :echo?], Bug6116) s.print "def\n" assert_equal("def\n", m.gets) } assert_send([s, :echo?]) s.print "ghi\n" assert_equal("ghi\r\n", m.gets) } end def test_raw_minchar helper {|m, s| len = 0 assert_equal([nil, 0], [s.getch(min: 0), len]) main = Thread.current go = false th = Thread.start { len += 1 m.print("a") m.flush sleep 0.01 until go and main.stop? len += 10 m.print("1234567890") m.flush } begin assert_equal(["a", 1], [s.getch(min: 1), len]) go = true assert_equal(["1", 11], [s.getch, len]) ensure th.join end } end def test_raw_timeout helper {|m, s| len = 0 assert_equal([nil, 0], [s.getch(min: 0, time: 0.1), len]) main = Thread.current th = Thread.start { sleep 0.01 until main.stop? len += 2 m.print("ab") } begin assert_equal(["a", 2], [s.getch(min: 1, time: 1), len]) assert_equal(["b", 2], [s.getch(time: 1), len]) ensure th.join end } end def test_cooked helper {|m, s| assert_send([s, :echo?]) s.raw { s.print "abc\n" assert_equal("abc\n", m.gets) assert_not_send([s, :echo?], Bug6116) s.cooked { assert_send([s, :echo?]) s.print "def\n" assert_equal("def\r\n", m.gets) } assert_not_send([s, :echo?], Bug6116) } assert_send([s, :echo?]) s.print "ghi\n" assert_equal("ghi\r\n", m.gets) } end def test_echo helper {|m, s| assert_send([s, :echo?]) m.print "a" assert_equal("a", m.readpartial(10)) } end def test_noecho helper {|m, s| s.noecho { assert_not_send([s, :echo?]) m.print "a" sleep 0.1 } m.print "b" assert_equal("b", m.readpartial(10)) } end def test_noecho2 helper {|m, s| assert_send([s, :echo?]) m.print "a\n" sleep 0.1 s.print "b\n" sleep 0.1 assert_equal("a\r\nb\r\n", m.readpartial(10)) assert_equal("a\n", s.readpartial(10)) s.noecho { assert_not_send([s, :echo?]) m.print "a\n" s.print "b\n" assert_equal("b\r\n", m.readpartial(10)) assert_equal("a\n", s.readpartial(10)) } assert_send([s, :echo?]) m.print "a\n" sleep 0.1 s.print "b\n" sleep 0.1 assert_equal("a\r\nb\r\n", m.readpartial(10)) assert_equal("a\n", s.readpartial(10)) } end def test_setecho helper {|m, s| assert_send([s, :echo?]) s.echo = false m.print "a" sleep 0.1 s.echo = true m.print "b" assert_equal("b", m.readpartial(10)) } end def test_setecho2 helper {|m, s| assert_send([s, :echo?]) m.print "a\n" sleep 0.1 s.print "b\n" sleep 0.1 assert_equal("a\r\nb\r\n", m.readpartial(10)) assert_equal("a\n", s.readpartial(10)) s.echo = false assert_not_send([s, :echo?]) m.print "a\n" s.print "b\n" assert_equal("b\r\n", m.readpartial(10)) assert_equal("a\n", s.readpartial(10)) s.echo = true assert_send([s, :echo?]) m.print "a\n" sleep 0.1 s.print "b\n" sleep 0.1 assert_equal("a\r\nb\r\n", m.readpartial(10)) assert_equal("a\n", s.readpartial(10)) } end def test_iflush helper {|m, s| m.print "a" s.iflush m.print "b\n" assert_equal("b\n", s.readpartial(10)) } end def test_oflush helper {|m, s| s.print "a" s.oflush # oflush may be issued after "a" is already sent. s.print "b" assert_include(["b", "ab"], m.readpartial(10)) } end def test_ioflush helper {|m, s| m.print "a" s.ioflush m.print "b\n" assert_equal("b\n", s.readpartial(10)) } end def test_ioflush2 helper {|m, s| s.print "a" s.ioflush # ioflush may be issued after "a" is already sent. s.print "b" assert_include(["b", "ab"], m.readpartial(10)) } end def test_winsize helper {|m, s| begin assert_equal([0, 0], s.winsize) rescue Errno::EINVAL # OpenSolaris 2009.06 TIOCGWINSZ causes Errno::EINVAL before TIOCSWINSZ. end } end if IO.console def test_close IO.console.close assert_kind_of(IO, IO.console) assert_nothing_raised(IOError) {IO.console.fileno} IO.console(:close) assert(IO.console(:tty?)) ensure IO.console(:close) end def test_sync assert(IO.console.sync, "console should be unbuffered") ensure IO.console(:close) end else def test_close assert_equal(["true"], run_pty("IO.console.close; p IO.console.fileno >= 0")) assert_equal(["true"], run_pty("IO.console(:close); p IO.console(:tty?)")) end def test_sync assert_equal(["true"], run_pty("p IO.console.sync")) end end private def helper m, s = PTY.open rescue RuntimeError skip $! else yield m, s ensure m.close if m s.close if s end def run_pty(src, n = 1) r, w, pid = PTY.spawn(EnvUtil.rubybin, "-rio/console", "-e", src) rescue RuntimeError skip $! else result = [] n.times {result << r.gets.chomp} Process.wait(pid) if block_given? yield result else result end ensure r.close if r w.close if w end end if defined?(PTY) and defined?(IO::console) class TestIO_Console < Test::Unit::TestCase case when Process.respond_to?(:daemon) noctty = [EnvUtil.rubybin, "-e", "Process.daemon(true)"] when !(rubyw = RbConfig::CONFIG["RUBYW_INSTALL_NAME"]).empty? dir, base = File.split(EnvUtil.rubybin) noctty = [File.join(dir, base.sub(/ruby/, rubyw))] end if noctty require 'tempfile' NOCTTY = noctty def test_noctty t = Tempfile.new("console") t.close t2 = Tempfile.new("console") t2.close cmd = [*NOCTTY[1..-1], '-e', 'open(ARGV[0], "w") {|f|', '-e', 'STDOUT.reopen(f)', '-e', 'STDERR.reopen(f)', '-e', 'require "io/console"', '-e', 'f.puts IO.console.inspect', '-e', 'f.flush', '-e', 'File.unlink(ARGV[1])', '-e', '}', '--', t.path, t2.path] assert_ruby_status(cmd, rubybin: NOCTTY[0]) 30.times do break unless File.exist?(t2.path) sleep 0.1 end t.open assert_equal("nil", t.gets(nil).chomp) ensure t.close! if t and !t.closed? t2.close! end end end if defined?(IO.console) class TestIO_Console < Test::Unit::TestCase def test_stringio_getch assert_separately %w"--disable=gems -rstringio -rio/console", %q{ assert_operator(StringIO, :method_defined?, :getch) } assert_separately %w"--disable=gems -rio/console -rstringio", %q{ assert_operator(StringIO, :method_defined?, :getch) } assert_separately %w"--disable=gems -rstringio", %q{ assert_not_operator(StringIO, :method_defined?, :getch) } end end