зеркало из https://github.com/github/ruby.git
75 строки
1.6 KiB
Ruby
75 строки
1.6 KiB
Ruby
require 'test/unit'
|
|
require_relative 'ruby/envutil'
|
|
|
|
begin
|
|
require 'curses'
|
|
require 'pty'
|
|
rescue LoadError
|
|
end
|
|
|
|
class TestCurses < Test::Unit::TestCase
|
|
def test_version
|
|
assert_instance_of(String, Curses::VERSION)
|
|
end
|
|
end if defined? Curses
|
|
|
|
class TestCurses
|
|
def run_curses(src, input = nil, timeout: 1)
|
|
r, w, pid = PTY.spawn({"TERM"=>ENV["TERM"]||"dumb"}, EnvUtil.rubybin, "-e", <<-"src")
|
|
require 'timeout'
|
|
require 'curses'
|
|
include Curses
|
|
init_screen
|
|
begin
|
|
result = Timeout.timeout(#{timeout}) do
|
|
print "!"
|
|
#{src}
|
|
end
|
|
rescue Exception => e
|
|
ensure
|
|
close_screen
|
|
puts "", [Marshal.dump([result, e])].pack('m').delete("\n")
|
|
print "\\0"
|
|
$stdio.flush
|
|
end
|
|
src
|
|
begin
|
|
wait = r.readpartial(1)
|
|
if wait != "!"
|
|
wait << r.readpartial(1000)
|
|
raise wait
|
|
end
|
|
if input
|
|
w.print(input)
|
|
w.flush
|
|
end
|
|
res = r.gets("\0")
|
|
return unless res
|
|
res.chomp!("\0")
|
|
res, error = Marshal.load(res[/(.*)\Z/, 1].unpack('m')[0])
|
|
raise error if error
|
|
return res
|
|
ensure
|
|
r.close unless r.closed?
|
|
w.close unless w.closed?
|
|
Process.wait(pid)
|
|
end
|
|
end
|
|
|
|
def test_getch
|
|
assert_equal("a", run_curses("getch", "a"))
|
|
end
|
|
def test_getch_cbreak
|
|
assert_equal("a", run_curses("cbreak; getch", "a"))
|
|
end
|
|
def test_getch_nocbreak
|
|
assert_raise(Timeout::Error) {run_curses("nocbreak; getch", "a")}
|
|
end
|
|
def test_getch_crmode
|
|
assert_equal("a", run_curses("crmode; getch", "a"))
|
|
end
|
|
def test_getch_nocrmode
|
|
assert_raise(Timeout::Error) {run_curses("nocrmode; getch", "a")}
|
|
end
|
|
end if defined? TestCurses and defined? PTY
|