ruby-prof 0.16.2 → 1.1.0

data/test/marshal_test.rb

@@ -0,0 +1,119 @@
+#!/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_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.root?, method_2.root?)
+      assert_equal(method_1.excluded?, method_2.excluded?)
+
+      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.callers, method_2.callers)
+      verify_call_infos(method_1.callees, method_2.callees)
+
+      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.count, call_infos_2.count)
+    call_infos_1.count.times do |i|
+      call_info_1 = call_infos_1[i]
+      call_info_2 = call_infos_2[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.parent, call_info_2.parent)
+    assert_equal(call_info_1.target, call_info_2.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

@@ -1,26 +1,385 @@
-#!/usr/bin/env ruby
-# encoding: UTF-8
-
-require File.expand_path('../test_helper', __FILE__)
-
-class MeasureAllocationsTest < TestCase
-  def test_allocations_mode
-    RubyProf::measure_mode = RubyProf::ALLOCATIONS
-    assert_equal(RubyProf::ALLOCATIONS, RubyProf::measure_mode)
-  end
-
-  def test_allocations_enabled_defined
-    assert(defined?(RubyProf::ALLOCATIONS_ENABLED))
-  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
-    end
-  end
-end
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+require File.expand_path('../test_helper', __FILE__)
+require_relative './measure_allocations'
+
+class MeasureAllocationsTest < TestCase
+  def setup
+    RubyProf::measure_mode = RubyProf::ALLOCATIONS
+  end
+
+  def test_allocations_mode
+    RubyProf::measure_mode = RubyProf::ALLOCATIONS
+    assert_equal(RubyProf::ALLOCATIONS, RubyProf::measure_mode)
+  end
+
+  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)
+
+    root_methods = thread.root_methods
+    assert_equal(1, root_methods.count)
+    assert_equal("MeasureAllocationsTest#test_allocations", root_methods[0].full_name)
+
+    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(1, method.callers.length)
+    call_info = method.callers[0]
+    assert_nil(call_info.parent)
+    assert_equal(20, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(20, call_info.children_time)
+
+    assert_equal(2, method.callees.length)
+    call_info = method.callees[0]
+    assert_equal('Class#new', call_info.target.full_name)
+    assert_equal(1, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(1, call_info.self_time)
+    assert_equal(0, call_info.children_time)
+
+    call_info = method.callees[1]
+    assert_equal('Allocator#run', call_info.target.full_name)
+    assert_equal(19, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(19, call_info.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.callers.length)
+    call_info = method.callers[0]
+    assert_equal('MeasureAllocationsTest#test_allocations', call_info.parent.full_name)
+    assert_equal(19, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(19, call_info.children_time)
+
+    assert_equal(1, method.callees.length)
+    call_info = method.callees[0]
+    assert_equal('Allocator#internal_run', call_info.target.full_name)
+    assert_equal(19, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(19, call_info.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.callers.length)
+    call_info = method.callers[0]
+    assert_equal('Allocator#run', call_info.parent.full_name)
+    assert_equal(19, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(19, call_info.children_time)
+
+    assert_equal(3, method.callees.length)
+    call_info = method.callees[0]
+    assert_equal('Allocator#make_arrays', call_info.target.full_name)
+    assert_equal(10, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(10, call_info.children_time)
+
+    call_info = method.callees[1]
+    assert_equal('Allocator#make_hashes', call_info.target.full_name)
+    assert_equal(5, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(5, call_info.children_time)
+
+    call_info = method.callees[2]
+    assert_equal('Allocator#make_strings', call_info.target.full_name)
+    assert_equal(4, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(1, call_info.self_time)
+    assert_equal(3, call_info.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.callers.length)
+    call_info = method.callers[0]
+    assert_equal('MeasureAllocationsTest#test_allocations', call_info.parent.full_name)
+    assert_equal(1, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(1, call_info.self_time)
+    assert_equal(0, call_info.children_time)
+
+    call_info = method.callers[1]
+    assert_equal('Integer#times', call_info.parent.full_name)
+    assert_equal(10, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(10, call_info.self_time)
+    assert_equal(0, call_info.children_time)
+
+    call_info = method.callers[2]
+    assert_equal('Allocator#make_hashes', call_info.parent.full_name)
+    assert_equal(5, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(5, call_info.self_time)
+    assert_equal(0, call_info.children_time)
+
+    call_info = method.callers[3]
+    assert_equal('Allocator#make_strings', call_info.parent.full_name)
+    assert_equal(2, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(1, call_info.self_time)
+    assert_equal(1, call_info.children_time)
+
+    assert_equal(4, method.callees.length)
+    call_info = method.callees[0]
+    assert_equal('BasicObject#initialize', call_info.target.full_name)
+    assert_equal(0, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(0, call_info.children_time)
+
+    call_info = method.callees[1]
+    assert_equal('Array#initialize', call_info.target.full_name)
+    assert_equal(0, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(0, call_info.children_time)
+
+    call_info = method.callees[2]
+    assert_equal('Hash#initialize', call_info.target.full_name)
+    assert_equal(0, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(0, call_info.children_time)
+
+    call_info = method.callees[3]
+    assert_equal('String#initialize', call_info.target.full_name)
+    assert_equal(1, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(1, call_info.self_time)
+    assert_equal(0, call_info.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.callers.length)
+    call_info = method.callers[0]
+    assert_equal('Allocator#internal_run', call_info.parent.full_name)
+    assert_equal(10, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(10, call_info.children_time)
+
+    assert_equal(1, method.callees.length)
+    call_info = method.callees[0]
+    assert_equal('Integer#times', call_info.target.full_name)
+    assert_equal(10, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(10, call_info.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.callers.length)
+    call_info = method.callers[0]
+    assert_equal('Allocator#make_arrays', call_info.parent.full_name)
+    assert_equal(10, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(10, call_info.children_time)
+
+    assert_equal(1, method.callees.length)
+    call_info = method.callees[0]
+    assert_equal('Class#new', call_info.target.full_name)
+    assert_equal(10, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(10, call_info.self_time)
+    assert_equal(0, call_info.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.callers.length)
+    call_info = method.callers[0]
+    assert_equal('Allocator#internal_run', call_info.parent.full_name)
+    assert_equal(5, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(5, call_info.children_time)
+
+    assert_equal(1, method.callees.length)
+    call_info = method.callees[0]
+    assert_equal('Class#new', call_info.target.full_name)
+    assert_equal(5, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(5, call_info.self_time)
+    assert_equal(0, call_info.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.callers.length)
+    call_info = method.callers[0]
+    assert_equal('Allocator#internal_run', call_info.parent.full_name)
+    assert_equal(4, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(1, call_info.self_time)
+    assert_equal(3, call_info.children_time)
+
+    assert_equal(2, method.callees.length)
+    call_info = method.callees[0]
+    assert_equal('String#*', call_info.target.full_name)
+    assert_equal(1, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(1, call_info.self_time)
+    assert_equal(0, call_info.children_time)
+
+    call_info = method.callees[1]
+    assert_equal('Class#new', call_info.target.full_name)
+    assert_equal(2, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(1, call_info.self_time)
+    assert_equal(1, call_info.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.callers.length)
+    call_info = method.callers[0]
+    assert_equal('Allocator#make_strings', call_info.parent.full_name)
+    assert_equal(1, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(1, call_info.self_time)
+    assert_equal(0, call_info.children_time)
+
+    assert_equal(0, method.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.callers.length)
+    call_info = method.callers[0]
+    assert_equal('Class#new', call_info.parent.full_name)
+    assert_equal(1, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(1, call_info.self_time)
+    assert_equal(0, call_info.children_time)
+
+    assert_equal(0, method.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.callers.length)
+    call_info = method.callers[0]
+    assert_equal('Class#new', call_info.parent.full_name)
+    assert_equal(0, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(0, call_info.children_time)
+
+    assert_equal(0, method.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.callers.length)
+    call_info = method.callers[0]
+    assert_equal('Class#new', call_info.parent.full_name)
+    assert_equal(0, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(0, call_info.children_time)
+
+    assert_equal(0, method.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.callers.length)
+    call_info = method.callers[0]
+    assert_equal('Class#new', call_info.parent.full_name)
+    assert_equal(0, call_info.total_time)
+    assert_equal(0, call_info.wait_time)
+    assert_equal(0, call_info.self_time)
+    assert_equal(0, call_info.children_time)
+
+    assert_equal(0, method.callees.length)
+  end
+end