summaryrefslogtreecommitdiff
path: root/test/dtrace/test_singleton_function.rb
blob: 3698a02c93eae8fc3cb74cf7d06ab6409a6844f2 (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
# frozen_string_literal: false
require_relative 'helper'

module DTrace
  class TestSingletonFunctionEntry < TestCase
    def test_entry
      probe = <<-eoprobe
ruby$target:::method-entry
/strstr(copyinstr(arg0), "Foo") != NULL/
{
  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_exit
      probe = <<-eoprobe
ruby$target:::method-return
{
  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 ruby_program
      <<-eoruby
      class Foo
	def self.foo; end
      end
      10.times { Foo.foo }
      eoruby
    end
  end
end if defined?(DTrace::TestCase)