debian-mirror-gitlab/spec/lib/bulk_imports/clients/http_spec.rb

131 lines
3.7 KiB
Ruby
Raw Normal View History

2021-01-03 14:25:43 +05:30
# frozen_string_literal: true
require 'spec_helper'
2021-01-29 00:20:46 +05:30
RSpec.describe BulkImports::Clients::Http do
2021-01-03 14:25:43 +05:30
include ImportSpecHelper
let(:uri) { 'http://gitlab.example' }
let(:token) { 'token' }
let(:resource) { 'resource' }
2021-06-08 01:23:25 +05:30
let(:response_double) { double(code: 200, success?: true, parsed_response: {}) }
2021-01-03 14:25:43 +05:30
subject { described_class.new(uri: uri, token: token) }
2021-06-08 01:23:25 +05:30
shared_examples 'performs network request' do
it 'performs network request' do
expect(Gitlab::HTTP).to receive(method).with(*expected_args).and_return(response_double)
2021-01-03 14:25:43 +05:30
2021-06-08 01:23:25 +05:30
subject.public_send(method, resource)
2021-01-03 14:25:43 +05:30
end
context 'error handling' do
context 'when error occurred' do
it 'raises ConnectionError' do
2021-06-08 01:23:25 +05:30
allow(Gitlab::HTTP).to receive(method).and_raise(Errno::ECONNREFUSED)
2021-01-03 14:25:43 +05:30
2021-06-08 01:23:25 +05:30
expect { subject.public_send(method, resource) }.to raise_exception(described_class::ConnectionError)
2021-01-03 14:25:43 +05:30
end
end
context 'when response is not success' do
it 'raises ConnectionError' do
response_double = double(code: 503, success?: false)
2021-06-08 01:23:25 +05:30
allow(Gitlab::HTTP).to receive(method).and_return(response_double)
2021-01-03 14:25:43 +05:30
2021-06-08 01:23:25 +05:30
expect { subject.public_send(method, resource) }.to raise_exception(described_class::ConnectionError)
2021-01-03 14:25:43 +05:30
end
end
end
2021-06-08 01:23:25 +05:30
end
describe '#get' do
let(:method) { :get }
include_examples 'performs network request' do
let(:expected_args) do
[
'http://gitlab.example:80/api/v4/resource',
hash_including(
query: {
page: described_class::DEFAULT_PAGE,
per_page: described_class::DEFAULT_PER_PAGE
},
headers: {
'Content-Type' => 'application/json',
'Authorization' => "Bearer #{token}"
}
)
]
end
end
2021-01-29 00:20:46 +05:30
describe '#each_page' do
let(:objects1) { [{ object: 1 }, { object: 2 }] }
let(:objects2) { [{ object: 3 }, { object: 4 }] }
let(:response1) { double(success?: true, headers: { 'x-next-page' => 2 }, parsed_response: objects1) }
let(:response2) { double(success?: true, headers: {}, parsed_response: objects2) }
before do
stub_http_get('groups', { page: 1, per_page: 30 }, response1)
stub_http_get('groups', { page: 2, per_page: 30 }, response2)
end
context 'with a block' do
it 'yields every retrieved page to the supplied block' do
pages = []
subject.each_page(:get, 'groups') { |page| pages << page }
expect(pages[0]).to be_an_instance_of(Array)
expect(pages[1]).to be_an_instance_of(Array)
expect(pages[0]).to eq(objects1)
expect(pages[1]).to eq(objects2)
end
end
context 'without a block' do
it 'returns an Enumerator' do
expect(subject.each_page(:get, :foo)).to be_an_instance_of(Enumerator)
end
end
private
def stub_http_get(path, query, response)
uri = "http://gitlab.example:80/api/v4/#{path}"
params = {
follow_redirects: false,
headers: {
"Authorization" => "Bearer token",
"Content-Type" => "application/json"
}
}.merge(query: query)
allow(Gitlab::HTTP).to receive(:get).with(uri, params).and_return(response)
end
end
2021-01-03 14:25:43 +05:30
end
2021-06-08 01:23:25 +05:30
describe '#post' do
let(:method) { :post }
include_examples 'performs network request' do
let(:expected_args) do
[
'http://gitlab.example:80/api/v4/resource',
hash_including(
body: {},
headers: {
'Content-Type' => 'application/json',
'Authorization' => "Bearer #{token}"
}
)
]
end
end
end
2021-01-03 14:25:43 +05:30
end