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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
# frozen_string_literal: true
# Released under the MIT License.
# Copyright, 2019-2024, by Samuel Williams.
require "protocol/http2/data_frame"
require "protocol/http2/a_frame"
describe Protocol::HTTP2::DataFrame do
let(:frame) {subject.new}
it_behaves_like Protocol::HTTP2::AFrame do
def before
frame.pack "Hello World!"
super
end
end
with "wire representation" do
let(:stream) {StringIO.new}
let(:payload) {"Hello World!"}
let(:data) do
[0, 12, 0x0, 0x1, 0x1].pack("CnCCNC*") + payload
end
it "should write frame to buffer" do
frame.set_flags(Protocol::HTTP2::END_STREAM)
frame.stream_id = 1
frame.payload = payload
frame.length = payload.bytesize
frame.write(stream)
expect(stream.string).to be == data
end
it "should read frame from buffer" do
stream.write(data)
stream.seek(0)
frame.read(stream)
expect(frame.length).to be == payload.bytesize
expect(frame.flags).to be == Protocol::HTTP2::END_STREAM
expect(frame.stream_id).to be == 1
expect(frame.payload).to be == payload
end
end
with "#pack" do
it "adds appropriate padding" do
frame.pack "Hello World!", padding_size: 4
expect(frame.length).to be == 17
expect(frame.payload[0].ord).to be == 4
expect(frame.unpack).to be == "Hello World!"
stream = StringIO.new
frame.write(stream)
expect(stream.string.bytesize).to be == 26
stream.rewind
frame2 = subject.new
frame2.read(stream)
expect(frame2.unpack).to be == "Hello World!"
expect(frame2).to be(:padded?)
end
it "detects invalid padding" do
frame.pack "Hello World!", padding_size: 4
expect(frame.length).to be == 17
expect(frame.payload[0].ord).to be == 4
expect(frame.unpack).to be == "Hello World!"
# Artifically set the padding to be the entire payload:
frame.payload[0] = (frame.payload.bytesize - 1).chr
expect(frame.unpack).to be == ""
# Artifically set the padding to be larger than the payload:
frame.payload[0] = (frame.payload.bytesize + 1).chr
expect do
frame.unpack
end.to raise_exception(Protocol::HTTP2::ProtocolError, message: be =~ /Invalid padding length/)
end
it "can pack end frame" do
frame.pack(nil)
expect(frame.length).to be == 0
expect(frame.flags).to be == Protocol::HTTP2::END_STREAM
expect(frame).to be(:end_stream?)
end
end
with "#unpack" do
it "removes padding" do
frame.pack "Hello World!"
expect(frame.unpack).to be == "Hello World!"
end
end
with "#inspect" do
it "can generate a string representation" do
expect(frame.inspect).to be =~ /Protocol::HTTP2::DataFrame stream_id=0 flags=0 0b/
end
end
end
|
