зеркало из https://github.com/github/ruby.git
110 строки
2.2 KiB
Ruby
110 строки
2.2 KiB
Ruby
require 'test/unit'
|
|
require_relative 'ruby/envutil'
|
|
require 'shellwords'
|
|
|
|
begin
|
|
require 'pty'
|
|
rescue LoadError
|
|
end
|
|
|
|
class TestPTY < Test::Unit::TestCase
|
|
RUBY = EnvUtil.rubybin
|
|
|
|
def test_spawn_without_block
|
|
r, w, pid = PTY.spawn(RUBY, '-e', 'puts "a"')
|
|
assert_equal("a\r\n", r.gets)
|
|
ensure
|
|
Process.wait pid if pid
|
|
end
|
|
|
|
def test_spawn_with_block
|
|
PTY.spawn(RUBY, '-e', 'puts "b"') {|r,w,pid|
|
|
assert_equal("b\r\n", r.gets)
|
|
Process.wait(pid)
|
|
}
|
|
end
|
|
|
|
def test_commandline
|
|
commandline = Shellwords.join([RUBY, '-e', 'puts "foo"'])
|
|
PTY.spawn(commandline) {|r,w,pid|
|
|
assert_equal("foo\r\n", r.gets)
|
|
Process.wait(pid)
|
|
}
|
|
end
|
|
|
|
def test_argv0
|
|
PTY.spawn([RUBY, "argv0"], '-e', 'puts "bar"') {|r,w,pid|
|
|
assert_equal("bar\r\n", r.gets)
|
|
Process.wait(pid)
|
|
}
|
|
end
|
|
|
|
def test_open_without_block
|
|
ret = PTY.open
|
|
assert_kind_of(Array, ret)
|
|
assert_equal(2, ret.length)
|
|
assert_equal(IO, ret[0].class)
|
|
assert_equal(File, ret[1].class)
|
|
master, slave = ret
|
|
assert(slave.tty?)
|
|
assert(File.chardev?(slave.path))
|
|
ensure
|
|
if ret
|
|
ret[0].close
|
|
ret[1].close
|
|
end
|
|
end
|
|
|
|
def test_open_with_block
|
|
r = nil
|
|
x = Object.new
|
|
y = PTY.open {|ret|
|
|
r = ret;
|
|
assert_kind_of(Array, ret)
|
|
assert_equal(2, ret.length)
|
|
assert_equal(IO, ret[0].class)
|
|
assert_equal(File, ret[1].class)
|
|
master, slave = ret
|
|
assert(slave.tty?)
|
|
assert(File.chardev?(slave.path))
|
|
x
|
|
}
|
|
assert(r[0].closed?)
|
|
assert(r[1].closed?)
|
|
assert_equal(y, x)
|
|
end
|
|
|
|
def test_close_in_block
|
|
PTY.open {|master, slave|
|
|
slave.close
|
|
master.close
|
|
assert(slave.closed?)
|
|
assert(master.closed?)
|
|
}
|
|
assert_nothing_raised {
|
|
PTY.open {|master, slave|
|
|
slave.close
|
|
master.close
|
|
}
|
|
}
|
|
end
|
|
|
|
def test_open
|
|
PTY.open {|master, slave|
|
|
slave.puts "foo"
|
|
assert_equal("foo", master.gets.chomp)
|
|
master.puts "bar"
|
|
assert_equal("bar", slave.gets.chomp)
|
|
}
|
|
end
|
|
|
|
def test_stat_slave
|
|
PTY.open {|master, slave|
|
|
s = File.stat(slave.path)
|
|
assert_equal(Process.uid, s.uid)
|
|
assert_equal(0600, s.mode & 0777)
|
|
}
|
|
end
|
|
end if defined? PTY
|
|
|