require "spec" require "../src/*" include Kemal class CustomTestHandler < HTTP::Handler def call(request) call_next request end end class CustomLogHandler < Kemal::BaseLogHandler; end def create_request_and_return_io(handler, request) io = MemoryIO.new response = HTTP::Server::Response.new(io) context = HTTP::Server::Context.new(request, response) handler.call(context) response.close io.rewind io end def create_ws_request_and_return_io(handler, request) io = MemoryIO.new response = HTTP::Server::Response.new(io) context = HTTP::Server::Context.new(request, response) begin handler.call context rescue IO::Error # Raises because the MemoryIO is empty end response.close io end 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) response.close io.rewind HTTP::Client::Response.from_io(io, decompress: false) 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::RouteHandler::INSTANCE.tree = Radix::Tree.new end