corntrace-rack-contrib 1.0.2

data/test/spec_rack_deflect.rb

@@ -0,0 +1,107 @@
+require 'test/spec'
+require 'rack/mock'
+require 'rack/contrib/deflect'
+
+context "Rack::Deflect" do
+
+  setup do
+    @app = lambda { |env| [200, { 'Content-Type' => 'text/plain' }, ['cookies']] }
+    @mock_addr_1 = '111.111.111.111'
+    @mock_addr_2 = '222.222.222.222'
+    @mock_addr_3 = '333.333.333.333'
+  end
+
+  def mock_env remote_addr, path = '/'
+    Rack::MockRequest.env_for path, { 'REMOTE_ADDR' => remote_addr }
+  end
+
+  def mock_deflect options = {}
+    Rack::Deflect.new @app, options
+  end
+
+  specify "should allow regular requests to follow through" do
+    app = mock_deflect
+    status, headers, body = app.call mock_env(@mock_addr_1)
+    status.should.equal 200
+    body.should.equal ['cookies']
+  end
+
+  specify "should deflect requests exceeding the request threshold" do
+    log = StringIO.new
+    app = mock_deflect :request_threshold => 5, :interval => 10, :block_duration => 10, :log => log
+    env = mock_env @mock_addr_1
+
+    # First 5 should be fine
+    5.times do
+      status, headers, body = app.call env
+      status.should.equal 200
+      body.should.equal ['cookies']
+    end
+
+    # Remaining requests should fail for 10 seconds
+    10.times do
+      status, headers, body = app.call env
+      status.should.equal 403
+      body.should.equal []
+    end
+
+    # Log should reflect that we have blocked an address
+    log.string.should.match(/^deflect\(\d+\/\d+\/\d+\): blocked 111.111.111.111\n/)
+  end
+
+  specify "should expire blocking" do
+    log = StringIO.new
+    app = mock_deflect :request_threshold => 5, :interval => 2, :block_duration => 2, :log => log
+    env = mock_env @mock_addr_1
+
+    # First 5 should be fine
+    5.times do
+      status, headers, body = app.call env
+      status.should.equal 200
+      body.should.equal ['cookies']
+    end
+
+    # Exceeds request threshold
+    status, headers, body = app.call env
+    status.should.equal 403
+    body.should.equal []
+
+    # Allow block to expire
+    sleep 3
+
+    # Another 5 is fine now
+    5.times do
+      status, headers, body = app.call env
+      status.should.equal 200
+      body.should.equal ['cookies']
+    end
+
+    # Log should reflect block and release
+    log.string.should.match(/deflect.*: blocked 111\.111\.111\.111\ndeflect.*: released 111\.111\.111\.111\n/)
+  end
+
+  specify "should allow whitelisting of remote addresses" do
+    app = mock_deflect :whitelist => [@mock_addr_1], :request_threshold => 5, :interval => 2
+    env = mock_env @mock_addr_1
+
+    # Whitelisted addresses are always fine
+    10.times do
+      status, headers, body = app.call env
+      status.should.equal 200
+      body.should.equal ['cookies']
+    end
+  end
+
+  specify "should allow blacklisting of remote addresses" do
+    app = mock_deflect :blacklist => [@mock_addr_2]
+
+    status, headers, body = app.call mock_env(@mock_addr_1)
+    status.should.equal 200
+    body.should.equal ['cookies']
+
+    status, headers, body = app.call mock_env(@mock_addr_2)
+    status.should.equal 403
+    body.should.equal []
+  end
+
+end

data/test/spec_rack_evil.rb

@@ -0,0 +1,19 @@
+require 'test/spec'
+require 'rack/mock'
+require 'rack/contrib/evil'
+require 'erb'
+
+context "Rack::Evil" do
+  app = lambda do |env|
+    template = ERB.new("<%= throw :response, [404, {'Content-Type' => 'text/html'}, 'Never know where it comes from'] %>")
+    [200, {'Content-Type' => 'text/plain'}, template.result(binding)]
+  end
+
+  specify "should enable the app to return the response from anywhere" do
+    status, headers, body = Rack::Evil.new(app).call({})
+
+    status.should.equal 404
+    headers['Content-Type'].should.equal 'text/html'
+    body.should.equal 'Never know where it comes from'
+  end
+end

