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
115
116
117
118
|
# frozen_string_literal: true
RSpec.describe FaradayMiddleware::Gzip, type: :response do
require 'brotli'
let(:middleware) do
described_class.new(lambda { |env|
Faraday::Response.new(env)
})
end
context 'request' do
it 'sets the Accept-Encoding request header' do
env = process('').env
expect(env[:request_headers][:accept_encoding]).to eq('gzip,deflate,br')
end
it 'doesnt overwrite existing Accept-Encoding request header' do
env = process('') do |e|
e[:request_headers][:accept_encoding] = 'zopfli'
end.env
expect(env[:request_headers][:accept_encoding]).to eq('zopfli')
end
end
context 'response' do
let(:uncompressed_body) do
'<html><head><title>Rspec</title></head><body>Hello, spec!</body></html>'
end
let(:gzipped_body) do
io = StringIO.new
gz = Zlib::GzipWriter.new(io)
gz.write(uncompressed_body)
gz.close
res = io.string
res.force_encoding('BINARY')
res
end
let(:deflated_body) do
Zlib::Deflate.deflate(uncompressed_body)
end
let(:raw_deflated_body) do
z = Zlib::Deflate.new(Zlib::DEFAULT_COMPRESSION, -Zlib::MAX_WBITS)
compressed_body = z.deflate(uncompressed_body, Zlib::FINISH)
z.close
compressed_body
end
let(:brotlied_body) do
Brotli.deflate(uncompressed_body)
end
let(:empty_body) do
''
end
shared_examples 'compressed response' do
it 'uncompresses the body' do
expect(process(body).body).to eq(uncompressed_body)
end
it 'sets the Content-Length' do
expect(process(body).headers['Content-Length']).to eq(uncompressed_body.length)
end
it 'removes the Content-Encoding' do
expect(process(body).headers['Content-Encoding']).to be_nil
end
end
context 'gzipped response' do
let(:body) { gzipped_body }
let(:headers) { { 'Content-Encoding' => 'gzip', 'Content-Length' => body.length } }
it_behaves_like 'compressed response'
end
context 'deflated response' do
let(:body) { deflated_body }
let(:headers) { { 'Content-Encoding' => 'deflate', 'Content-Length' => body.length } }
it_behaves_like 'compressed response'
end
context 'raw deflated response' do
let(:body) { raw_deflated_body }
let(:headers) { { 'Content-Encoding' => 'deflate', 'Content-Length' => body.length } }
it_behaves_like 'compressed response'
end
context 'brotlied response' do
let(:body) { brotlied_body }
let(:headers) { { 'Content-Encoding' => 'br', 'Content-Length' => body.length } }
it_behaves_like 'compressed response'
end
context 'empty response' do
let(:body) { empty_body }
let(:headers) { { 'Content-Encoding' => 'gzip', 'Content-Length' => body.length } }
it 'sets the Content-Length' do
expect(process(body).headers['Content-Length']).to eq(empty_body.length)
end
it 'removes the Content-Encoding' do
expect(process(body).headers['Content-Encoding']).to be_nil
end
end
context 'identity response' do
let(:body) { uncompressed_body }
it 'does not modify the body' do
expect(process(body).body).to eq(uncompressed_body)
end
end
end
end
|
