summaryrefslogtreecommitdiff
path: root/spec/ruby/shared/kernel/at_exit.rb
blob: 16d41cb01c93b180d1d39da670f5968fd0845645 (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
70
describe :kernel_at_exit, shared: true do
  it "runs after all other code" do
    ruby_exe("#{@method} { print 5 }; print 6").should == "65"
  end

  it "runs in reverse order of registration" do
    code = "#{@method} { print 4 }; #{@method} { print 5 }; print 6; #{@method} { print 7 }"
    ruby_exe(code).should == "6754"
  end

  it "allows calling exit inside a handler" do
    code = "#{@method} { print 3 }; #{@method} { print 4; exit; print 5 }; #{@method} { print 6 }"
    ruby_exe(code).should == "643"
  end

  it "gives access to the last raised exception - global variables $! and $@" do
    code = <<-EOC
      #{@method} {
        puts "The exception matches: \#{$! == $exception && $@ == $exception.backtrace} (message=\#{$!.message})"
      }

      begin
        raise "foo"
      rescue => $exception
        raise
      end
    EOC

    result = ruby_exe(code, args: "2>&1", exit_status: 1)
    result.lines.should.include?("The exception matches: true (message=foo)\n")
  end

  it "both exceptions in a handler and in the main script are printed" do
    code = "#{@method} { raise 'at_exit_error' }; raise 'main_script_error'"
    result = ruby_exe(code, args: "2>&1", exit_status: 1)
    result.should.include?('at_exit_error (RuntimeError)')
    result.should.include?('main_script_error (RuntimeError)')
  end

  it "decides the exit status if both at_exit and the main script raise SystemExit" do
    ruby_exe("#{@method} { exit 43 }; exit 42", args: "2>&1", exit_status: 43)
    $?.exitstatus.should == 43
  end

  it "runs all handlers even if some raise exceptions" do
    code = "#{@method} { STDERR.puts 'last' }; #{@method} { exit 43 }; #{@method} { STDERR.puts 'first' }; exit 42"
    result = ruby_exe(code, args: "2>&1", exit_status: 43)
    result.should == "first\nlast\n"
    $?.exitstatus.should == 43
  end

  it "runs handlers even if the main script fails to parse" do
    script = fixture(__FILE__, "#{@method}.rb")
    result = ruby_exe('{', options: "-r#{script}", args: "2>&1", exit_status: 1)
    $?.should_not.success?
    result.should.include?("handler ran\n")

    # it's tempting not to rely on error message and rely only on exception class name,
    # but CRuby before 3.2 doesn't print class name for syntax error
    result.should include_any_of("syntax error", "SyntaxError")
  end

  it "calls the nested handler right after the outer one if a handler is nested into another handler" do
    ruby_exe(<<~ruby).should == "last\nbefore\nafter\nnested\nfirst\n"
        #{@method} { puts :first }
        #{@method} { puts :before; #{@method} { puts :nested }; puts :after };
        #{@method} { puts :last }
    ruby
  end
end