data/test/spec_rack_expectation_cascade.rb

@@ -0,0 +1,72 @@
+require 'test/spec'
+require 'rack/mock'
+require 'rack/contrib/expectation_cascade'
+
+context "Rack::ExpectationCascade" do
+  specify "with no apps returns a 404 if no expectation header was set" do
+    app = Rack::ExpectationCascade.new
+    env = {}
+    response = app.call(env)
+    response[0].should.equal 404
+    env.should.equal({})
+  end
+
+  specify "with no apps returns a 417 if expectation header was set" do
+    app = Rack::ExpectationCascade.new
+    env = {"Expect" => "100-continue"}
+    response = app.call(env)
+    response[0].should.equal 417
+    env.should.equal({"Expect" => "100-continue"})
+  end
+
+  specify "returns first successful response" do
+    app = Rack::ExpectationCascade.new do |cascade|
+      cascade << lambda { |env| [417, {"Content-Type" => "text/plain"}, []] }
+      cascade << lambda { |env| [200, {"Content-Type" => "text/plain"}, ["OK"]] }
+    end
+    response = app.call({})
+    response[0].should.equal 200
+    response[2][0].should.equal "OK"
+  end
+
+  specify "expectation is set if it has not been already" do
+    app = Rack::ExpectationCascade.new do |cascade|
+      cascade << lambda { |env| [200, {"Content-Type" => "text/plain"}, ["Expect: #{env["Expect"]}"]] }
+    end
+    response = app.call({})
+    response[0].should.equal 200
+    response[2][0].should.equal "Expect: 100-continue"
+  end
+
+  specify "returns a 404 if no apps where matched and no expectation header was set" do
+    app = Rack::ExpectationCascade.new do |cascade|
+      cascade << lambda { |env| [417, {"Content-Type" => "text/plain"}, []] }
+    end
+    response = app.call({})
+    response[0].should.equal 404
+    response[2][0].should.equal nil
+  end
+
+  specify "returns a 417 if no apps where matched and a expectation header was set" do
+    app = Rack::ExpectationCascade.new do |cascade|
+      cascade << lambda { |env| [417, {"Content-Type" => "text/plain"}, []] }
+    end
+    response = app.call({"Expect" => "100-continue"})
+    response[0].should.equal 417
+    response[2][0].should.equal nil
+  end
+
+  specify "nests expectation cascades" do
+    app = Rack::ExpectationCascade.new do |c1|
+      c1 << Rack::ExpectationCascade.new do |c2|
+        c2 << lambda { |env| [417, {"Content-Type" => "text/plain"}, []] }
+      end
+      c1 << Rack::ExpectationCascade.new do |c2|
+        c2 << lambda { |env| [200, {"Content-Type" => "text/plain"}, ["OK"]] }
+      end
+    end
+    response = app.call({})
+    response[0].should.equal 200
+    response[2][0].should.equal "OK"
+  end
+end

data/test/spec_rack_garbagecollector.rb

@@ -0,0 +1,13 @@
+require 'test/spec'
+require 'rack/mock'
+require 'rack/contrib/garbagecollector'
+
+context 'Rack::GarbageCollector' do
+
+  specify 'starts the garbage collector after each request' do
+    app = lambda { |env|
+      [200, {'Content-Type'=>'text/plain'}, ['Hello World']] }
+    Rack::GarbageCollector.new(app).call({})
+  end
+
+end

data/test/spec_rack_host_meta.rb

