ruby-prof 0.18.0 → 1.2.0

data/test/gc_test.rb

@@ -0,0 +1,105 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+require File.expand_path('../test_helper', __FILE__)
+
+class GcTest < TestCase
+  def setup
+    GC.stress = true
+  end
+
+  def teardown
+    GC.stress = false
+  end
+
+  def some_method
+    Array.new(3 * 4)
+  end
+
+  def run_profile
+    RubyProf.profile do
+      self.some_method
+    end
+  end
+
+  def test_hold_onto_thread
+    threads = 5.times.reduce(Array.new) do |array, i|
+      array.concat(run_profile.threads)
+      array
+    end
+
+    threads.each do |thread|
+      error = assert_raises(RuntimeError) do
+        thread.id
+      end
+      assert_match(/has already been freed/, error.message)
+    end
+    assert(true)
+  end
+
+  def test_hold_onto_root_call_info
+    call_trees = 5.times.reduce(Array.new) do |array, i|
+      array.concat(run_profile.threads.map(&:call_tree))
+      array
+    end
+
+    call_trees.each do |call_tree|
+      error = assert_raises(RuntimeError) do
+        call_tree.source_file
+      end
+      assert_match(/has already been freed/, error.message)
+    end
+    assert(true)
+  end
+
+  def test_hold_onto_method
+    methods = 5.times.reduce(Array.new) do |array, i|
+      profile = run_profile
+      methods_2 = profile.threads.map(&:methods).flatten
+      array.concat(methods_2)
+      array
+    end
+
+    methods.each do |method|
+      error = assert_raises(RuntimeError) do
+        method.method_name
+      end
+      assert_match(/has already been freed/, error.message)
+    end
+    assert(true)
+  end
+
+  def test_hold_onto_call_trees
+    method_call_infos = 5.times.reduce(Array.new) do |array, i|
+      profile = run_profile
+      call_trees = profile.threads.map(&:methods).flatten.map(&:call_trees).flatten
+      array.concat(call_trees)
+      array
+    end
+
+    method_call_infos.each do |call_trees|
+      error = assert_raises(RuntimeError) do
+        call_trees.call_trees
+      end
+      assert_match(/has already been freed/, error.message)
+    end
+    assert(true)
+  end
+
+  def test_hold_onto_measurements
+    measurements = 5.times.reduce(Array.new) do |array, i|
+      profile = run_profile
+      measurements = profile.threads.map(&:methods).flatten.map(&:measurement)
+      array.concat(measurements)
+      array
+    end
+
+    measurements.each do |measurement|
+      error = assert_raises(RuntimeError) do
+        measurement.total_time
+      end
+      assert_match(/has already been freed/, error.message)
+    end
+    assert(true)
+  end
+end

data/test/inverse_call_tree_test.rb

