зеркало из https://github.com/github/ruby.git
53 строки
1.4 KiB
Ruby
53 строки
1.4 KiB
Ruby
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
|