@@ -0,0 +1,50 @@
+require 'test/spec'
+require 'rack/mock'
+require 'rack/contrib/host_meta'
+require 'rack/contrib/not_found'
+
+context "Rack::HostMeta" do
+
+  setup do
+    app = Rack::Builder.new do
+      use Rack::Lint
+      use Rack::ContentLength
+      use Rack::HostMeta do
+        link :uri => '/robots.txt', :rel => 'robots'
+        link :uri => '/w3c/p3p.xml', :rel => 'privacy', :type => 'application/p3p.xml'
+        link :pattern => '{uri};json_schema', :rel => 'describedby', :type => 'application/x-schema+json'
+      end
+      run Rack::NotFound.new('test/404.html')
+    end
+    @response = Rack::MockRequest.new(app).get('/host-meta')
+  end
+
+  specify "should respond to /host-meta" do
+    @response.status.should.equal 200
+  end
+
+  specify "should respond with the correct media type" do
+    @response['Content-Type'].should.equal 'application/host-meta'
+  end
+
+  specify "should include a Link entry for each Link item in the config block" do
+    @response.body.should.match(/Link:\s*<\/robots.txt>;.*\n/)
+    @response.body.should.match(/Link:\s*<\/w3c\/p3p.xml>;.*/)
+  end
+
+  specify "should include a Link-Pattern entry for each Link-Pattern item in the config" do
+    @response.body.should.match(/Link-Pattern:\s*<\{uri\};json_schema>;.*/)
+  end
+
+  specify "should include a rel attribute for each Link or Link-Pattern entry where specified" do
+    @response.body.should.match(/rel="robots"/)
+    @response.body.should.match(/rel="privacy"/)
+    @response.body.should.match(/rel="describedby"/)
+  end
+
+  specify "should include a type attribute for each Link or Link-Pattern entry where specified" do
+    @response.body.should.match(/Link:\s*<\/w3c\/p3p.xml>;.*type.*application\/p3p.xml/)
+    @response.body.should.match(/Link-Pattern:\s*<\{uri\};json_schema>;.*type.*application\/x-schema\+json/)
+  end
+
+end

data/test/spec_rack_jsonp.rb

@@ -0,0 +1,83 @@
+require 'test/spec'
+require 'rack/mock'
+# require File.expand_path(File.dirname(__FILE__)+"/../lib/rack/contrib/jsonp")
+require 'rack/contrib/jsonp'
+
+context "Rack::JSONP" do
+
+  context "when a callback parameter is provided" do
+    specify "should wrap the response body in the Javascript callback if JSON" do
+      test_body = '{"bar":"foo"}'
+      callback = 'foo'
+      app = lambda { |env| [200, {'Content-Type' => 'application/json'}, [test_body]] }
+      request = Rack::MockRequest.env_for("/", :params => "foo=bar&callback=#{callback}")
+      body = Rack::JSONP.new(app).call(request).last
+      body.should.equal ["#{callback}(#{test_body})"]
+    end
+    
+    specify "should not wrap the response body in a callback if body is not JSON" do
+      test_body = '{"bar":"foo"}'
+      callback = 'foo'
+      app = lambda { |env| [200, {'Content-Type' => 'text/plain'}, [test_body]] }
+      request = Rack::MockRequest.env_for("/", :params => "foo=bar&callback=#{callback}")
+      body = Rack::JSONP.new(app).call(request).last
+      body.should.equal ['{"bar":"foo"}']
+    end
+    
+    specify "should update content length if it was set" do
+      test_body = '{"bar":"foo"}'
+      callback = 'foo'
+      app = lambda { |env| [200, {'Content-Type' => 'application/json', 'Content-Length' => test_body.length}, [test_body]] }
+      request = Rack::MockRequest.env_for("/", :params => "foo=bar&callback=#{callback}")
+
+      headers = Rack::JSONP.new(app).call(request)[1]
+      expected_length = test_body.length + callback.length + "()".length
+      headers['Content-Length'].should.equal (expected_length.to_s)
+    end
+    
+    specify "should not touch content length if not set" do
+      test_body = '{"bar":"foo"}'
+      callback = 'foo'
+      app = lambda { |env| [200, {'Content-Type' => 'application/json'}, [test_body]] }
+      request = Rack::MockRequest.env_for("/", :params => "foo=bar&callback=#{callback}")
+      headers = Rack::JSONP.new(app).call(request)[1]
+      headers['Content-Length'].should.equal nil
+    end
+    
+    specify "should modify the content type to application/javascript" do
+      test_body = '{"bar":"foo"}'
+      callback = 'foo'
+      app = lambda { |env| [200, {'Content-Type' => 'application/json'}, [test_body]] }
+      request = Rack::MockRequest.env_for("/", :params => "foo=bar&callback=#{callback}")
+      headers = Rack::JSONP.new(app).call(request)[1]
+      headers['Content-Type'].should.equal ('application/javascript')
+    end
+    
+    specify "should write status code to response and set status to 200 if status is 40x or 50x" do
+      callback = 'foo'
+      app = lambda { |env| [404, {'Content-Type' => 'application/json'}, [""]] }
+      request = Rack::MockRequest.env_for("/", :params => "foo=bar&callback=#{callback}")
+      status, headers, body = Rack::JSONP.new(app).call(request)
+      status.should.equal 200
+      body.should.equal ["#{callback}({\"errorCode\":404})"]
+    end
+    
+  end
+
+  specify "should not change anything if no callback param is provided" do
+    test_body = ['{"bar":"foo"}']
+    app = lambda { |env| [200, {'Content-Type' => 'application/json'}, test_body] }
+    request = Rack::MockRequest.env_for("/", :params => "foo=bar")
+    body = Rack::JSONP.new(app).call(request).last
+    body.should.equal test_body
+  end
+
+  specify "should not change anything if it's not a json response" do
+    test_body = '<html><body>404 Not Found</body></html>'
+    app = lambda { |env| [404, {'Content-Type' => 'text/html'}, [test_body]] }
+    request = Rack::MockRequest.env_for("/", :params => "callback=foo", 'HTTP_ACCEPT' => 'application/json')
+    body = Rack::JSONP.new(app).call(request).last
+    body.should.equal [test_body]
+  end
+
+end

