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
92
93
94
95
96
97
98
99
100
101
102
103
104
|
#!/usr/bin/env ruby
# encoding: UTF-8
require File.expand_path('../test_helper', __FILE__)
# Test data
# A
# / \
# B C
# \
# B
class MSTPT
def a
100.times{b}
300.times{c}
c;c;c
end
def b
sleep 0
end
def c
5.times{b}
end
end
class MultiPrinterTest < TestCase
def setup
# Need to use wall time for this test due to the sleep calls
RubyProf::measure_mode = RubyProf::WALL_TIME
end
def test_refuses_io_objects
# we don't need a real profile for this test
p = RubyProf::MultiPrinter.new nil
begin
p.print(STDOUT)
flunk "should have raised an ArgumentError"
rescue ArgumentError => e
assert_match(/IO/, e.to_s)
end
end
def test_refuses_non_hashes
# we don't need a real profile for this test
p = RubyProf::MultiPrinter.new nil
begin
p.print([])
flunk "should have raised an ArgumentError"
rescue ArgumentError => e
assert_match(/hash/, e.to_s)
end
end
def test_all_profiles_can_be_created
start_time = Time.now
RubyProf.start
5.times{MSTPT.new.a}
result = RubyProf.stop
end_time = Time.now
expected_time = end_time - start_time
graph = print(result)[1]
re = Regexp.new('
\s*<table>
\s*<tr>
\s*<th>Thread ID</th>
\s*(<th>Fiber ID</th>)?
\s*<th>Total Time</th>
\s*</tr>
\s*
\s*<tr>
\s*(<td>([\.0-9]+)</td>)?
\s*<td><a href="#-?\d+">-?\d+</a></td>
\s*<td>([\.0-9e]+)</td>
\s*</tr>
\s*
\s*</table>')
assert_match(re, graph)
graph =~ re
display_time = $4.to_f
assert_in_delta expected_time, display_time, 0.001
end
private
def print(result)
test = caller.first =~ /in `(.*)'/ ? $1 : "test"
path = RubyProf.tmpdir
profile = "ruby_prof_#{test}"
printer = RubyProf::MultiPrinter.new(result)
printer.print(:path => path, :profile => profile,
:threshold => 0, :min_percent => 0, :title => "ruby_prof #{test}")
if RUBY_PLATFORM =~ /darwin/ && ENV['SHOW_RUBY_PROF_PRINTER_OUTPUT']=="1"
system("open '#{printer.stack_profile}'")
end
# if GC.respond_to?(:dump_file_and_line_info)
# GC.start
# GC.dump_file_and_line_info("heap.dump")
# end
[File.read(printer.stack_profile), File.read(printer.graph_profile)]
end
end
|
