ruby-prof 0.18.0 → 1.2.0

data/test/marshal_test.rb

@@ -0,0 +1,115 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+require File.expand_path("../test_helper", __FILE__)
+require 'stringio'
+
+class MarshalTest < TestCase
+  def verify_profile(profile_1, profile_2)
+    verify_threads(profile_1.threads, profile_2.threads)
+  end
+
+  def verify_threads(threads_1, threads_2)
+    assert_equal(threads_1.count, threads_2.count)
+    threads_1.count.times do |i|
+      thread_1 = threads_1[i]
+      thread_2 = threads_2[i]
+      assert_nil(thread_2.id)
+      assert_equal(thread_1.fiber_id, thread_2.fiber_id)
+      verify_call_info(thread_1.call_tree, thread_2.call_tree)
+
+      verify_methods(thread_1.methods, thread_2.methods)
+    end
+  end
+
+  def verify_methods(methods_1, methods_2)
+    assert_equal(methods_1.count, methods_2.count)
+
+    methods_1.count.times do |i|
+      method_1 = methods_1[i]
+      method_2 = methods_2[i]
+
+      assert_equal(method_1.klass_name, method_2.klass_name)
+      assert_equal(method_1.klass_flags, method_2.klass_flags)
+
+      assert_equal(method_1.method_name, method_2.method_name)
+      assert_equal(method_1.full_name, method_2.full_name)
+
+      assert_equal(method_1.recursive?, method_2.recursive?)
+
+      assert_equal(method_1.source_file, method_2.source_file)
+      assert_equal(method_1.line, method_2.line)
+
+      verify_measurement(method_1.measurement, method_2.measurement)
+      verify_call_infos(method_1.call_trees, method_2.call_trees)
+      verify_allocations(method_1.allocations, method_2.allocations)
+    end
+  end
+
+  def verify_allocations(allocations_1, allocations_2)
+    assert_equal(allocations_1.count, allocations_2.count)
+
+    allocations_1.count.times do |i|
+      allocation_1 = allocations_1[i]
+      allocation_2 = allocations_2[i]
+
+      assert_equal(allocation_1.klass_name, allocation_2.klass_name)
+      assert_equal(allocation_1.klass_flags, allocation_2.klass_flags)
+
+      assert_equal(allocation_1.count, allocation_2.count)
+      assert_equal(allocation_1.memory, allocation_2.memory)
+
+      assert_equal(allocation_1.source_file, allocation_2.source_file)
+      assert_equal(allocation_1.line, allocation_2.line)
+    end
+  end
+
+  def verify_call_infos(call_infos_1, call_infos_2)
+    assert_equal(call_infos_1.call_trees.count, call_infos_2.call_trees.count)
+    call_infos_1.call_trees.count.times do |i|
+      call_info_1 = call_infos_1.call_trees[i]
+      call_info_2 = call_infos_2.call_trees[i]
+      verify_call_info(call_info_1, call_info_2)
+    end
+  end
+
+  def verify_call_info(call_info_1, call_info_2)
+    assert_equal(call_info_1.target, call_info_2.target)
+    assert_equal(call_info_1.parent&.target, call_info_2.parent&.target)
+
+    assert_equal(call_info_1.depth, call_info_2.depth)
+    assert_equal(call_info_1.source_file, call_info_2.source_file)
+    assert_equal(call_info_1.line, call_info_2.line)
+
+    verify_measurement(call_info_1.measurement, call_info_2.measurement)
+  end
+
+  def verify_measurement(measurement_1, measurement_2)
+    assert_equal(measurement_1.total_time, measurement_2.total_time)
+    assert_equal(measurement_1.self_time, measurement_2.self_time)
+    assert_equal(measurement_1.wait_time, measurement_2.wait_time)
+    assert_equal(measurement_1.called, measurement_2.called)
+  end
+
+  def test_marshal
+    profile_1 = RubyProf.profile do
+      1.times { RubyProf::C1.new.sleep_wait }
+    end
+
+    data = Marshal.dump(profile_1)
+    profile_2 = Marshal.load(data)
+
+    verify_profile(profile_1, profile_2)
+  end
+
+  def test_singleton
+    profile_1 = RubyProf.profile do
+      SingletonTest.instance.busy_wait
+    end
+
+    data = Marshal.dump(profile_1)
+    profile_2 = Marshal.load(data)
+
+    verify_profile(profile_1, profile_2)
+  end
+end

