Permalink
Fetching contributors…
Cannot retrieve contributors at this time
161 lines (137 sloc) 5.14 KB
require 'minitest/autorun'
require 'fileutils'
require 'rack/lint'
require 'rack/sendfile'
require 'rack/mock'
require 'tmpdir'
describe Rack::Sendfile do
def sendfile_body
FileUtils.touch File.join(Dir.tmpdir, "rack_sendfile")
res = ['Hello World']
def res.to_path ; File.join(Dir.tmpdir, "rack_sendfile") ; end
res
end
def simple_app(body=sendfile_body)
lambda { |env| [200, {'Content-Type' => 'text/plain'}, body] }
end
def sendfile_app(body, mappings = [])
Rack::Lint.new Rack::Sendfile.new(simple_app(body), nil, mappings)
end
def request(headers={}, body=sendfile_body, mappings=[])
yield Rack::MockRequest.new(sendfile_app(body, mappings)).get('/', headers)
end
def open_file(path)
Class.new(File) do
unless method_defined?(:to_path)
alias :to_path :path
end
end.open(path, 'wb+')
end
it "does nothing when no X-Sendfile-Type header present" do
request do |response|
response.must_be :ok?
response.body.must_equal 'Hello World'
response.headers.wont_include 'X-Sendfile'
end
end
it "sets X-Sendfile response header and discards body" do
request 'HTTP_X_SENDFILE_TYPE' => 'X-Sendfile' do |response|
response.must_be :ok?
response.body.must_be :empty?
response.headers['Content-Length'].must_equal '0'
response.headers['X-Sendfile'].must_equal File.join(Dir.tmpdir, "rack_sendfile")
end
end
it "sets X-Lighttpd-Send-File response header and discards body" do
request 'HTTP_X_SENDFILE_TYPE' => 'X-Lighttpd-Send-File' do |response|
response.must_be :ok?
response.body.must_be :empty?
response.headers['Content-Length'].must_equal '0'
response.headers['X-Lighttpd-Send-File'].must_equal File.join(Dir.tmpdir, "rack_sendfile")
end
end
it "sets X-Accel-Redirect response header and discards body" do
headers = {
'HTTP_X_SENDFILE_TYPE' => 'X-Accel-Redirect',
'HTTP_X_ACCEL_MAPPING' => "#{Dir.tmpdir}/=/foo/bar/"
}
request headers do |response|
response.must_be :ok?
response.body.must_be :empty?
response.headers['Content-Length'].must_equal '0'
response.headers['X-Accel-Redirect'].must_equal '/foo/bar/rack_sendfile'
end
end
it 'writes to rack.error when no X-Accel-Mapping is specified' do
request 'HTTP_X_SENDFILE_TYPE' => 'X-Accel-Redirect' do |response|
response.must_be :ok?
response.body.must_equal 'Hello World'
response.headers.wont_include 'X-Accel-Redirect'
response.errors.must_include 'X-Accel-Mapping'
end
end
it 'does nothing when body does not respond to #to_path' do
request({'HTTP_X_SENDFILE_TYPE' => 'X-Sendfile'}, ['Not a file...']) do |response|
response.body.must_equal 'Not a file...'
response.headers.wont_include 'X-Sendfile'
end
end
it "sets X-Accel-Redirect response header and discards body when initialized with multiple mappings" do
begin
dir1 = Dir.mktmpdir
dir2 = Dir.mktmpdir
first_body = open_file(File.join(dir1, 'rack_sendfile'))
first_body.puts 'hello world'
second_body = open_file(File.join(dir2, 'rack_sendfile'))
second_body.puts 'goodbye world'
mappings = [
["#{dir1}/", '/foo/bar/'],
["#{dir2}/", '/wibble/']
]
request({'HTTP_X_SENDFILE_TYPE' => 'X-Accel-Redirect'}, first_body, mappings) do |response|
response.must_be :ok?
response.body.must_be :empty?
response.headers['Content-Length'].must_equal '0'
response.headers['X-Accel-Redirect'].must_equal '/foo/bar/rack_sendfile'
end
request({'HTTP_X_SENDFILE_TYPE' => 'X-Accel-Redirect'}, second_body, mappings) do |response|
response.must_be :ok?
response.body.must_be :empty?
response.headers['Content-Length'].must_equal '0'
response.headers['X-Accel-Redirect'].must_equal '/wibble/rack_sendfile'
end
ensure
FileUtils.remove_entry_secure dir1
FileUtils.remove_entry_secure dir2
end
end
it "sets X-Accel-Redirect response header and discards body when initialized with multiple mappings via header" do
begin
dir1 = Dir.mktmpdir
dir2 = Dir.mktmpdir
first_body = open_file(File.join(dir1, 'rack_sendfile'))
first_body.puts 'hello world'
second_body = open_file(File.join(dir2, 'rack_sendfile'))
second_body.puts 'goodbye world'
headers = {
'HTTP_X_SENDFILE_TYPE' => 'X-Accel-Redirect',
'HTTP_X_ACCEL_MAPPING' => "#{dir1}/=/foo/bar/, #{dir2}/=/wibble/"
}
request(headers, first_body) do |response|
response.must_be :ok?
response.body.must_be :empty?
response.headers['Content-Length'].must_equal '0'
response.headers['X-Accel-Redirect'].must_equal '/foo/bar/rack_sendfile'
end
request(headers, second_body) do |response|
response.must_be :ok?
response.body.must_be :empty?
response.headers['Content-Length'].must_equal '0'
response.headers['X-Accel-Redirect'].must_equal '/wibble/rack_sendfile'
end
ensure
FileUtils.remove_entry_secure dir1
FileUtils.remove_entry_secure dir2
end
end
end