mirror of
https://gitea.invidious.io/iv-org/shard-kemal.git
synced 2024-08-15 00:53:36 +00:00
43 lines
900 B
Crystal
43 lines
900 B
Crystal
require "spec"
|
|
require "../src/kemal/*"
|
|
require "../src/kemal/middleware/*"
|
|
|
|
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
|
|
|
|
Spec.before_each do
|
|
config = Kemal.config
|
|
config.env = "development"
|
|
config.setup_logging
|
|
config.handlers.clear
|
|
end
|