1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
# frozen_string_literal: true
#
# purpose:
# Profile memory usage of each tests.
#
# usage:
# RUBY_TEST_ALL_PROFILE=[file] make test-all
#
# output:
# [file] specified by RUBY_TEST_ALL_PROFILE
# If [file] is 'true', then it is ./test_all_profile
#
# collected information:
# - ObjectSpace.memsize_of_all
# - GC.stat
# - /proc/meminfo (some fields, if exists)
# - /proc/self/status (some fields, if exists)
# - /proc/self/statm (if exists)
#
require 'objspace'
class Test::Unit::TestCase
alias orig_run run
file = ENV['RUBY_TEST_ALL_PROFILE']
file = 'test-all-profile-result' if file == 'true'
TEST_ALL_PROFILE_OUT = open(file, 'w')
TEST_ALL_PROFILE_GC_STAT_HASH = {}
TEST_ALL_PROFILE_BANNER = ['name']
TEST_ALL_PROFILE_PROCS = []
def self.add *name, &b
TEST_ALL_PROFILE_BANNER.concat name
TEST_ALL_PROFILE_PROCS << b
end
add 'failed?' do |result, tc|
result << (tc.passed? ? 0 : 1)
end
add 'memsize_of_all' do |result, *|
result << ObjectSpace.memsize_of_all
end
add(*GC.stat.keys) do |result, *|
GC.stat(TEST_ALL_PROFILE_GC_STAT_HASH)
result.concat TEST_ALL_PROFILE_GC_STAT_HASH.values
end
def self.add_proc_meminfo file, fields
return unless FileTest.exist?(file)
regexp = /(#{fields.join("|")}):\s*(\d+) kB/
# check = {}; fields.each{|e| check[e] = true}
add(*fields) do |result, *|
text = File.read(file)
text.scan(regexp){
# check.delete $1
result << $2
''
}
# raise check.inspect unless check.empty?
end
end
add_proc_meminfo '/proc/meminfo', %w(MemTotal MemFree)
add_proc_meminfo '/proc/self/status', %w(VmPeak VmSize VmHWM VmRSS)
if FileTest.exist?('/proc/self/statm')
add 'size', 'resident', 'share', 'text', 'lib', 'data', 'dt' do |result, *|
result.concat File.read('/proc/self/statm').split(/\s+/)
end
end
def memprofile_test_all_result_result
result = ["#{self.class}\##{self.__name__.to_s.gsub(/\s+/, '')}"]
TEST_ALL_PROFILE_PROCS.each{|proc|
proc.call(result, self)
}
result.join("\t")
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
TEST_ALL_PROFILE_OUT.puts TEST_ALL_PROFILE_BANNER.join("\t")
end
|
