forked from rack/rack-contrib
-
Notifications
You must be signed in to change notification settings - Fork 0
/
spec_rack_profiler.rb
36 lines (28 loc) · 1.45 KB
/
spec_rack_profiler.rb
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
require 'rack/mock'
require 'rack/profiler'
context 'Rack::Profiler' do
specify 'printer defaults to RubyProf::CallTreePrinter' do
profiler = Rack::Profiler.new(nil)
profiler.instance_variable_get('@printer').should == RubyProf::CallTreePrinter
end
specify 'CallTreePrinter has correct headers' do
app = lambda { |env| [200, {'Content-Type' => 'text/plain'}, 'Oh hai der'] }
request = Rack::MockRequest.env_for("/", :input => "profile=process_time")
headers = Rack::Profiler.new(app).call(request)[1]
headers.should == {"Content-Disposition"=>"attachment; filename=\"/.process_time.tree\")", "Content-Type"=>"application/octet-stream"}
end
specify 'FlatPrinter and GraphPrinter has Content-Type text/plain' do
app = lambda { |env| [200, {'Content-Type' => 'text/plain'}, 'Oh hai der'] }
request = Rack::MockRequest.env_for("/", :input => "profile=process_time")
%w(flat graph).each do |printer|
headers = Rack::Profiler.new(app, :printer => printer.to_sym).call(request)[1]
headers.should == {"Content-Type"=>"text/plain"}
end
end
specify 'GraphHtmlPrinter has Content-Type text/html' do
app = lambda { |env| [200, {'Content-Type' => 'text/plain'}, 'Oh hai der'] }
request = Rack::MockRequest.env_for("/", :input => "profile=process_time")
headers = Rack::Profiler.new(app, :printer => :graphhtml).call(request)[1]
headers.should == {"Content-Type"=>"text/html"}
end
end