data/test/measure_allocations.rb

@@ -0,0 +1,30 @@
+# Some classes used in measurement tests
+class Allocator
+  def make_arrays
+    10.times {|i| Array.new}
+  end
+
+  def make_hashes
+    Hash.new
+    Hash.new
+    Hash.new
+    Hash.new
+    Hash.new
+  end
+
+  def make_strings
+    a_string = 'a'
+    b_string = a_string * 100
+    String.new(b_string)
+  end
+
+  def internal_run
+    make_arrays
+    make_hashes
+    make_strings
+  end
+
+  def run
+    internal_run
+  end
+end

data/test/measure_allocations_test.rb

@@ -2,25 +2,374 @@
 # encoding: UTF-8
 
 require File.expand_path('../test_helper', __FILE__)
+require_relative './measure_allocations'
 
 class MeasureAllocationsTest < TestCase
-  def test_allocations_mode
+  def setup
     RubyProf::measure_mode = RubyProf::ALLOCATIONS
-    assert_equal(RubyProf::ALLOCATIONS, RubyProf::measure_mode)
   end
 
-  def test_allocations_enabled_defined
-    assert(defined?(RubyProf::ALLOCATIONS_ENABLED))
+  def test_allocations_mode
+    RubyProf::measure_mode = RubyProf::ALLOCATIONS
+    assert_equal(RubyProf::ALLOCATIONS, RubyProf::measure_mode)
   end
 
-  if RubyProf::ALLOCATIONS_ENABLED
-    def test_allocations
-      t = RubyProf.measure_allocations
-      refute_empty("a" + "b")
-      u = RubyProf.measure_allocations
-      assert_kind_of Integer, t
-      assert_kind_of Integer, u
-      assert_operator t, :<, u
+  def test_allocations
+    result = RubyProf.profile do
+      allocator = Allocator.new
+      allocator.run
     end