data/test/spec_rack_lighttpd_script_name_fix.rb

@@ -0,0 +1,16 @@
+require 'test/spec'
+require 'rack/mock'
+require 'rack/contrib/lighttpd_script_name_fix'
+
+context "Rack::LighttpdScriptNameFix" do
+  specify "corrects SCRIPT_NAME and PATH_INFO set by lighttpd " do
+    env = {
+      "PATH_INFO" => "/foo/bar/baz",
+      "SCRIPT_NAME" => "/hello"
+    }
+    app = lambda { |_| [200, {'Content-Type' => 'text/plain'}, ["Hello, World!"]] }
+    response = Rack::LighttpdScriptNameFix.new(app).call(env)
+    env['SCRIPT_NAME'].should.be.empty
+    env['PATH_INFO'].should.equal '/hello/foo/bar/baz'
+  end
+end

data/test/spec_rack_mailexceptions.rb

@@ -0,0 +1,97 @@
+require 'test/spec'
+require 'rack/mock'
+
+begin
+  require 'tmail'
+  require 'rack/contrib/mailexceptions'
+
+  require File.dirname(__FILE__) + '/mail_settings.rb'
+
+  class TestError < RuntimeError
+  end
+
+  def test_exception
+    raise TestError, 'Suffering Succotash!'
+  rescue => boom
+    return boom
+  end
+
+  context 'Rack::MailExceptions' do
+
+    setup do
+      @app = lambda { |env| raise TestError, 'Why, I say' }
+      @env = Rack::MockRequest.env_for("/foo",
+        'FOO' => 'BAR',
+        :method => 'GET',
+        :input => 'THE BODY'
+      )
+      @smtp_settings = {
+        :server         => 'example.com',
+        :domain         => 'example.com',
+        :port           => 500,
+        :authentication => :login,
+        :user_name      => 'joe',
+        :password       => 'secret'
+      }
+    end
+
+    specify 'yields a configuration object to the block when created' do
+      called = false
+      mailer =
+        Rack::MailExceptions.new(@app) do |mail|
+          called = true
+          mail.to 'foo@example.org'
+          mail.from 'bar@example.org'
+          mail.subject '[ERROR] %s'
+          mail.smtp @smtp_settings
+        end
+      called.should.be == true
+    end
+
+    specify 'generates a TMail object with configured settings' do
+      mailer =
+        Rack::MailExceptions.new(@app) do |mail|
+          mail.to 'foo@example.org'
+          mail.from 'bar@example.org'
+          mail.subject '[ERROR] %s'
+          mail.smtp @smtp_settings
+        end
+
+      tmail = mailer.send(:generate_mail, test_exception, @env)
+      tmail.to.should.equal ['foo@example.org']
+      tmail.from.should.equal ['bar@example.org']
+      tmail.subject.should.equal '[ERROR] Suffering Succotash!'
+      tmail.body.should.not.be.nil
+      tmail.body.should.be =~ /FOO:\s+"BAR"/
+      tmail.body.should.be =~ /^\s*THE BODY\s*$/
+    end
+
+    specify 'catches exceptions raised from app, sends mail, and re-raises' do
+      mailer =
+        Rack::MailExceptions.new(@app) do |mail|
+          mail.to 'foo@example.org'
+          mail.from 'bar@example.org'
+          mail.subject '[ERROR] %s'
+          mail.smtp @smtp_settings
+        end
+      lambda { mailer.call(@env) }.should.raise(TestError)
+      @env['mail.sent'].should.be == true
+    end
+
+    if TEST_SMTP && ! TEST_SMTP.empty?
+      specify 'sends mail' do
+        mailer =
+          Rack::MailExceptions.new(@app) do |mail|
+            mail.config.merge! TEST_SMTP
+          end
+        lambda { mailer.call(@env) }.should.raise(TestError)
+        @env['mail.sent'].should.be == true
+      end
+    else
+      STDERR.puts 'WARN: Skipping SMTP tests (edit test/mail_settings.rb to enable)'
+    end
+
+  end
+rescue LoadError => boom
+  STDERR.puts "WARN: Skipping Rack::MailExceptions tests (tmail not installed)"
+end

