Fresh start

1 files changed, 332 insertions, 0 deletions

diff --git a/jni/ruby/test/io/console/test_io_console.rb b/jni/ruby/test/io/console/test_io_console.rb
new file mode 100644
index 0000000..3481a2b
--- /dev/null
+++ b/jni/ruby/test/io/console/test_io_console.rb
@@ -0,0 +1,332 @@
+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