162 lines
4.3 KiB
Ruby
162 lines
4.3 KiB
Ruby
|
require File.expand_path(File.dirname(__FILE__) + '/helper')
|
||
|
|
||
|
describe "Application" do
|
||
|
before { Padrino.clear! }
|
||
|
after { remove_views }
|
||
|
|
||
|
context 'CSRF protection' do
|
||
|
context "with CSRF protection on" do
|
||
|
before do
|
||
|
mock_app do
|
||
|
enable :sessions
|
||
|
enable :protect_from_csrf
|
||
|
post('/'){ 'HI' }
|
||
|
end
|
||
|
end
|
||
|
|
||
|
should "not allow requests without tokens" do
|
||
|
post "/"
|
||
|
assert_equal 403, status
|
||
|
end
|
||
|
|
||
|
should "allow requests with correct tokens" do
|
||
|
post "/", {"authenticity_token" => "a"}, 'rack.session' => {:csrf => "a"}
|
||
|
assert_equal 200, status
|
||
|
end
|
||
|
|
||
|
should "not allow requests with incorrect tokens" do
|
||
|
post "/", {"authenticity_token" => "a"}, 'rack.session' => {:csrf => "b"}
|
||
|
assert_equal 403, status
|
||
|
end
|
||
|
end
|
||
|
|
||
|
context "without CSRF protection on" do
|
||
|
before do
|
||
|
mock_app do
|
||
|
enable :sessions
|
||
|
disable :protect_from_csrf
|
||
|
post('/'){ 'HI' }
|
||
|
end
|
||
|
end
|
||
|
|
||
|
should "allows requests without tokens" do
|
||
|
post "/"
|
||
|
assert_equal 200, status
|
||
|
end
|
||
|
|
||
|
should "allow requests with correct tokens" do
|
||
|
post "/", {"authenticity_token" => "a"}, 'rack.session' => {:csrf => "a"}
|
||
|
assert_equal 200, status
|
||
|
end
|
||
|
|
||
|
should "allow requests with incorrect tokens" do
|
||
|
post "/", {"authenticity_token" => "a"}, 'rack.session' => {:csrf => "b"}
|
||
|
assert_equal 200, status
|
||
|
end
|
||
|
end
|
||
|
|
||
|
context "with optional CSRF protection" do
|
||
|
before do
|
||
|
mock_app do
|
||
|
enable :sessions
|
||
|
enable :protect_from_csrf
|
||
|
set :allow_disabled_csrf, true
|
||
|
post('/on') { 'HI' }
|
||
|
post('/off', :csrf_protection => false) { 'HI' }
|
||
|
end
|
||
|
end
|
||
|
|
||
|
should "allow access to routes with csrf_protection off" do
|
||
|
post "/off"
|
||
|
assert_equal 200, status
|
||
|
end
|
||
|
|
||
|
should "not allow access to routes with csrf_protection on" do
|
||
|
post "/on"
|
||
|
assert_equal 403, status
|
||
|
end
|
||
|
end
|
||
|
|
||
|
context "with :except option that is using Proc" do
|
||
|
before do
|
||
|
mock_app do
|
||
|
enable :sessions
|
||
|
set :protect_from_csrf, :except => proc{|env| ["/", "/foo"].any?{|path| path == env['PATH_INFO'] }}
|
||
|
post("/") { "Hello" }
|
||
|
post("/foo") { "Hello, foo" }
|
||
|
post("/bar") { "Hello, bar" }
|
||
|
end
|
||
|
end
|
||
|
|
||
|
should "allow ignoring CSRF protection on specific routes" do
|
||
|
post "/"
|
||
|
assert_equal 200, status
|
||
|
post "/foo"
|
||
|
assert_equal 200, status
|
||
|
post "/bar"
|
||
|
assert_equal 403, status
|
||
|
end
|
||
|
end
|
||
|
|
||
|
context "with :except option that is using String and Regexp" do
|
||
|
before do
|
||
|
mock_app do
|
||
|
enable :sessions
|
||
|
set :protect_from_csrf, :except => ["/a", %r{^/a.c$}]
|
||
|
post("/a") { "a" }
|
||
|
post("/abc") { "abc" }
|
||
|
post("/foo") { "foo" }
|
||
|
end
|
||
|
end
|
||
|
|
||
|
should "allow ignoring CSRF protection on specific routes" do
|
||
|
post "/a"
|
||
|
assert_equal 200, status
|
||
|
post "/abc"
|
||
|
assert_equal 200, status
|
||
|
post "/foo"
|
||
|
assert_equal 403, status
|
||
|
end
|
||
|
end
|
||
|
|
||
|
context "with custom protection options" do
|
||
|
before do
|
||
|
mock_app do
|
||
|
enable :sessions
|
||
|
set :protect_from_csrf, :authenticity_param => 'foobar'
|
||
|
post("/a") { "a" }
|
||
|
end
|
||
|
end
|
||
|
|
||
|
should "allow configuring protection options" do
|
||
|
post "/a", {"foobar" => "a"}, 'rack.session' => {:csrf => "a"}
|
||
|
assert_equal 200, status
|
||
|
end
|
||
|
end
|
||
|
|
||
|
context "with middleware" do
|
||
|
before do
|
||
|
class Middleware < Sinatra::Base
|
||
|
post("/middleware") { "Hello, middleware" }
|
||
|
post("/dummy") { "Hello, dummy" }
|
||
|
end
|
||
|
mock_app do
|
||
|
enable :sessions
|
||
|
set :protect_from_csrf, :except => proc{|env| ["/", "/middleware"].any?{|path| path == env['PATH_INFO'] }}
|
||
|
use Middleware
|
||
|
post("/") { "Hello" }
|
||
|
end
|
||
|
end
|
||
|
|
||
|
should "allow ignoring CSRF protection on specific routes of middleware" do
|
||
|
post "/"
|
||
|
assert_equal 200, status
|
||
|
post "/middleware"
|
||
|
assert_equal 200, status
|
||
|
post "/dummy"
|
||
|
assert_equal 403, status
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
end
|