mirror of
https://gitea.invidious.io/iv-org/shard-kemal.git
synced 2024-08-15 00:53:36 +00:00
Middleware ordering (#236)
Improve and correct request middleware Request -> Middleware -> Filter -> Route
This commit is contained in:
parent
1b27f5c011
commit
922d6de4d1
6 changed files with 79 additions and 24 deletions
30
spec/handler_spec.cr
Normal file
30
spec/handler_spec.cr
Normal file
|
@ -0,0 +1,30 @@
|
|||
require "./spec_helper"
|
||||
|
||||
class CustomTestHandler < HTTP::Handler
|
||||
def call(env)
|
||||
env.response << "Kemal"
|
||||
call_next env
|
||||
end
|
||||
end
|
||||
|
||||
describe "Handler" do
|
||||
it "adds custom handler before before_*" do
|
||||
filter_middleware = Kemal::Middleware::Filter.new
|
||||
filter_middleware._add_route_filter("GET", "/", :before) do |env|
|
||||
env.response << " is"
|
||||
end
|
||||
|
||||
filter_middleware._add_route_filter("GET", "/", :before) do |env|
|
||||
env.response << " so"
|
||||
end
|
||||
add_handler CustomTestHandler.new
|
||||
|
||||
get "/" do |env|
|
||||
" Great"
|
||||
end
|
||||
request = HTTP::Request.new("GET", "/")
|
||||
client_response = call_request_on_app(request)
|
||||
client_response.status_code.should eq(200)
|
||||
client_response.body.should eq("Kemal is so Great")
|
||||
end
|
||||
end
|
|
@ -30,7 +30,7 @@ describe "Macros" do
|
|||
it "sets a custom logger" do
|
||||
config = Kemal::Config::INSTANCE
|
||||
logger CustomLogHandler.new
|
||||
config.handlers.last.should be_a(CustomLogHandler)
|
||||
config.handlers[4].should be_a(CustomLogHandler)
|
||||
config.logger.should be_a(CustomLogHandler)
|
||||
end
|
||||
end
|
||||
|
@ -77,7 +77,6 @@ describe "Macros" do
|
|||
"Content-Type" => "text/plain",
|
||||
}
|
||||
end
|
||||
|
||||
request = HTTP::Request.new("GET", "/headers")
|
||||
response = call_request_on_app(request)
|
||||
response.headers["Access-Control-Allow-Origin"].should eq("*")
|
||||
|
@ -126,11 +125,7 @@ describe "Macros" do
|
|||
describe "#gzip" do
|
||||
it "adds HTTP::DeflateHandler to handlers" do
|
||||
gzip true
|
||||
Kemal.config.handlers.last.is_a?(HTTP::DeflateHandler).should eq true
|
||||
end
|
||||
|
||||
it "doesn't add HTTP::DeflateHandler to handlers by default" do
|
||||
Kemal.config.handlers.last.is_a?(HTTP::DeflateHandler).should eq false
|
||||
Kemal.config.handlers[4].should be_a(HTTP::DeflateHandler)
|
||||
end
|
||||
end
|
||||
|
||||
|
|
|
@ -3,14 +3,9 @@ require "../src/*"
|
|||
|
||||
include Kemal
|
||||
|
||||
class CustomTestHandler < HTTP::Handler
|
||||
def call(request)
|
||||
call_next request
|
||||
end
|
||||
end
|
||||
|
||||
class CustomLogHandler < Kemal::BaseLogHandler
|
||||
def call(context)
|
||||
def call(env)
|
||||
call_next env
|
||||
end
|
||||
|
||||
def write(message)
|
||||
|
@ -44,20 +39,30 @@ def call_request_on_app(request)
|
|||
io = MemoryIO.new
|
||||
response = HTTP::Server::Response.new(io)
|
||||
context = HTTP::Server::Context.new(request, response)
|
||||
Kemal::RouteHandler::INSTANCE.call(context)
|
||||
main_handler = build_main_handler
|
||||
main_handler.call context
|
||||
response.close
|
||||
io.rewind
|
||||
HTTP::Client::Response.from_io(io, decompress: false)
|
||||
end
|
||||
|
||||
def build_main_handler
|
||||
main_handler = Kemal.config.handlers.first
|
||||
current_handler = main_handler
|
||||
Kemal.config.handlers.each_with_index do |handler, index|
|
||||
current_handler.next = handler
|
||||
current_handler = handler
|
||||
end
|
||||
main_handler
|
||||
end
|
||||
|
||||
Spec.before_each do
|
||||
config = Kemal.config
|
||||
config.env = "development"
|
||||
config.setup
|
||||
config.add_handler Kemal::RouteHandler::INSTANCE
|
||||
end
|
||||
|
||||
Spec.after_each do
|
||||
Kemal.config.handlers.clear
|
||||
Kemal.config.clear
|
||||
Kemal::RouteHandler::INSTANCE.tree = Radix::Tree(Route).new
|
||||
end
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue