mirror of
https://gitea.invidious.io/iv-org/shard-kemal.git
synced 2024-08-15 00:53:36 +00:00
60 lines
2 KiB
Crystal
60 lines
2 KiB
Crystal
|
require "./spec_helper"
|
||
|
|
|
||
|
|
describe "Kemal::WebsocketHandler" do
|
||
|
|
it "doesn't match on wrong route" do
|
||
|
|
handler = Kemal::WebsocketHandler.new "/" { }
|
||
|
|
headers = HTTP::Headers{
|
||
|
|
"Upgrade": "websocket",
|
||
|
|
"Connection": "Upgrade",
|
||
|
|
"Sec-WebSocket-Key": "dGhlIHNhbXBsZSBub25jZQ==",
|
||
|
|
}
|
||
|
|
request = HTTP::Request.new("GET", "/asd", headers)
|
||
|
|
response = handler.call request
|
||
|
|
response.status_code.should eq(404)
|
||
|
|
end
|
||
|
|
|
||
|
|
it "matches on given route" do
|
||
|
|
handler = Kemal::WebsocketHandler.new "/" { }
|
||
|
|
headers = HTTP::Headers{
|
||
|
|
"Upgrade": "websocket",
|
||
|
|
"Connection": "Upgrade",
|
||
|
|
"Sec-WebSocket-Key": "dGhlIHNhbXBsZSBub25jZQ==",
|
||
|
|
}
|
||
|
|
request = HTTP::Request.new("GET", "/", headers)
|
||
|
|
response = handler.call request
|
||
|
|
response.status_code.should eq(101)
|
||
|
|
response.headers["Upgrade"].should eq("websocket")
|
||
|
|
response.headers["Connection"].should eq("Upgrade")
|
||
|
|
response.headers["Sec-WebSocket-Accept"].should eq("s3pPLMBiTxaQ9kYGzzhZRbK+xOo=")
|
||
|
|
response.upgrade_handler.should_not be_nil
|
||
|
|
end
|
||
|
|
|
||
|
|
it "doesn't mix http and ws on same route" do
|
||
|
|
kemal = Kemal::Handler.new
|
||
|
|
kemal.add_route "GET", "/" do |env|
|
||
|
|
"hello #{env.params["message"]}"
|
||
|
|
end
|
||
|
|
|
||
|
|
ws_handler = Kemal::WebsocketHandler.new "/" { }
|
||
|
|
headers = HTTP::Headers{
|
||
|
|
"Upgrade": "websocket",
|
||
|
|
"Connection": "Upgrade",
|
||
|
|
"Sec-WebSocket-Key": "dGhlIHNhbXBsZSBub25jZQ==",
|
||
|
|
}
|
||
|
|
|
||
|
|
# HTTP Request
|
||
|
|
request = HTTP::Request.new("GET", "/?message=world")
|
||
|
|
response = kemal.call(request)
|
||
|
|
response.body.should eq("hello world")
|
||
|
|
|
||
|
|
# Websocket request
|
||
|
|
request = HTTP::Request.new("GET", "/", headers)
|
||
|
|
response = ws_handler.call request
|
||
|
|
response.status_code.should eq(101)
|
||
|
|
response.headers["Upgrade"].should eq("websocket")
|
||
|
|
response.headers["Connection"].should eq("Upgrade")
|
||
|
|
response.headers["Sec-WebSocket-Accept"].should eq("s3pPLMBiTxaQ9kYGzzhZRbK+xOo=")
|
||
|
|
response.upgrade_handler.should_not be_nil
|
||
|
|
end
|
||
|
|
end
|