data/test/spec_rack_nested_params.rb

@@ -0,0 +1,46 @@
+require 'test/spec'
+require 'rack/mock'
+require 'rack/contrib/nested_params'
+require 'rack/methodoverride'
+
+context Rack::NestedParams do
+
+  App = lambda { |env| [200, {'Content-Type' => 'text/plain'}, Rack::Request.new(env)] }
+
+  def env_for_post_with_headers(path, headers, body)
+    Rack::MockRequest.env_for(path, {:method => "POST", :input => body}.merge(headers))
+  end
+
+  def form_post(params, content_type = 'application/x-www-form-urlencoded')
+    params = Rack::Utils.build_query(params) if Hash === params
+    env_for_post_with_headers('/', {'CONTENT_TYPE' => content_type}, params)
+  end
+
+  def middleware
+    Rack::NestedParams.new(App)
+  end
+
+  specify "should handle requests with POST body Content-Type of application/x-www-form-urlencoded" do
+    req = middleware.call(form_post({'foo[bar][baz]' => 'nested'})).last
+    req.POST.should.equal({"foo" => { "bar" => { "baz" => "nested" }}})
+  end
+
+  specify "should not parse requests with other Content-Type" do
+    req = middleware.call(form_post({'foo[bar][baz]' => 'nested'}, 'text/plain')).last
+    req.POST.should.equal({})
+  end
+
+  specify "should work even after another middleware already parsed the request" do
+    app = Rack::MethodOverride.new(middleware)
+    req = app.call(form_post({'_method' => 'put', 'foo[bar]' => 'nested'})).last
+    req.POST.should.equal({'_method' => 'put', "foo" => { "bar" => "nested" }})
+    req.put?.should.equal true
+  end
+
+  specify "should make first boolean have precedence even after request already parsed" do
+    app = Rack::MethodOverride.new(middleware)
+    req = app.call(form_post("foo=1&foo=0")).last
+    req.POST.should.equal({"foo" => '1'})
+  end
+
+end

data/test/spec_rack_not_found.rb

@@ -0,0 +1,17 @@
+require 'test/spec'
+require 'rack/mock'
+require 'rack/contrib/not_found'
+
+context "Rack::NotFound" do
+
+  specify "should render the file at the given path for all requests" do
+    app = Rack::Builder.new do
+      use Rack::Lint
+      run Rack::NotFound.new('test/404.html')
+    end
+    response = Rack::MockRequest.new(app).get('/')
+    response.body.should.equal('Not Found')
+    response.status.should.equal(404)
+  end
+
+end

data/test/spec_rack_post_body_content_type_parser.rb

