2009-01-13 17:53:53 +00:00
|
|
|
require File.dirname(__FILE__) + '/helper'
|
2008-12-13 21:06:02 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
class BaseTest < Test::Unit::TestCase
|
2009-05-17 03:36:04 +00:00
|
|
|
def test_default
|
|
|
|
assert true
|
|
|
|
end
|
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
describe 'Sinatra::Base subclasses' do
|
|
|
|
class TestApp < Sinatra::Base
|
|
|
|
get '/' do
|
|
|
|
'Hello World'
|
|
|
|
end
|
|
|
|
end
|
2009-02-22 08:53:43 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
it 'include Rack::Utils' do
|
|
|
|
assert TestApp.included_modules.include?(Rack::Utils)
|
2009-02-22 08:53:43 +00:00
|
|
|
end
|
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
it 'processes requests with #call' do
|
|
|
|
assert TestApp.respond_to?(:call)
|
2008-12-13 21:06:02 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
request = Rack::MockRequest.new(TestApp)
|
|
|
|
response = request.get('/')
|
|
|
|
assert response.ok?
|
|
|
|
assert_equal 'Hello World', response.body
|
|
|
|
end
|
2008-12-13 21:06:02 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
class TestApp < Sinatra::Base
|
|
|
|
get '/state' do
|
2009-05-17 03:36:04 +00:00
|
|
|
@foo ||= "new"
|
2009-03-26 15:42:13 +00:00
|
|
|
body = "Foo: #{@foo}"
|
|
|
|
@foo = 'discard'
|
|
|
|
body
|
|
|
|
end
|
|
|
|
end
|
2008-12-13 21:06:02 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
it 'does not maintain state between requests' do
|
|
|
|
request = Rack::MockRequest.new(TestApp)
|
|
|
|
2.times do
|
|
|
|
response = request.get('/state')
|
|
|
|
assert response.ok?
|
2009-05-17 03:36:04 +00:00
|
|
|
assert_equal 'Foo: new', response.body
|
2009-03-26 15:42:13 +00:00
|
|
|
end
|
2009-02-22 08:53:43 +00:00
|
|
|
end
|
2009-03-31 16:52:49 +00:00
|
|
|
|
|
|
|
it "passes the subclass to configure blocks" do
|
|
|
|
ref = nil
|
|
|
|
TestApp.configure { |app| ref = app }
|
|
|
|
assert_equal TestApp, ref
|
|
|
|
end
|
|
|
|
|
|
|
|
it "allows the configure block arg to be omitted and does not change context" do
|
|
|
|
context = nil
|
|
|
|
TestApp.configure { context = self }
|
|
|
|
assert_equal self, context
|
|
|
|
end
|
2009-02-22 08:53:43 +00:00
|
|
|
end
|
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
describe "Sinatra::Base as Rack middleware" do
|
|
|
|
app = lambda { |env|
|
|
|
|
headers = {'X-Downstream' => 'true'}
|
|
|
|
headers['X-Route-Missing'] = env['sinatra.route-missing'] || ''
|
|
|
|
[210, headers, ['Hello from downstream']] }
|
|
|
|
|
|
|
|
class TestMiddleware < Sinatra::Base
|
2009-02-22 08:53:43 +00:00
|
|
|
end
|
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
it 'creates a middleware that responds to #call with .new' do
|
|
|
|
middleware = TestMiddleware.new(app)
|
|
|
|
assert middleware.respond_to?(:call)
|
|
|
|
end
|
2009-02-22 08:53:43 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
it 'exposes the downstream app' do
|
2011-03-12 14:52:43 +00:00
|
|
|
middleware = TestMiddleware.new!(app)
|
2009-03-26 15:42:13 +00:00
|
|
|
assert_same app, middleware.app
|
|
|
|
end
|
2009-02-22 08:53:43 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
class TestMiddleware < Sinatra::Base
|
|
|
|
def route_missing
|
|
|
|
env['sinatra.route-missing'] = '1'
|
|
|
|
super
|
|
|
|
end
|
2009-02-22 08:53:43 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
get '/' do
|
|
|
|
'Hello from middleware'
|
|
|
|
end
|
|
|
|
end
|
2009-02-22 08:53:43 +00:00
|
|
|
|
|
|
|
middleware = TestMiddleware.new(app)
|
2009-03-26 15:42:13 +00:00
|
|
|
request = Rack::MockRequest.new(middleware)
|
2008-12-13 21:06:02 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
it 'intercepts requests' do
|
|
|
|
response = request.get('/')
|
|
|
|
assert response.ok?
|
|
|
|
assert_equal 'Hello from middleware', response.body
|
2009-03-11 21:44:09 +00:00
|
|
|
end
|
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
it 'automatically forwards requests downstream when no matching route found' do
|
|
|
|
response = request.get('/missing')
|
|
|
|
assert_equal 210, response.status
|
|
|
|
assert_equal 'Hello from downstream', response.body
|
2009-02-22 08:53:43 +00:00
|
|
|
end
|
2008-12-13 21:06:02 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
it 'calls #route_missing before forwarding downstream' do
|
|
|
|
response = request.get('/missing')
|
|
|
|
assert_equal '1', response['X-Route-Missing']
|
|
|
|
end
|
2009-02-22 08:53:43 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
class TestMiddleware < Sinatra::Base
|
|
|
|
get '/low-level-forward' do
|
|
|
|
app.call(env)
|
|
|
|
end
|
|
|
|
end
|
2009-03-11 21:44:09 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
it 'can call the downstream app directly and return result' do
|
|
|
|
response = request.get('/low-level-forward')
|
|
|
|
assert_equal 210, response.status
|
|
|
|
assert_equal 'true', response['X-Downstream']
|
|
|
|
assert_equal 'Hello from downstream', response.body
|
2009-02-22 08:53:43 +00:00
|
|
|
end
|
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
class TestMiddleware < Sinatra::Base
|
|
|
|
get '/explicit-forward' do
|
|
|
|
response['X-Middleware'] = 'true'
|
|
|
|
res = forward
|
|
|
|
assert_nil res
|
|
|
|
assert_equal 210, response.status
|
|
|
|
assert_equal 'true', response['X-Downstream']
|
|
|
|
assert_equal ['Hello from downstream'], response.body
|
|
|
|
'Hello after explicit forward'
|
|
|
|
end
|
|
|
|
end
|
2009-02-22 08:53:43 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
it 'forwards the request downstream and integrates the response into the current context' do
|
|
|
|
response = request.get('/explicit-forward')
|
2009-02-22 08:53:43 +00:00
|
|
|
assert_equal 210, response.status
|
|
|
|
assert_equal 'true', response['X-Downstream']
|
2009-03-26 15:42:13 +00:00
|
|
|
assert_equal 'Hello after explicit forward', response.body
|
|
|
|
assert_equal '28', response['Content-Length']
|
2009-02-22 08:53:43 +00:00
|
|
|
end
|
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
app_content_length = lambda {|env|
|
|
|
|
[200, {'Content-Length' => '16'}, 'From downstream!']}
|
2009-02-24 15:43:53 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
class TestMiddlewareContentLength < Sinatra::Base
|
|
|
|
get '/forward' do
|
|
|
|
res = forward
|
|
|
|
'From after explicit forward!'
|
|
|
|
end
|
2009-02-24 15:43:53 +00:00
|
|
|
end
|
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
middleware_content_length = TestMiddlewareContentLength.new(app_content_length)
|
|
|
|
request_content_length = Rack::MockRequest.new(middleware_content_length)
|
2009-02-24 15:43:53 +00:00
|
|
|
|
2009-03-26 15:42:13 +00:00
|
|
|
it "sets content length for last response" do
|
|
|
|
response = request_content_length.get('/forward')
|
|
|
|
assert_equal '28', response['Content-Length']
|
|
|
|
end
|
2008-12-13 21:06:02 +00:00
|
|
|
end
|
|
|
|
end
|