77 lines
2.1 KiB
Ruby
77 lines
2.1 KiB
Ruby
require 'test/spec'
|
|
|
|
require 'rack/recursive'
|
|
require 'rack/urlmap'
|
|
require 'rack/response'
|
|
require 'rack/mock'
|
|
|
|
context "Rack::Recursive" do
|
|
setup do
|
|
|
|
@app1 = lambda { |env|
|
|
res = Rack::Response.new
|
|
res["X-Path-Info"] = env["PATH_INFO"]
|
|
res["X-Query-String"] = env["QUERY_STRING"]
|
|
res.finish do |res|
|
|
res.write "App1"
|
|
end
|
|
}
|
|
|
|
@app2 = lambda { |env|
|
|
Rack::Response.new.finish do |res|
|
|
res.write "App2"
|
|
_, _, body = env['rack.recursive.include'].call(env, "/app1")
|
|
body.each { |b|
|
|
res.write b
|
|
}
|
|
end
|
|
}
|
|
|
|
@app3 = lambda { |env|
|
|
raise Rack::ForwardRequest.new("/app1")
|
|
}
|
|
|
|
@app4 = lambda { |env|
|
|
raise Rack::ForwardRequest.new("http://example.org/app1/quux?meh")
|
|
}
|
|
|
|
end
|
|
|
|
specify "should allow for subrequests" do
|
|
res = Rack::MockRequest.new(Rack::Recursive.new(
|
|
Rack::URLMap.new("/app1" => @app1,
|
|
"/app2" => @app2))).
|
|
get("/app2")
|
|
|
|
res.should.be.ok
|
|
res.body.should.equal "App2App1"
|
|
end
|
|
|
|
specify "should raise error on requests not below the app" do
|
|
app = Rack::URLMap.new("/app1" => @app1,
|
|
"/app" => Rack::Recursive.new(
|
|
Rack::URLMap.new("/1" => @app1,
|
|
"/2" => @app2)))
|
|
|
|
lambda {
|
|
Rack::MockRequest.new(app).get("/app/2")
|
|
}.should.raise(ArgumentError).
|
|
message.should =~ /can only include below/
|
|
end
|
|
|
|
specify "should support forwarding" do
|
|
app = Rack::Recursive.new(Rack::URLMap.new("/app1" => @app1,
|
|
"/app3" => @app3,
|
|
"/app4" => @app4))
|
|
|
|
res = Rack::MockRequest.new(app).get("/app3")
|
|
res.should.be.ok
|
|
res.body.should.equal "App1"
|
|
|
|
res = Rack::MockRequest.new(app).get("/app4")
|
|
res.should.be.ok
|
|
res.body.should.equal "App1"
|
|
res["X-Path-Info"].should.equal "/quux"
|
|
res["X-Query-String"].should.equal "meh"
|
|
end
|
|
end
|