+
+    thread = result.threads.first
+    assert_in_delta(20, thread.total_time, 1)
+
+    methods = result.threads.first.methods.sort.reverse
+    assert_equal(13, methods.length)
+
+    # Method 0
+    method = methods[0]
+    assert_equal('MeasureAllocationsTest#test_allocations',  method.full_name)
+    assert_in_delta(20, method.total_time, 1)
+    assert_equal(0, method.wait_time)
+    assert_equal(0, method.self_time)
+    assert_in_delta(20, method.children_time, 1)
+
+    assert_equal(0, method.call_trees.callers.length)
+
+    assert_equal(2, method.call_trees.callees.length)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('Class#new', call_tree.target.full_name)
+    assert_equal(1, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(1, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    call_tree = method.call_trees.callees[1]
+    assert_equal('Allocator#run', call_tree.target.full_name)
+    assert_equal(19, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(19, call_tree.children_time)
+
+    # Method 1
+    method = methods[1]
+    assert_equal('Allocator#run',method.full_name)
+    assert_equal(19, method.total_time)
+    assert_equal(0, method.wait_time)
+    assert_equal(0, method.self_time)
+    assert_equal(19, method.children_time)
+
+    assert_equal(1, method.call_trees.callers.length)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('MeasureAllocationsTest#test_allocations', call_tree.parent.target.full_name)
+    assert_equal(19, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(19, call_tree.children_time)
+
+    assert_equal(1, method.call_trees.callees.length)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('Allocator#internal_run', call_tree.target.full_name)
+    assert_equal(19, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(19, call_tree.children_time)
+
+    # Method 2
+    method = methods[2]
+    assert_equal('Allocator#internal_run',  method.full_name)
+    assert_equal(19, method.total_time)
+    assert_equal(0, method.wait_time)
+    assert_equal(0, method.self_time)
+    assert_equal(19, method.children_time)
+
+    assert_equal(1, method.call_trees.callers.length)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('Allocator#run', call_tree.parent.target.full_name)
+    assert_equal(19, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(19, call_tree.children_time)
+
+    assert_equal(3, method.call_trees.callees.length)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('Allocator#make_arrays', call_tree.target.full_name)
+    assert_equal(10, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(10, call_tree.children_time)
+
+    call_tree = method.call_trees.callees[1]
+    assert_equal('Allocator#make_hashes', call_tree.target.full_name)
+    assert_equal(5, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(5, call_tree.children_time)
+
+    call_tree = method.call_trees.callees[2]
+    assert_equal('Allocator#make_strings', call_tree.target.full_name)
+    assert_equal(4, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(1, call_tree.self_time)
+    assert_equal(3, call_tree.children_time)
+
+    # Method 3
+    method = methods[3]
+    assert_equal('Class#new', method.full_name)
+    assert_equal(18, method.total_time)
+    assert_equal(0, method.wait_time)
+    assert_equal(17, method.self_time)
+    assert_equal(1, method.children_time)
+
+    assert_equal(4, method.call_trees.callers.length)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('MeasureAllocationsTest#test_allocations', call_tree.parent.target.full_name)
+    assert_equal(1, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(1, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    call_tree = method.call_trees.callers[1]
+    assert_equal('Integer#times', call_tree.parent.target.full_name)
+    assert_equal(10, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(10, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    call_tree = method.call_trees.callers[2]
+    assert_equal('Allocator#make_hashes', call_tree.parent.target.full_name)
+    assert_equal(5, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(5, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    call_tree = method.call_trees.callers[3]
+    assert_equal('Allocator#make_strings', call_tree.parent.target.full_name)
+    assert_equal(2, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(1, call_tree.self_time)
+    assert_equal(1, call_tree.children_time)
+
+    assert_equal(4, method.call_trees.callees.length)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('BasicObject#initialize', call_tree.target.full_name)
+    assert_equal(0, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    call_tree = method.call_trees.callees[1]
+    assert_equal('Array#initialize', call_tree.target.full_name)
+    assert_equal(0, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    call_tree = method.call_trees.callees[2]
+    assert_equal('Hash#initialize', call_tree.target.full_name)
+    assert_equal(0, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    call_tree = method.call_trees.callees[3]
+    assert_equal('String#initialize', call_tree.target.full_name)
+    assert_equal(1, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(1, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    # Method 4
+    method = methods[4]
+    assert_equal('Allocator#make_arrays', method.full_name)
+    assert_equal(10, method.total_time)
+    assert_equal(0, method.wait_time)
+    assert_equal(0, method.self_time)
+    assert_equal(10, method.children_time)
+
+    assert_equal(1, method.call_trees.callers.length)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('Allocator#internal_run', call_tree.parent.target.full_name)
+    assert_equal(10, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(10, call_tree.children_time)
+
+    assert_equal(1, method.call_trees.callees.length)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('Integer#times', call_tree.target.full_name)
+    assert_equal(10, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(10, call_tree.children_time)
+
+    # Method 5
+    method = methods[5]
+    assert_equal('Integer#times', method.full_name)
+    assert_equal(10, method.total_time)
+    assert_equal(0, method.wait_time)
+    assert_equal(0, method.self_time)
+    assert_equal(10, method.children_time)
+
+    assert_equal(1, method.call_trees.callers.length)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('Allocator#make_arrays', call_tree.parent.target.full_name)
+    assert_equal(10, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(10, call_tree.children_time)
+
+    assert_equal(1, method.call_trees.callees.length)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('Class#new', call_tree.target.full_name)
+    assert_equal(10, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(10, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    # Method 6
+    method = methods[6]
+    assert_equal('Allocator#make_hashes', method.full_name)
+    assert_equal(5, method.total_time)
+    assert_equal(0, method.wait_time)
+    assert_equal(0, method.self_time)
+    assert_equal(5, method.children_time)
+
+    assert_equal(1, method.call_trees.callers.length)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('Allocator#internal_run', call_tree.parent.target.full_name)
+    assert_equal(5, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(5, call_tree.children_time)
+
+    assert_equal(1, method.call_trees.callees.length)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('Class#new', call_tree.target.full_name)
+    assert_equal(5, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(5, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    # Method 7
+    method = methods[7]
+    assert_equal('Allocator#make_strings', method.full_name)
+    assert_equal(4, method.total_time)
+    assert_equal(0, method.wait_time)
+    assert_equal(1, method.self_time)
+    assert_equal(3, method.children_time)
+
+    assert_equal(1, method.call_trees.callers.length)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('Allocator#internal_run', call_tree.parent.target.full_name)
+    assert_equal(4, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(1, call_tree.self_time)
+    assert_equal(3, call_tree.children_time)
+
+    assert_equal(2, method.call_trees.callees.length)
+    call_tree = method.call_trees.callees[0]
+    assert_equal('String#*', call_tree.target.full_name)
+    assert_equal(1, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(1, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    call_tree = method.call_trees.callees[1]
+    assert_equal('Class#new', call_tree.target.full_name)
+    assert_equal(2, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(1, call_tree.self_time)
+    assert_equal(1, call_tree.children_time)
+
+    # Method 8
+    method = methods[8]
+    assert_equal('String#*', method.full_name)
+    assert_equal(1, method.total_time)
+    assert_equal(0, method.wait_time)
+    assert_equal(1, method.self_time)
+    assert_equal(0, method.children_time)
+
+    assert_equal(1, method.call_trees.callers.length)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('Allocator#make_strings', call_tree.parent.target.full_name)
+    assert_equal(1, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(1, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    assert_equal(0, method.call_trees.callees.length)
+
+    # Method 9
+    method = methods[9]
+    assert_equal('String#initialize', method.full_name)
+    assert_equal(1, method.total_time)
+    assert_equal(0, method.wait_time)
+    assert_equal(1, method.self_time)
+    assert_equal(0, method.children_time)
+
+    assert_equal(1, method.call_trees.callers.length)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('Class#new', call_tree.parent.target.full_name)
+    assert_equal(1, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(1, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    assert_equal(0, method.call_trees.callees.length)
+
+    # Method 10
+    method = methods[10]
+    assert_equal('BasicObject#initialize', method.full_name)
+    assert_equal(0, method.total_time)
+    assert_equal(0, method.wait_time)
+    assert_equal(0, method.self_time)
+    assert_equal(0, method.children_time)
+
+    assert_equal(1, method.call_trees.callers.length)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('Class#new', call_tree.parent.target.full_name)
+    assert_equal(0, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    assert_equal(0, method.call_trees.callees.length)
+
+    # Method 11
+    method = methods[11]
+    assert_equal('Hash#initialize', method.full_name)
+    assert_equal(0, method.total_time)
+    assert_equal(0, method.wait_time)
+    assert_equal(0, method.self_time)
+    assert_equal(0, method.children_time)
+
+    assert_equal(1, method.call_trees.callers.length)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('Class#new', call_tree.parent.target.full_name)
+    assert_equal(0, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    assert_equal(0, method.call_trees.callees.length)
+
+    # Method 12
+    method = methods[12]
+    assert_equal('Array#initialize', method.full_name)
+    assert_equal(0, method.total_time)
+    assert_equal(0, method.wait_time)
+    assert_equal(0, method.self_time)
+    assert_equal(0, method.children_time)
+
+    assert_equal(1, method.call_trees.callers.length)
+    call_tree = method.call_trees.callers[0]
+    assert_equal('Class#new', call_tree.parent.target.full_name)
+    assert_equal(0, call_tree.total_time)
+    assert_equal(0, call_tree.wait_time)
+    assert_equal(0, call_tree.self_time)
+    assert_equal(0, call_tree.children_time)
+
+    assert_equal(0, method.call_trees.callees.length)
   end
 end