From 32623a16276b8c99a8c142c570cb2f50dc3df737 Mon Sep 17 00:00:00 2001
From: ko1 <ko1@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Fri, 29 Oct 2010 14:35:53 +0000
Subject: * test/profile_test_all.rb: added.   You can use test-all profiler
 with the following command:   RUBY_TEST_ALL_PROFILE=true make test-all   This
 command generates ./test_all_profile and you can analyse   which tests
 consume memories. * test/runner.rb: ditto.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@29627 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 test/profile_test_all.rb | 52 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 test/profile_test_all.rb

(limited to 'test/profile_test_all.rb')

diff --git a/test/profile_test_all.rb b/test/profile_test_all.rb
new file mode 100644
index 0000000000..52eaf364aa
--- /dev/null
+++ b/test/profile_test_all.rb
@@ -0,0 +1,52 @@
+require 'objspace'
+
+#
+# purpose:
+#  Profile memory usage of each tests.
+#
+# usage:
+#   RUBY_TEST_ALL_PROFILE=true make test-all
+#
+# output:
+#   ./test_all_profile
+#
+# collected information:
+#   - ObjectSpace.memsize_of_all
+#   - GC.stat
+#   - /proc/self/statm (if it exists)
+#
+
+class MiniTest::Unit::TestCase
+  alias orig_run run
+
+  $test_all_profile_out = open('test_all_profile', 'w')
+  $test_all_profile_gc_stat_hash = {}
+
+  if FileTest.exist?('/proc/self/statm')
+    # for Linux (only?)
+    $test_all_profile_out.puts "name\tmemsize_of_all\t" +
+                                 (GC.stat.keys +
+                                  %w(size resident share text lib data dt)).join("\t")
+
+    def memprofile_test_all_result_result
+      "#{self.class}\##{self.__name__}\t" \
+      "#{ObjectSpace.memsize_of_all}\t" \
+      "#{GC.stat($test_all_profile_gc_stat_hash).values.join("\t")}\t" \
+      "#{File.read('/proc/self/statm').split(/\s+/).join("\t")}"
+    end
+  else
+    $test_all_profile_out.puts "name\tmemsize_of_alls\t" + GC.stat.keys.join("\t")
+    def memprofile_test_all_result_result
+      "#{self.class}\##{self.__name__}\t" \
+      "#{ObjectSpace.memsize_of_all}\t" \
+      "#{GC.stat($test_all_profile_gc_stat_hash).values.join("\t")}"
+    end
+  end
+
+  def run runner
+    result = orig_run(runner)
+    $test_all_profile_out.puts memprofile_test_all_result_result
+    $test_all_profile_out.flush
+    result
+  end
+end
-- 
cgit v1.2.3