1 files changed, 49 insertions, 0 deletions
diff --git a/test/dtrace/test_cmethod.rb b/test/dtrace/test_cmethod.rb
new file mode 100644
index 0000000000..138ec55ab3
--- /dev/null
+++ b/test/dtrace/test_cmethod.rb
@@ -0,0 +1,49 @@
+require 'dtrace/helper'
+
+module DTrace
+  class TestCMethod < TestCase
+    def test_entry
+      probe = <<-eoprobe
+ruby$target:::cmethod-entry
+{
+  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[1] == 'times'
+	}
+
+	assert_equal 1, foo_calls.length
+      }
+    end
+
+    def test_exit
+      probe = <<-eoprobe
+ruby$target:::cmethod-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[1] == 'times'
+	}
+
+	assert_equal 1, foo_calls.length
+      }
+    end
+
+    def ruby_program
+      <<-eoruby
+      class Foo
+	def self.foo; end
+      end
+      10.times { Foo.foo }
+      eoruby
+    end
+  end
+end if defined?(DTrace::TestCase)
+