summaryrefslogtreecommitdiff
path: root/test/test_curses.rb
blob: 7bcb172e96b4fb80d8e3e62703e6600d5debff61 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
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)
    PTY.spawn({"TERM"=>ENV["TERM"]||"dumb"}, EnvUtil.rubybin, "-e", <<-"src") {|r, w, pid|
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
      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
    }
  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