1 files changed, 132 insertions, 0 deletions
diff --git a/spec/lib/bulk_imports/clients/http_spec.rb b/spec/lib/bulk_imports/clients/http_spec.rb
new file mode 100644
index 00000000000..2d841b7fac2
--- /dev/null
+++ b/spec/lib/bulk_imports/clients/http_spec.rb
@@ -0,0 +1,132 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe BulkImports::Clients::Http do
+  include ImportSpecHelper
+
+  let(:uri) { 'http://gitlab.example' }
+  let(:token) { 'token' }
+  let(:resource) { 'resource' }
+
+  subject { described_class.new(uri: uri, token: token) }
+
+  describe '#get' do
+    let(:response_double) { double(code: 200, success?: true, parsed_response: {}) }
+
+    shared_examples 'performs network request' do
+      it 'performs network request' do
+        expect(Gitlab::HTTP).to receive(:get).with(*expected_args).and_return(response_double)
+
+        subject.get(resource)
+      end
+    end
+
+    describe 'request query' do
+      include_examples 'performs network request' do
+        let(:expected_args) do
+          [
+            anything,
+            hash_including(
+              query: {
+                page: described_class::DEFAULT_PAGE,
+                per_page: described_class::DEFAULT_PER_PAGE
+              }
+            )
+          ]
+        end
+      end
+    end
+
+    describe 'request headers' do
+      include_examples 'performs network request' do
+        let(:expected_args) do
+          [
+            anything,
+            hash_including(
+              headers: {
+                'Content-Type' => 'application/json',
+                'Authorization' => "Bearer #{token}"
+              }
+            )
+          ]
+        end
+      end
+    end
+
+    describe 'request uri' do
+      include_examples 'performs network request' do
+        let(:expected_args) do
+          ['http://gitlab.example:80/api/v4/resource', anything]
+        end
+      end
+    end
+
+    context 'error handling' do
+      context 'when error occurred' do
+        it 'raises ConnectionError' do
+          allow(Gitlab::HTTP).to receive(:get).and_raise(Errno::ECONNREFUSED)
+
+          expect { subject.get(resource) }.to raise_exception(described_class::ConnectionError)
+        end
+      end
+
+      context 'when response is not success' do
+        it 'raises ConnectionError' do
+          response_double = double(code: 503, success?: false)
+
+          allow(Gitlab::HTTP).to receive(:get).and_return(response_double)
+
+          expect { subject.get(resource) }.to raise_exception(described_class::ConnectionError)
+        end
+      end
+    end
+
+    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
+  end
+end