@@ -0,0 +1,175 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+require File.expand_path('../test_helper', __FILE__)
+
+class InverseCallTreeTest < TestCase
+  INVERSE_DEPTH = 5
+
+  def setup
+    # Need to use wall time for this test due to the sleep calls
+    RubyProf::measure_mode = RubyProf::WALL_TIME
+  end
+
+  INVERSE_DEPTH.times do |i|
+    if i == 0
+      define_method("method_#{i}") do
+        sleep_amount = (i + 1) * 0.05
+        RubyProf.start
+        sleep(sleep_amount)
+      end
+    else
+      define_method("method_#{i}") do
+        method_name = "method_#{i-1}"
+        sleep_amount = (i + 1) * 0.05
+        self.send(method_name.to_sym)
+        sleep(sleep_amount)
+      end
+    end
+  end
+
+  def test_inverse
+    method_name = "method_#{INVERSE_DEPTH - 1}"
+    self.send(method_name.to_sym)
+    result = profile = RubyProf.stop
+
+    assert_equal(1, profile.threads.count)
+
+    thread = profile.threads.first
+    assert_in_delta(0.79, thread.total_time, 0.05)
+
+    assert_equal(7, thread.methods.length)
+    methods = thread.methods.sort.reverse
+
+    # InverseCallTreeTest#test_inverse
+    method = methods[0]
+    assert_equal('InverseCallTreeTest#test_inverse', method.full_name)
+    assert_equal(34, method.line)
+
+    assert_equal(0, method.call_trees.callers.count)
+
+    assert_equal(1, method.call_trees.callees.count)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('InverseCallTreeTest#method_4', call_tree.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    # InverseCallTreeTest#method_4
+    method = methods[1]
+    assert_equal('InverseCallTreeTest#method_4', method.full_name)
+    assert_equal(26, method.line)
+
+    assert_equal(1, method.call_trees.callers.count)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('InverseCallTreeTest#test_inverse', call_tree.parent.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    assert_equal(2, method.call_trees.callees.count)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('InverseCallTreeTest#method_3', call_tree.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    # Kernel#sleep
+    method = methods[2]
+    assert_equal('Kernel#sleep', method.full_name)
+    assert_equal(0, method.line)
+
+    assert_equal(5, method.call_trees.callers.count)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('InverseCallTreeTest#method_0', call_tree.parent.target.full_name)
+    assert_equal(19, call_tree.line)
+
+    call_tree = method.call_trees.callers[1]
+    assert_equal('InverseCallTreeTest#method_1', call_tree.parent.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    call_tree = method.call_trees.callers[2]
+    assert_equal('InverseCallTreeTest#method_2', call_tree.parent.target.full_name)
+    assert_equal(26, call_tree.line)
+    call_tree = method.call_trees.callers[3]
+
+    assert_equal('InverseCallTreeTest#method_3', call_tree.parent.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    call_tree = method.call_trees.callers[4]
+    assert_equal('InverseCallTreeTest#method_4', call_tree.parent.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    assert_equal(0, method.call_trees.callees.count)
+
+    # InverseCallTreeTest#method_3
+    method = methods[3]
+    assert_equal('InverseCallTreeTest#method_3', method.full_name)
+    assert_equal(26, method.line)
+
+    assert_equal(1, method.call_trees.callers.count)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('InverseCallTreeTest#method_4', call_tree.parent.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    assert_equal(2, method.call_trees.callees.count)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('InverseCallTreeTest#method_2', call_tree.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    call_tree = method.call_trees.callees[1]
+    assert_equal('Kernel#sleep', call_tree.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    # InverseCallTreeTest#method_2
+    method = methods[4]
+    assert_equal('InverseCallTreeTest#method_2', method.full_name)
+    assert_equal(26, method.line)
+
+    assert_equal(1, method.call_trees.callers.count)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('InverseCallTreeTest#method_3', call_tree.parent.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    assert_equal(2, method.call_trees.callees.count)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('InverseCallTreeTest#method_1', call_tree.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    call_tree = method.call_trees.callees[1]
+    assert_equal('Kernel#sleep', call_tree.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    call_tree = method.call_trees.callees[1]
+    assert_equal('Kernel#sleep', call_tree.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    # InverseCallTreeTest#method_1
+    method = methods[5]
+    assert_equal('InverseCallTreeTest#method_1', method.full_name)
+    assert_equal(26, method.line)
+
+    assert_equal(1, method.call_trees.callers.count)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('InverseCallTreeTest#method_2', call_tree.parent.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    assert_equal(2, method.call_trees.callees.count)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('InverseCallTreeTest#method_0', call_tree.target.full_name)
+    assert_equal(19, call_tree.line)
+
+    call_tree = method.call_trees.callees[1]
+    assert_equal('Kernel#sleep', call_tree.target.full_name)
+    assert_equal(26, call_tree.line)
+
+    # InverseCallTreeTest#method_0
+    method = methods[6]
+    assert_equal('InverseCallTreeTest#method_0', method.full_name)
+    assert_equal(19, method.line)
+
+    assert_equal(1, method.call_trees.callers.count)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('InverseCallTreeTest#method_1', call_tree.parent.target.full_name)
+    assert_equal(19, call_tree.line)
+
+    assert_equal(1, method.call_trees.callees.count)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('Kernel#sleep', call_tree.target.full_name)
+    assert_equal(19, call_tree.line)
+  end
+end

data/test/line_number_test.rb

@@ -4,15 +4,27 @@
 require File.expand_path('../test_helper', __FILE__)
 
 class LineNumbers
-  def method1
+  def method_1
+    method_2
+    filler = 1
+    method_3
   end
 
-  def method2
-    method1
+  def method_2
+    filler = 1
+    2.times do |i|
+      filler = 2
+      method_3
+    end
+  end
+
+  def method_3
+    sleep(0.3)
+    method_4
   end
 
-  def method3
-    sleep(1)
+  def method_4
+    sleep(1.2)
   end
 end
 
@@ -22,59 +34,125 @@ class LineNumbersTest < TestCase
     numbers = LineNumbers.new
 
     result = RubyProf.profile do
-      numbers.method2
+      numbers.method_1
     end
 
     methods = result.threads.first.methods.sort.reverse
-    assert_equal(3, methods.length)
+    assert_equal(7, methods.length)
 
+    # Method 0
     method = methods[0]
     assert_equal('LineNumbersTest#test_function_line_no', method.full_name)
-    assert_equal(25, method.line)
+    assert_equal(37, method.line)
 
+    assert_equal(0, method.call_trees.callers.count)
+
+    assert_equal(1, method.call_trees.callees.count)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('LineNumbers#method_1', call_tree.target.full_name)
+    assert_equal(37, call_tree.line)
+
+    # Method 1
     method = methods[1]
-    assert_equal('LineNumbers#method2', method.full_name)
-    if RUBY_VERSION < "2.5"
-      assert_equal(10, method.line)
-    else
-      assert_equal(11, method.line)
-    end
+    assert_equal('LineNumbers#method_1', method.full_name)
+    assert_equal(7, method.line)
+
+    assert_equal(1, method.call_trees.callers.count)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('LineNumbersTest#test_function_line_no', call_tree.parent.target.full_name)
+    assert_equal(37, call_tree.line)
+
+    assert_equal(2, method.call_trees.callees.count)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('LineNumbers#method_2', call_tree.target.full_name)
+    assert_equal(8, call_tree.line)
 
+    call_tree = method.call_trees.callees[1]
+    assert_equal('LineNumbers#method_3', call_tree.target.full_name)
+    assert_equal(10, call_tree.line)
+
+    # Method 2
     method = methods[2]
-    assert_equal('LineNumbers#method1', method.full_name)
-    assert_equal(7, method.line)
-  end
+    assert_equal('LineNumbers#method_3', method.full_name)
+    assert_equal(21, method.line)
 
-  def test_c_function
-    numbers = LineNumbers.new
+    assert_equal(2, method.call_trees.callers.count)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('Integer#times', call_tree.parent.target.full_name)
+    assert_equal(17, call_tree.line)
 
-    result = RubyProf.profile do
-      numbers.method3
-    end
+    call_tree = method.call_trees.callers[1]
+    assert_equal('LineNumbers#method_1', call_tree.parent.target.full_name)
+    assert_equal(10, call_tree.line)
 
-    methods = result.threads.first.methods.sort_by {|method| method.full_name}
-    assert_equal(3, methods.length)
+    assert_equal(2, method.call_trees.callees.count)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('Kernel#sleep', call_tree.target.full_name)
+    assert_equal(22, call_tree.line)
 
-    # Methods:
-    #   LineNumbers#method3
-    #   LineNumbersTest#test_c_function
-    #   Kernel#sleep
+    call_tree = method.call_trees.callees[1]
+    assert_equal('LineNumbers#method_4', call_tree.target.full_name)
+    assert_equal(23, call_tree.line)
 
-    method = methods[0]
+    # Method 3
+    method = methods[3]
     assert_equal('Kernel#sleep', method.full_name)
     assert_equal(0, method.line)
 
-    method = methods[1]
-    assert_equal('LineNumbers#method3', method.full_name)
-    if RUBY_VERSION < "2.5"
-      assert_equal(14, method.line)
-    else
-      assert_equal(15, method.line)
-    end
+    assert_equal(2, method.call_trees.callers.count)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('LineNumbers#method_3', call_tree.parent.target.full_name)
+    assert_equal(22, call_tree.line)
+
+    call_tree = method.call_trees.callers[1]
+    assert_equal('LineNumbers#method_4', call_tree.parent.target.full_name)
+    assert_equal(27, call_tree.line)
+
+    assert_equal(0, method.call_trees.callees.count)
+
+    # Method 4
+    method = methods[4]
+    assert_equal('LineNumbers#method_4', method.full_name)
+    assert_equal(26, method.line)
+
+    assert_equal(1, method.call_trees.callers.count)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('LineNumbers#method_3', call_tree.parent.target.full_name)
+    assert_equal(23, call_tree.line)
+
+    assert_equal(1, method.call_trees.callees.count)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('Kernel#sleep', call_tree.target.full_name)
+    assert_equal(27, call_tree.line)
+
+    # Method 5
+    method = methods[5]
+    assert_equal('LineNumbers#method_2', method.full_name)
+    assert_equal(13, method.line)
+
+    assert_equal(1, method.call_trees.callers.count)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('LineNumbers#method_1', call_tree.parent.target.full_name)
+    assert_equal(8, call_tree.line)
+
+    assert_equal(1, method.call_trees.callees.count)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('Integer#times', call_tree.target.full_name)
+    assert_equal(15, call_tree.line)
+
+    # Method 6
+    method = methods[6]
+    assert_equal('Integer#times', method.full_name)
+    assert_equal(0, method.line)
 
-    method = methods[2]
-    assert_equal('LineNumbersTest#test_c_function', method.full_name)
-    assert_equal(52, method.line)
-  end
+    assert_equal(1, method.call_trees.callers.count)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('LineNumbers#method_2', call_tree.parent.target.full_name)
+    assert_equal(15, call_tree.line)
 
+    assert_equal(1, method.call_trees.callees.count)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('LineNumbers#method_3', call_tree.target.full_name)
+    assert_equal(17, call_tree.line)
+  end
 end