103 lines
3.3 KiB
Ruby
103 lines
3.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
RSpec.describe Bitbucket::Connection, feature_category: :integrations do
|
|
let(:token) { 'token' }
|
|
|
|
before do
|
|
allow_next_instance_of(described_class) do |instance|
|
|
allow(instance).to receive(:provider).and_return(double(app_id: '', app_secret: ''))
|
|
end
|
|
end
|
|
|
|
describe '#get' do
|
|
it 'calls OAuth2::AccessToken::get' do
|
|
expected_client_options = {
|
|
site: OmniAuth::Strategies::Bitbucket.default_options[:client_options]['site'],
|
|
authorize_url: OmniAuth::Strategies::Bitbucket.default_options[:client_options]['authorize_url'],
|
|
token_url: OmniAuth::Strategies::Bitbucket.default_options[:client_options]['token_url']
|
|
}
|
|
|
|
expect_next_instance_of(described_class) do |instance|
|
|
expect(instance).to receive(:retry_with_exponential_backoff).and_call_original
|
|
end
|
|
|
|
expect(OAuth2::Client)
|
|
.to receive(:new)
|
|
.with(anything, anything, expected_client_options)
|
|
|
|
expect_next_instance_of(OAuth2::AccessToken) do |instance|
|
|
expect(instance).to receive(:get).and_return(double(parsed: true))
|
|
end
|
|
|
|
connection = described_class.new({ token: token })
|
|
|
|
connection.get('/users')
|
|
end
|
|
|
|
context 'when the API returns an error' do
|
|
before do
|
|
allow_next_instance_of(OAuth2::AccessToken) do |instance|
|
|
allow(instance).to receive(:get).and_raise(OAuth2::Error, 'some error')
|
|
end
|
|
|
|
stub_const('Bitbucket::ExponentialBackoff::INITIAL_DELAY', 0.0)
|
|
allow(Random).to receive(:rand).and_return(0.001)
|
|
end
|
|
|
|
it 'logs the retries and raises an error if it does not succeed on retry' do
|
|
expect(Gitlab::BitbucketImport::Logger).to receive(:info)
|
|
.with(message: 'Retrying in 0.0 seconds due to some error')
|
|
.twice
|
|
|
|
connection = described_class.new({ token: token })
|
|
|
|
expect { connection.get('/users') }.to raise_error(Bitbucket::ExponentialBackoff::RateLimitError)
|
|
end
|
|
end
|
|
|
|
context 'when the bitbucket_importer_exponential_backoff feature flag is disabled' do
|
|
before do
|
|
stub_feature_flags(bitbucket_importer_exponential_backoff: false)
|
|
end
|
|
|
|
it 'does not run with exponential backoff' do
|
|
expect_next_instance_of(described_class) do |instance|
|
|
expect(instance).not_to receive(:retry_with_exponential_backoff).and_call_original
|
|
end
|
|
|
|
expect_next_instance_of(OAuth2::AccessToken) do |instance|
|
|
expect(instance).to receive(:get).and_return(double(parsed: true))
|
|
end
|
|
|
|
connection = described_class.new({ token: token })
|
|
|
|
connection.get('/users')
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#expired?' do
|
|
it 'calls connection.expired?' do
|
|
expect_next_instance_of(OAuth2::AccessToken) do |instance|
|
|
expect(instance).to receive(:expired?).and_return(true)
|
|
end
|
|
|
|
expect(described_class.new({ token: token }).expired?).to be_truthy
|
|
end
|
|
end
|
|
|
|
describe '#refresh!' do
|
|
it 'calls connection.refresh!' do
|
|
response = double(token: token, expires_at: nil, expires_in: nil, refresh_token: nil)
|
|
|
|
expect_next_instance_of(OAuth2::AccessToken) do |instance|
|
|
expect(instance).to receive(:refresh!).and_return(response)
|
|
end
|
|
|
|
described_class.new({ token: token }).refresh!
|
|
end
|
|
end
|
|
end
|