@@ -0,0 +1,32 @@
+require 'test/spec'
+require 'rack/mock'
+
+begin
+  require 'rack/contrib/post_body_content_type_parser'
+
+  context "Rack::PostBodyContentTypeParser" do
+
+    specify "should handle requests with POST body Content-Type of application/json" do
+      app = lambda { |env| [200, {'Content-Type' => 'text/plain'}, Rack::Request.new(env).POST] }
+      env = env_for_post_with_headers('/', {'Content_Type'.upcase => 'application/json'}, {:body => "asdf", :status => "12"}.to_json)
+      body = Rack::PostBodyContentTypeParser.new(app).call(env).last
+      body['body'].should.equal "asdf"
+      body['status'].should.equal "12"
+    end
+
+    specify "should change nothing when the POST body content type isn't application/json" do
+      app = lambda { |env| [200, {'Content-Type' => 'text/plain'}, Rack::Request.new(env).POST] }
+      body = Rack::PostBodyContentTypeParser.new(app).call(Rack::MockRequest.env_for("/", {:method => 'POST', :params => "body=asdf&status=12"})).last
+      body['body'].should.equal "asdf"
+      body['status'].should.equal "12"
+    end
+
+  end
+
+  def env_for_post_with_headers(path, headers, body)
+    Rack::MockRequest.env_for(path, {:method => "POST", :input => body}.merge(headers))
+  end
+rescue LoadError => e
+  # Missing dependency JSON, skipping tests.
+  STDERR.puts "WARN: Skipping Rack::PostBodyContentTypeParser tests (json not installed)"
+end

data/test/spec_rack_proctitle.rb

@@ -0,0 +1,26 @@
+require 'test/spec'
+require 'rack/mock'
+require 'rack/contrib/proctitle'
+
+context "Rack::ProcTitle" do
+  F = ::File
+
+  progname = File.basename($0)
+  appname = F.expand_path(__FILE__).split('/')[-3]
+
+  def simple_app(body=['Hello World!'])
+    lambda { |env| [200, {'Content-Type' => 'text/plain'}, body] }
+  end
+
+  specify "should set the process title when created" do
+    Rack::ProcTitle.new(simple_app)
+    $0.should.equal "#{progname} [#{appname}] init ..."
+  end
+
+  specify "should set the process title on each request" do
+    app = Rack::ProcTitle.new(simple_app)
+    req = Rack::MockRequest.new(app)
+    10.times { req.get('/hello') }
+    $0.should.equal "#{progname} [#{appname}/80] (10) GET /hello"
+  end
+end

data/test/spec_rack_profiler.rb

@@ -0,0 +1,37 @@
+require 'test/spec'
+require 'rack/mock'
+
+begin
+  require 'rack/contrib/profiler'
+
+  context 'Rack::Profiler' do
+    app = lambda { |env| [200, {'Content-Type' => 'text/plain'}, 'Oh hai der'] }
+    request = Rack::MockRequest.env_for("/", :params => "profile=process_time")
+
+    specify 'printer defaults to RubyProf::CallTreePrinter' do
+      profiler = Rack::Profiler.new(nil)
+      profiler.instance_variable_get('@printer').should.equal RubyProf::CallTreePrinter
+      profiler.instance_variable_get('@times').should.equal 1
+    end
+
+    specify 'CallTreePrinter has correct headers' do
+      headers = Rack::Profiler.new(app).call(request)[1]
+      headers.should.equal "Content-Disposition"=>"attachment; filename=\"/.process_time.tree\"", "Content-Type"=>"application/octet-stream"
+    end
+
+    specify 'FlatPrinter and GraphPrinter has Content-Type text/plain' do
+      %w(flat graph).each do |printer|
+        headers = Rack::Profiler.new(app, :printer => printer.to_sym).call(request)[1]
+        headers.should.equal "Content-Type"=>"text/plain"
+      end
+    end
+
+    specify 'GraphHtmlPrinter has Content-Type text/html' do
+      headers = Rack::Profiler.new(app, :printer => :graph_html).call(request)[1]
+      headers.should.equal "Content-Type"=>"text/html"
+    end
+  end
+
+rescue LoadError => boom
+  $stderr.puts "WARN: Skipping Rack::Profiler tests (ruby-prof not installed)"
+end