debian-mirror-gitlab/spec/lib/gitlab/exclusive_lease_helpers_spec.rb

120 lines
3.9 KiB
Ruby
Raw Normal View History

2019-12-26 22:10:19 +05:30
# frozen_string_literal: true
2018-11-08 19:23:39 +05:30
require 'spec_helper'
2020-07-28 23:09:34 +05:30
RSpec.describe Gitlab::ExclusiveLeaseHelpers, :clean_gitlab_redis_shared_state do
2018-11-08 19:23:39 +05:30
include ::ExclusiveLeaseHelpers
let(:class_instance) { (Class.new { include ::Gitlab::ExclusiveLeaseHelpers }).new }
let(:unique_key) { SecureRandom.hex(10) }
describe '#in_lock' do
subject { class_instance.in_lock(unique_key, **options) { } }
let(:options) { {} }
2019-02-15 15:39:39 +05:30
context 'when unique key is not set' do
let(:unique_key) { }
it 'raises an error' do
expect { subject }.to raise_error ArgumentError
end
end
2018-11-08 19:23:39 +05:30
context 'when the lease is not obtained yet' do
2020-05-24 23:13:21 +05:30
let!(:lease) { stub_exclusive_lease(unique_key, 'uuid') }
2018-11-08 19:23:39 +05:30
it 'calls the given block' do
2021-01-03 14:25:43 +05:30
expect { |b| class_instance.in_lock(unique_key, &b) }
.to yield_with_args(false, an_instance_of(described_class::SleepingLock))
2018-11-08 19:23:39 +05:30
end
it 'calls the given block continuously' do
2021-01-03 14:25:43 +05:30
expect { |b| class_instance.in_lock(unique_key, &b) }
.to yield_with_args(false, an_instance_of(described_class::SleepingLock))
expect { |b| class_instance.in_lock(unique_key, &b) }
.to yield_with_args(false, an_instance_of(described_class::SleepingLock))
expect { |b| class_instance.in_lock(unique_key, &b) }
.to yield_with_args(false, an_instance_of(described_class::SleepingLock))
2018-11-08 19:23:39 +05:30
end
it 'cancels the exclusive lease after the block' do
2020-05-24 23:13:21 +05:30
expect(lease).to receive(:cancel).once
2018-11-08 19:23:39 +05:30
subject
end
end
context 'when the lease is obtained already' do
let!(:lease) { stub_exclusive_lease_taken(unique_key) }
it 'retries to obtain a lease and raises an error' do
expect(lease).to receive(:try_obtain).exactly(11).times
expect { subject }.to raise_error('Failed to obtain a lock')
end
context 'when ttl is specified' do
let(:options) { { ttl: 10.minutes } }
it 'receives the specified argument' do
expect(Gitlab::ExclusiveLease).to receive(:new).with(unique_key, { timeout: 10.minutes } )
expect { subject }.to raise_error('Failed to obtain a lock')
end
end
context 'when retry count is specified' do
let(:options) { { retries: 3 } }
it 'retries for the specified times' do
expect(lease).to receive(:try_obtain).exactly(4).times
expect { subject }.to raise_error('Failed to obtain a lock')
end
2019-09-30 21:07:59 +05:30
context 'when lease is granted after retry' do
it 'yields block with true' do
expect(lease).to receive(:try_obtain).exactly(3).times { nil }
expect(lease).to receive(:try_obtain).once { unique_key }
2021-01-03 14:25:43 +05:30
expect { |b| class_instance.in_lock(unique_key, &b) }
.to yield_with_args(true, an_instance_of(described_class::SleepingLock))
2019-09-30 21:07:59 +05:30
end
end
2018-11-08 19:23:39 +05:30
end
2020-05-24 23:13:21 +05:30
context 'when we specify no retries' do
let(:options) { { retries: 0 } }
it 'never sleeps' do
expect(class_instance).not_to receive(:sleep)
expect { subject }.to raise_error('Failed to obtain a lock')
end
end
2018-11-08 19:23:39 +05:30
context 'when sleep second is specified' do
2020-05-24 23:13:21 +05:30
let(:options) { { retries: 1, sleep_sec: 0.05.seconds } }
it 'receives the specified argument' do
expect_any_instance_of(Object).to receive(:sleep).with(0.05.seconds).once
expect { subject }.to raise_error('Failed to obtain a lock')
end
end
context 'when sleep second is specified as a lambda' do
let(:options) { { retries: 2, sleep_sec: ->(num) { 0.1 + num } } }
2018-11-08 19:23:39 +05:30
it 'receives the specified argument' do
2020-05-24 23:13:21 +05:30
expect_any_instance_of(Object).to receive(:sleep).with(1.1.seconds).once
expect_any_instance_of(Object).to receive(:sleep).with(2.1.seconds).once
2018-11-08 19:23:39 +05:30
expect { subject }.to raise_error('Failed to obtain a lock')
end
end
end
end
end