shard-kemal/spec/kemal_ws_handler_spec.cr
2015-12-16 17:06:06 +02:00

59 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