test/ruby/test_trace.rb


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

require 'test/unit'

class TestTrace < Test::Unit::TestCase
  def test_trace
    $x = 1234
    $y = 0
    trace_var :$x, proc{$y = $x}
    $x = 40414
    assert_equal($x, $y)

    untrace_var :$x
    $x = 19660208
    assert_not_equal($x, $y)

    trace_var :$x, proc{$x *= 2}
    $x = 5
    assert_equal(10, $x)

    untrace_var :$x
  end

  def test_trace_tainted_proc
    $x = 1234
    s = proc { $y = :foo }
    trace_var :$x, s
    s.taint
    $x = 42
    assert_equal(:foo, $y)
  ensure
    untrace_var :$x
  end

  def test_trace_proc_that_raises_exception
    $x = 1234
    trace_var :$x, proc { raise }
    assert_raise(RuntimeError) { $x = 42 }
  ensure
    untrace_var :$x
  end

  def test_trace_string
    $x = 1234
    trace_var :$x, "$y = :bar"
    $x = 42
    assert_equal(:bar, $y)
  ensure
    untrace_var :$x
  end

  def test_trace_break
    bug2722 = '[ruby-core:31783]'
    a = Object.new.extend(Enumerable)
    def a.each
      yield
    end
    assert(Thread.start {
             Thread.current.add_trace_func(proc{})
             a.any? {true}
           }.value, bug2722)
  end
end