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 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202
|
require 'spec_helper'
describe HTTP::Client do
StubbedClient = Class.new(HTTP::Client) do
def perform(request, options)
stubs.fetch(request.uri.to_s) { super(request, options) }
end
def stubs
@stubs ||= {}
end
def stub(stubs)
@stubs = stubs
self
end
end
def redirect_response(location, status = 302)
HTTP::Response.new(status, '1.1', {'Location' => location}, '')
end
def simple_response(body, status = 200)
HTTP::Response.new(status, '1.1', {}, body)
end
describe 'following redirects' do
it 'returns response of new location' do
client = StubbedClient.new(:follow => true).stub(
'http://example.com/' => redirect_response('http://example.com/blog'),
'http://example.com/blog' => simple_response('OK')
)
expect(client.get('http://example.com/').to_s).to eq 'OK'
end
it 'prepends previous request uri scheme and host if needed' do
client = StubbedClient.new(:follow => true).stub(
'http://example.com/' => redirect_response('/index'),
'http://example.com/index' => redirect_response('/index.html'),
'http://example.com/index.html' => simple_response('OK')
)
expect(client.get('http://example.com/').to_s).to eq 'OK'
end
it 'fails upon endless redirects' do
client = StubbedClient.new(:follow => true).stub(
'http://example.com/' => redirect_response('/')
)
expect { client.get('http://example.com/') } \
.to raise_error(HTTP::Redirector::EndlessRedirectError)
end
it 'fails if max amount of hops reached' do
client = StubbedClient.new(:follow => 5).stub(
'http://example.com/' => redirect_response('/1'),
'http://example.com/1' => redirect_response('/2'),
'http://example.com/2' => redirect_response('/3'),
'http://example.com/3' => redirect_response('/4'),
'http://example.com/4' => redirect_response('/5'),
'http://example.com/5' => redirect_response('/6'),
'http://example.com/6' => simple_response('OK')
)
expect { client.get('http://example.com/') } \
.to raise_error(HTTP::Redirector::TooManyRedirectsError)
end
end
describe 'parsing params' do
let(:client) { HTTP::Client.new }
before { allow(client).to receive :perform }
it 'accepts params within the provided URL' do
expect(HTTP::Request).to receive(:new) do |_, uri|
expect(CGI.parse uri.query).to eq('foo' => %w[bar])
end
client.get('http://example.com/?foo=bar')
end
it 'combines GET params from the URI with the passed in params' do
expect(HTTP::Request).to receive(:new) do |_, uri|
expect(CGI.parse uri.query).to eq('foo' => %w[bar], 'baz' => %w[quux])
end
client.get('http://example.com/?foo=bar', :params => {:baz => 'quux'})
end
it 'merges duplicate values' do
expect(HTTP::Request).to receive(:new) do |_, uri|
expect(uri.query).to match(/^(a=1&a=2|a=2&a=1)$/)
end
client.get('http://example.com/?a=1', :params => {:a => 2})
end
it 'does not modifies query part if no params were given' do
expect(HTTP::Request).to receive(:new) do |_, uri|
expect(uri.query).to eq 'deadbeef'
end
client.get('http://example.com/?deadbeef')
end
it 'does not corrupts index-less arrays' do
expect(HTTP::Request).to receive(:new) do |_, uri|
expect(CGI.parse uri.query).to eq 'a[]' => %w[b c], 'd' => %w[e]
end
client.get('http://example.com/?a[]=b&a[]=c', :params => {:d => 'e'})
end
end
describe 'passing json' do
it 'encodes given object' do
client = HTTP::Client.new
allow(client).to receive(:perform)
expect(HTTP::Request).to receive(:new) do |*args|
expect(args.last).to eq('{"foo":"bar"}')
end
client.get('http://example.com/', :json => {:foo => :bar})
end
end
describe '#request' do
context 'with explicitly given `Host` header' do
let(:headers) { {'Host' => 'another.example.com'} }
let(:client) { described_class.new :headers => headers }
it 'keeps `Host` header as is' do
expect(client).to receive(:perform) do |req, options|
expect(req['Host']).to eq 'another.example.com'
end
client.request(:get, 'http://example.com/')
end
end
end
describe '#perform' do
let(:client) { described_class.new }
it 'calls finish_response before actual performance' do
TCPSocket.stub(:open) { throw :halt }
expect(client).to receive(:finish_response)
catch(:halt) { client.head "http://127.0.0.1:#{ExampleService::PORT}/" }
end
it 'calls finish_response once body was fully flushed' do
expect(client).to receive(:finish_response).twice.and_call_original
client.get("http://127.0.0.1:#{ExampleService::PORT}/").to_s
end
context 'with HEAD request' do
it 'does not iterates through body' do
expect(client).to_not receive(:readpartial)
client.head("http://127.0.0.1:#{ExampleService::PORT}/")
end
it 'finishes response after headers were received' do
expect(client).to receive(:finish_response).twice.and_call_original
client.head("http://127.0.0.1:#{ExampleService::PORT}/")
end
end
context 'when server fully flushes response in one chunk' do
before do
socket_spy = double
chunks = [
<<-RESPONSE.gsub(/^\s*\| */, '').gsub(/\n/, "\r\n")
| HTTP/1.1 200 OK
| Content-Type: text/html
| Server: WEBrick/1.3.1 (Ruby/1.9.3/2013-11-22)
| Date: Mon, 24 Mar 2014 00:32:22 GMT
| Content-Length: 15
| Connection: Keep-Alive
|
| <!doctype html>
RESPONSE
]
socket_spy.stub(:close) { nil }
socket_spy.stub(:closed?) { true }
socket_spy.stub(:readpartial) { chunks.shift }
socket_spy.stub(:<<) { nil }
TCPSocket.stub(:open) { socket_spy }
end
it 'properly reads body' do
body = client.get("http://127.0.0.1:#{ExampleService::PORT}/").to_s
expect(body).to eq '<!doctype html>'
end
end
end
end
|