File: hijack.rb

package info (click to toggle)
ruby-protocol-http1 0.35.2-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 376 kB
  • sloc: ruby: 2,367; makefile: 4
file content (60 lines) | stat: -rw-r--r-- 1,654 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
# frozen_string_literal: true

# Released under the MIT License.
# Copyright, 2019-2025, by Samuel Williams.
# Copyright, 2024, by Anton Zhuravsky.
# Copyright, 2024, by Thomas Morgan.

require "protocol/http1/connection"
require "protocol/http/body/buffered"
require "connection_context"

describe Protocol::HTTP1::Connection do
	include_context ConnectionContext
	
	with "#hijack" do
		let(:response_version) {Protocol::HTTP1::Connection::HTTP10}
		let(:body) {Protocol::HTTP::Body::Buffered.new}
		let(:text) {"Hello World!"}
		
		it "should not be persistent after hijack" do
			server_wrapper = server.hijack!
			expect(server.persistent).to be == false
		end
		
		it "should repord itself as #hijacked? after the hijack" do
			expect(server.hijacked?).to be == false
			server.hijack!
			expect(server.hijacked?).to be == true
		end
		
		it "should use non-chunked output" do
			server.open!
			
			expect(body).to receive(:ready?).and_return(false)
			expect(body).to receive(:each).and_return(nil)
			server.write_response(response_version, 101, {"upgrade" => "websocket"})
			server.write_body(response_version, body)
			
			server_stream = server.hijack!
			
			client.open!
			
			version, status, reason, headers, body = client.read_response("GET")
			
			expect(version).to be == response_version
			expect(status).to be == 101
			expect(headers).to have_keys(
				"upgrade" => be == ["websocket"],
			)
			expect(body).to be_a(::Protocol::HTTP1::Body::Remainder) # due to 101 status
			
			client_stream = client.hijack!
			
			client_stream.write(text)
			client_stream.close
			
			expect(server_stream.read).to be == text
		end
	end
end