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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
#!/usr/bin/env ruby
# encoding: UTF-8
require File.expand_path('../test_helper', __FILE__)
class FakeRackApp
def call(env)
end
end
module Rack
class Request
def initialize(env)
if env == :fake_env
@env = {}
else
@env = env
end
end
def path
@env[:path] || '/path/to/resource.json'
end
end
end
module Rack
class RubyProf
attr_reader :_profiler
def public_delete_profiler!
delete_profiler!
end
end
end
class RackTest < TestCase
def test_create_print_path
path = Dir.mktmpdir
Dir.delete(path)
Rack::RubyProf.new(FakeRackApp.new, :path => path)
assert(Dir.exist?(path))
end
def test_create_profile_reports
path = Dir.mktmpdir
adapter = Rack::RubyProf.new(FakeRackApp.new, :path => path)
adapter.call(:fake_env)
%w(flat.txt graph.txt graph.html call_stack.html).each do |base_name|
file_path = ::File.join(path, "path-to-resource.json-#{base_name}")
assert(File.exist?(file_path))
end
end
def test_skip_paths
path = Dir.mktmpdir
adapter = Rack::RubyProf.new(FakeRackApp.new, :path => path, :skip_paths => [%r{\.json$}])
adapter.call(:fake_env)
%w(flat.txt graph.txt graph.html call_stack.html).each do |base_name|
file_path = ::File.join(path, "path-to-resource.json-#{base_name}")
assert(!File.exist?(file_path))
end
end
def test_only_paths
path = Dir.mktmpdir
adapter = Rack::RubyProf.new(FakeRackApp.new, :path => path, :only_paths => [%r{\.json$}])
adapter.call({path: '/path/to/resource.json'})
%w(flat.txt graph.txt graph.html call_stack.html).each do |base_name|
file_path = ::File.join(path, "path-to-resource.json-#{base_name}")
assert(File.exist?(file_path))
end
adapter.call({path: '/path/to/resource.html'})
%w(flat.txt graph.txt graph.html call_stack.html).each do |base_name|
file_path = ::File.join(path, "path-to-resource.html-#{base_name}")
assert(!File.exist?(file_path))
end
end
def test_allows_lazy_filename_setting
path = Dir.mktmpdir
printer = {::RubyProf::FlatPrinter => lambda { 'dynamic.txt' }}
adapter = Rack::RubyProf.new(FakeRackApp.new, :path => path, :printers => printer)
adapter.call(:fake_env)
file_path = ::File.join(path, 'path-to-resource.json-dynamic.txt')
assert(File.exist?(file_path))
end
def test_works_for_multiple_requests
path = Dir.mktmpdir
adapter = Rack::RubyProf.new(FakeRackApp.new, :path => path, :max_requests => 2)
# make a 1st request, and check that this didn't create any files
adapter.call(:fake_env)
assert(Dir["#{path}/*"].empty?)
# now a second request should create all the expected files
adapter.call(:fake_env)
%w(flat.txt graph.txt graph.html call_stack.html).each do |base_name|
file_path = ::File.join(path, "multi-requests-2-#{base_name}")
assert(File.exist?(file_path))
end
# let's clean up
FileUtils.rm_rf(Dir["#{path}/*"])
# and do the same again for the next 2 requests
adapter.call(:fake_env)
assert(Dir["#{path}/*"].empty?)
adapter.call(:fake_env)
%w(flat.txt graph.txt graph.html call_stack.html).each do |base_name|
file_path = ::File.join(path, "multi-requests-2-#{base_name}")
assert(File.exist?(file_path))
end
end
def test_tries_to_print_results_if_shut_down_before_max_requests_reached
path = Dir.mktmpdir
adapter = Rack::RubyProf.new(FakeRackApp.new, :path => path, :max_requests => 100)
# make a 1st request, and check that this didn't create any files
adapter.call(:fake_env)
assert(Dir["#{path}/*"].empty?)
adapter.public_delete_profiler!
%w(flat.txt graph.txt graph.html call_stack.html).each do |base_name|
file_path = ::File.join(path, "multi-requests-1-#{base_name}")
assert(File.exist?(file_path))
end
end
def test_it_uses_separate_profilers_if_not_aggregating_multiple_requests
adapter1 = Rack::RubyProf.new(FakeRackApp.new)
adapter2 = Rack::RubyProf.new(FakeRackApp.new)
assert(adapter1.object_id != adapter2.object_id)
end
end
|
