89 lines
1.9 KiB
Ruby
89 lines
1.9 KiB
Ruby
# frozen_string_literal: false
|
|
require_relative 'helper'
|
|
|
|
module DTrace
|
|
class TestFunctionEntry < TestCase
|
|
def test_function_entry
|
|
probe = <<-eoprobe
|
|
ruby$target:::method-entry
|
|
/arg0 && arg1 && arg2/
|
|
{
|
|
printf("%s %s %s %d\\n", copyinstr(arg0), copyinstr(arg1), copyinstr(arg2), arg3);
|
|
}
|
|
eoprobe
|
|
|
|
trap_probe(probe, ruby_program) { |d_file, rb_file, probes|
|
|
foo_calls = probes.map { |line| line.split }.find_all { |row|
|
|
row.first == 'Foo' && row[1] == 'foo'
|
|
}
|
|
|
|
assert_equal 10, foo_calls.length
|
|
line = '2'
|
|
foo_calls.each { |f| assert_equal line, f[3] }
|
|
foo_calls.each { |f| assert_equal rb_file, f[2] }
|
|
}
|
|
end
|
|
|
|
def test_function_return
|
|
probe = <<-eoprobe
|
|
ruby$target:::method-return
|
|
/arg0 && arg1 && arg2/
|
|
{
|
|
printf("%s %s %s %d\\n", copyinstr(arg0), copyinstr(arg1), copyinstr(arg2), arg3);
|
|
}
|
|
eoprobe
|
|
|
|
trap_probe(probe, ruby_program) { |d_file, rb_file, probes|
|
|
foo_calls = probes.map { |line| line.split }.find_all { |row|
|
|
row.first == 'Foo' && row[1] == 'foo'
|
|
}
|
|
|
|
assert_equal 10, foo_calls.length
|
|
line = '2'
|
|
foo_calls.each { |f| assert_equal line, f[3] }
|
|
foo_calls.each { |f| assert_equal rb_file, f[2] }
|
|
}
|
|
end
|
|
|
|
def test_return_from_raise
|
|
program = <<-eoruby
|
|
class Foo
|
|
def bar; raise; end
|
|
def baz
|
|
bar
|
|
rescue
|
|
end
|
|
end
|
|
|
|
Foo.new.baz
|
|
eoruby
|
|
|
|
probe = <<-eoprobe
|
|
ruby$target:::method-return
|
|
/arg0 && arg1 && arg2/
|
|
{
|
|
printf("%s %s %s %d\\n", copyinstr(arg0), copyinstr(arg1), copyinstr(arg2), arg3);
|
|
}
|
|
eoprobe
|
|
|
|
trap_probe(probe, program) { |d_file, rb_file, probes|
|
|
foo_calls = probes.map { |line| line.split }.find_all { |row|
|
|
row.first == 'Foo' && row[1] == 'bar'
|
|
}
|
|
assert foo_calls.any?
|
|
}
|
|
end
|
|
|
|
private
|
|
def ruby_program
|
|
<<-eoruby
|
|
class Foo
|
|
def foo; end
|
|
end
|
|
x = Foo.new
|
|
10.times { x.foo }
|
|
eoruby
|
|
end
|
|
end
|
|
end if defined?(DTrace::TestCase)
|