# Copyright 2013 Google Inc. All Rights Reserved. # Licensed under the Apache License, Version 2.0, found in the LICENSE file. require_relative "../spec_helper" describe Rack::AllocationStats do before do @app = HelloWorldApp.new @vanilla_request_env = Rack::MockRequest.env_for("/") @traced_request_env = Rack::MockRequest.env_for("/", :params => "ras[trace]=true") end it "returns something when called with a tracing request" do status, headers, _ = Rack::AllocationStats.new(@app).call(@traced_request_env) expect(headers["Content-Type"]).to eq "text/plain" expect(status).to be 200 end it "does not interfere when not a trace request" do _, _, body = Rack::AllocationStats.new(@app).call(@vanilla_request_env) expect(body).to eq ["Hello Rack!"] end it "deletes ras[] params before hitting next app" do allocation_stats = Rack::AllocationStats.new(@app) @app.should_receive(:call).with(query_string({})) status, headers, _ = allocation_stats.call(@traced_request_env) end it "returns the correct body when called with a tracing request" do _, _, body = Rack::AllocationStats.new(@app).call(@traced_request_env) expected_body = [ "#{HelloWorldApp::FULL_PATH}:#{@app.allocating_lines[1]} allocated 4 `String`\n", "#{HelloWorldApp::FULL_PATH}:#{@app.allocating_lines[1]} allocated 1 `Array`\n", "#{HelloWorldApp::FULL_PATH}:#{@app.allocating_lines[1]} allocated 1 `Array`\n", "#{HelloWorldApp::FULL_PATH}:#{@app.allocating_lines[1]} allocated 1 `Hash`\n", "#{HelloWorldApp::FULL_PATH}:#{@app.allocating_lines[0]} allocated 1 `String`\n" ] expect(body).to include(expected_body[0]) expect(body).to include(expected_body[1]) expect(body).to include(expected_body[2]) expect(body).to include(expected_body[3]) expect(body).to include(expected_body[4]) end it "returns the correct body when called with alias_paths" do request_env = Rack::MockRequest.env_for("/", :params => "ras[trace]=true&ras[alias_paths]=true") _, _, body = Rack::AllocationStats.new(@app).call(request_env) expected_body = [ "/spec/hello_world_app.rb:#{@app.allocating_lines[1]} allocated 4 `String`\n", "/spec/hello_world_app.rb:#{@app.allocating_lines[1]} allocated 1 `Array`\n", "/spec/hello_world_app.rb:#{@app.allocating_lines[1]} allocated 1 `Array`\n", "/spec/hello_world_app.rb:#{@app.allocating_lines[1]} allocated 1 `Hash`\n", "/spec/hello_world_app.rb:#{@app.allocating_lines[0]} allocated 1 `String`\n" ] expect(body).to include(expected_body[0]) expect(body).to include(expected_body[1]) expect(body).to include(expected_body[2]) expect(body).to include(expected_body[3]) expect(body).to include(expected_body[4]) end it "returns correct body when called with a tracing request with times" do request_env = Rack::MockRequest.env_for("/", :params => "ras[trace]=true&ras[times]=4") _, _, body = Rack::AllocationStats.new(@app).call(request_env) expected_body = [ "#{HelloWorldApp::FULL_PATH}:#{@app.allocating_lines[1]} allocated 16 `String`\n", "#{HelloWorldApp::FULL_PATH}:#{@app.allocating_lines[1]} allocated 4 `Array`\n", "#{HelloWorldApp::FULL_PATH}:#{@app.allocating_lines[1]} allocated 4 `Array`\n", "#{HelloWorldApp::FULL_PATH}:#{@app.allocating_lines[1]} allocated 4 `Hash`\n", "#{HelloWorldApp::FULL_PATH}:#{@app.allocating_lines[0]} allocated 4 `String`\n" ] expect(body).to include(expected_body[0]) expect(body).to include(expected_body[1]) expect(body).to include(expected_body[2]) expect(body).to include(expected_body[3]) expect(body).to include(expected_body[4]) end it "returns correct body when called on just local files" do yaml_app = YamlApp.new request_env = Rack::MockRequest.env_for("/", :params => "ras[trace]=true&ras[scope]=.") _, _, body = Rack::AllocationStats.new(yaml_app).call(request_env) # should be: # yaml_app.rb:15 allocated 4 `String` # yaml_app.rb:14 allocated 2 `String` # yaml_app.rb:14 allocated 1 `Array<>` # yaml_app.rb:14 allocated 1 `Array` # yaml_app.rb:15 allocated 1 `Array` # yaml_app.rb:15 allocated 1 `Array` # yaml_app.rb:15 allocated 1 `Hash` expect(body.size).to eq 7 end it "returns the correct body when called on a specific directory" do yaml_app = YamlApp.new psych_request_env = Rack::MockRequest.env_for("/", :params => "ras[trace]=true&ras[scope]=psych") _, _, body = Rack::AllocationStats.new(yaml_app).call(psych_request_env) expect(body.size).to eq 44 end it "returns HTML5 in response to an interactive request" do interactive_request_env = Rack::MockRequest.env_for("/", :params => "ras[trace]=true&ras[output]=interactive") _, _, body = Rack::AllocationStats.new(@app).call(interactive_request_env) expect(body[0]).to match /^/ end it "returns the corrent Content-Length in the headers" do file = "/foo/bar.rb" line = 7 # TODO 5 comes from factories.rb; it should come from here. # TODO String comes from factories.rb; it should come from here. expected_body = "#{file}:#{line} allocated 5 `String`\n" stats = FactoryGirl.build(:stats, files: [file], sourceline: line) AllocationStats.stub(:trace) { stats } allocation_stats = Rack::AllocationStats.new(@app) _, headers, _ = allocation_stats.call(@traced_request_env) headers["Content-Length"].to_i.should be expected_body.length end context("scoping") do before do stats = FactoryGirl.build(:stats, files: ["/foo/bar.rb", File.join(Dir.pwd, "baz.rb")]) AllocationStats.stub(:trace) { stats } @allocation_stats = Rack::AllocationStats.new(@app) end it "returns all allocations when no scope is specified" do _, _, body = @allocation_stats.call(@traced_request_env) text = body.join text.should include("bar.rb") text.should include("baz.rb") end it "returns locally-scoped allocations" do request_env = Rack::MockRequest.env_for("/", :params => "ras[trace]=true&ras[scope]=.") _, _, body = @allocation_stats.call(request_env) text = body.join text.should include("baz.rb") text.should_not include("bar.rb") end it "returns allocations limited by scope" do request_env = Rack::MockRequest.env_for("/", :params => "ras[trace]=true&ras[scope]=foo") _, _, body = @allocation_stats.call(request_env) text = body.join text.should include("bar.rb") text.should_not include("baz.rb") end end it "returns JSON when res[output]=json" do stats = FactoryGirl.build(:stats, files: ["/foo/bar.rb", File.join(Dir.pwd, "baz.rb")]) AllocationStats.stub(:trace) { stats } @allocation_stats = Rack::AllocationStats.new(@app) json_request_env = Rack::MockRequest.env_for("/", :params => "ras[trace]=true&ras[output]=json") _, _, body = @allocation_stats.call(json_request_env) allocation = stats.allocations.all.first first_group_key = ["/foo/bar.rb", 7, String] text = body.join text.should include(allocation.to_json) text.should include(first_group_key.to_json) hash = {group_key: first_group_key, allocations: [allocation, allocation, allocation]} text.should include(Yajl::Encoder.encode(hash)) end it "returns the right Content-Type and status for JSON" do json_request_env = Rack::MockRequest.env_for("/", :params => "ras[trace]=true&ras[output]=json") status, headers, _ = Rack::AllocationStats.new(@app).call(json_request_env) expect(headers["Content-Type"]).to eq "application/json" expect(status).to be 200 end it "returns the right Content-Type and status for help" do help_request_env = Rack::MockRequest.env_for("/", :params => "ras[help]") status, headers, _ = Rack::AllocationStats.new(@app).call(help_request_env) expect(headers["Content-Type"]).to eq "text/plain" expect(status).to be 200 end it "returns the right body for help" do help_request_env = Rack::MockRequest.env_for("/", :params => "ras[help]") _, _, body = Rack::AllocationStats.new(@app).call(help_request_env) body.first.should include("Rack AllocationStats help\n") end end