68 lines
2.6 KiB
Crystal
68 lines
2.6 KiB
Crystal
require "./spec_helper"
|
|
|
|
describe "Kemal::WebSocketHandler" do
|
|
it "doesn't match on wrong route" do
|
|
handler = Kemal::WebSocketHandler::INSTANCE
|
|
handler.next = Kemal::RouteHandler::INSTANCE
|
|
ws "/" { }
|
|
headers = HTTP::Headers{
|
|
"Upgrade" => "websocket",
|
|
"Connection" => "Upgrade",
|
|
"Sec-WebSocket-Key" => "dGhlIHNhbXBsZSBub25jZQ==",
|
|
}
|
|
request = HTTP::Request.new("GET", "/asd", headers)
|
|
io = IO::Memory.new
|
|
response = HTTP::Server::Response.new(io)
|
|
context = HTTP::Server::Context.new(request, response)
|
|
|
|
expect_raises(Kemal::Exceptions::RouteNotFound) do
|
|
handler.call context
|
|
end
|
|
end
|
|
|
|
it "matches on given route" do
|
|
handler = Kemal::WebSocketHandler::INSTANCE
|
|
ws "/" { |socket| socket.send("Match") }
|
|
ws "/no_match" { |socket| socket.send "No Match" }
|
|
headers = HTTP::Headers{
|
|
"Upgrade" => "websocket",
|
|
"Connection" => "Upgrade",
|
|
"Sec-WebSocket-Key" => "dGhlIHNhbXBsZSBub25jZQ==",
|
|
"Sec-WebSocket-Version" => "13",
|
|
}
|
|
request = HTTP::Request.new("GET", "/", headers)
|
|
|
|
io_with_context = create_ws_request_and_return_io_and_context(handler, request)[0]
|
|
io_with_context.to_s.should eq("HTTP/1.1 101 Switching Protocols\r\nUpgrade: websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n\r\n\x81\u0005Match")
|
|
end
|
|
|
|
it "fetches named url parameters" do
|
|
handler = Kemal::WebSocketHandler::INSTANCE
|
|
ws "/:id" { |_, c| c.ws_route_lookup.params["id"] }
|
|
headers = HTTP::Headers{
|
|
"Upgrade" => "websocket",
|
|
"Connection" => "Upgrade",
|
|
"Sec-WebSocket-Key" => "dGhlIHNhbXBsZSBub25jZQ==",
|
|
"Sec-WebSocket-Version" => "13",
|
|
}
|
|
request = HTTP::Request.new("GET", "/1234", headers)
|
|
io_with_context = create_ws_request_and_return_io_and_context(handler, request)[0]
|
|
io_with_context.to_s.should eq("HTTP/1.1 101 Switching Protocols\r\nUpgrade: websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n\r\n")
|
|
end
|
|
|
|
it "matches correct verb" do
|
|
handler = Kemal::WebSocketHandler::INSTANCE
|
|
handler.next = Kemal::RouteHandler::INSTANCE
|
|
ws "/" { }
|
|
get "/" { "get" }
|
|
request = HTTP::Request.new("GET", "/")
|
|
io = IO::Memory.new
|
|
response = HTTP::Server::Response.new(io)
|
|
context = HTTP::Server::Context.new(request, response)
|
|
handler.call(context)
|
|
response.close
|
|
io.rewind
|
|
client_response = HTTP::Client::Response.from_io(io, decompress: false)
|
|
client_response.body.should eq("get")
|
|
end
|
|
end
|