shard-kemal/spec/spec_helper.cr

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
  Kemal::RouteHandler::INSTANCE.tree = Radix::Tree.new
end