88 lines
2.7 KiB
Ruby
88 lines
2.7 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
require 'spec_helper'
|
||
|
|
|
||
|
|
describe Gitlab::Throttle do
|
||
|
|
describe '.protected_paths_enabled?' do
|
||
|
|
subject { described_class.protected_paths_enabled? }
|
||
|
|
|
||
|
|
context 'when omnibus protected paths throttle should be used' do
|
||
|
|
before do
|
||
|
|
expect(described_class).to receive(:should_use_omnibus_protected_paths?).and_return(true)
|
||
|
|
end
|
||
|
|
|
||
|
|
it { is_expected.to be_falsey }
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'when omnibus protected paths throttle should not be used' do
|
||
|
|
before do
|
||
|
|
expect(described_class).to receive(:should_use_omnibus_protected_paths?).and_return(false)
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'returns Application Settings throttle_protected_paths_enabled?' do
|
||
|
|
expect(Gitlab::CurrentSettings.current_application_settings).to receive(:throttle_protected_paths_enabled?)
|
||
|
|
|
||
|
|
subject
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
describe '.should_use_omnibus_protected_paths?' do
|
||
|
|
subject { described_class.should_use_omnibus_protected_paths? }
|
||
|
|
|
||
|
|
context 'when rack_attack.admin_area_protected_paths_enabled config is unspecified' do
|
||
|
|
context 'when the omnibus protected paths throttle has been recently used (it has data)' do
|
||
|
|
before do
|
||
|
|
expect(described_class).to receive(:omnibus_protected_paths_present?).and_return(true)
|
||
|
|
end
|
||
|
|
|
||
|
|
it { is_expected.to be_truthy }
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'when the omnibus protected paths throttle has not been recently used' do
|
||
|
|
before do
|
||
|
|
expect(described_class).to receive(:omnibus_protected_paths_present?).and_return(false)
|
||
|
|
end
|
||
|
|
|
||
|
|
it { is_expected.to be_falsey }
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'when rack_attack.admin_area_protected_paths_enabled config is false' do
|
||
|
|
before do
|
||
|
|
stub_config(rack_attack: {
|
||
|
|
admin_area_protected_paths_enabled: false
|
||
|
|
})
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'when the omnibus protected paths throttle has been recently used (it has data)' do
|
||
|
|
before do
|
||
|
|
expect(described_class).to receive(:omnibus_protected_paths_present?).and_return(true)
|
||
|
|
end
|
||
|
|
|
||
|
|
it { is_expected.to be_truthy }
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'when the omnibus protected paths throttle has not been recently used' do
|
||
|
|
before do
|
||
|
|
expect(described_class).to receive(:omnibus_protected_paths_present?).and_return(false)
|
||
|
|
end
|
||
|
|
|
||
|
|
it { is_expected.to be_falsey }
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'when rack_attack.admin_area_protected_paths_enabled config is true' do
|
||
|
|
before do
|
||
|
|
stub_config(rack_attack: {
|
||
|
|
admin_area_protected_paths_enabled: true
|
||
|
|
})
|
||
|
|
|
||
|
|
expect(described_class).not_to receive(:omnibus_protected_paths_present?)
|
||
|
|
end
|
||
|
|
|
||
|
|
it { is_expected.to be_falsey }
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|