68 lines
1.7 KiB
Ruby
68 lines
1.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
RSpec.describe Gitlab::PollingInterval do
|
|
let(:polling_interval) { described_class }
|
|
|
|
describe '.set_header' do
|
|
let(:headers) { {} }
|
|
let(:response) { double(headers: headers) }
|
|
|
|
context 'when polling is disabled' do
|
|
before do
|
|
stub_application_setting(polling_interval_multiplier: 0)
|
|
end
|
|
|
|
it 'sets value to -1' do
|
|
polling_interval.set_header(response, interval: 10_000)
|
|
|
|
expect(headers['Poll-Interval']).to eq('-1')
|
|
end
|
|
end
|
|
|
|
context 'when polling is enabled' do
|
|
before do
|
|
stub_application_setting(polling_interval_multiplier: 0.33333)
|
|
end
|
|
|
|
it 'applies modifier to base interval' do
|
|
polling_interval.set_header(response, interval: 10_000)
|
|
|
|
expect(headers['Poll-Interval']).to eq('3333')
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '.set_api_header' do
|
|
let(:context) { double(Grape::Endpoint) }
|
|
|
|
before do
|
|
allow(context).to receive(:header)
|
|
end
|
|
|
|
context 'when polling is disabled' do
|
|
before do
|
|
stub_application_setting(polling_interval_multiplier: 0)
|
|
end
|
|
|
|
it 'sets value to -1' do
|
|
expect(context).to receive(:header).with('Poll-Interval', '-1')
|
|
|
|
polling_interval.set_api_header(context, interval: 10_000)
|
|
end
|
|
end
|
|
|
|
context 'when polling is enabled' do
|
|
before do
|
|
stub_application_setting(polling_interval_multiplier: 0.33333)
|
|
end
|
|
|
|
it 'applies modifier to base interval' do
|
|
expect(context).to receive(:header).with('Poll-Interval', '3333')
|
|
|
|
polling_interval.set_api_header(context, interval: 10_000)
|
|
end
|
|
end
|
|
end
|
|
end
|