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

52 lines
1.8 KiB
Ruby
Raw Normal View History

2019-12-04 20:38:33 +05:30
# frozen_string_literal: true
2017-08-17 22:00:37 +05:30
require 'spec_helper'
2020-03-13 15:44:24 +05:30
describe Gitlab::RequestContext, :request_store do
subject { described_class.instance }
2019-12-21 20:55:43 +05:30
2020-05-24 23:13:21 +05:30
before do
allow(subject).to receive(:enabled?).and_return(true)
end
2020-03-13 15:44:24 +05:30
it { is_expected.to have_attributes(client_ip: nil, start_thread_cpu_time: nil, request_start_time: nil) }
2017-08-17 22:00:37 +05:30
2020-03-13 15:44:24 +05:30
describe '#request_deadline' do
let(:request_start_time) { 1575982156.206008 }
2019-05-03 19:53:19 +05:30
2020-03-13 15:44:24 +05:30
before do
allow(subject).to receive(:request_start_time).and_return(request_start_time)
end
2019-05-03 19:53:19 +05:30
2020-03-13 15:44:24 +05:30
it "sets the time to #{Settings.gitlab.max_request_duration_seconds} seconds in the future" do
expect(subject.request_deadline).to eq(request_start_time + Settings.gitlab.max_request_duration_seconds)
expect(subject.request_deadline).to be_a(Float)
end
2019-05-03 19:53:19 +05:30
2020-03-13 15:44:24 +05:30
it 'returns nil if there is no start time' do
allow(subject).to receive(:request_start_time).and_return(nil)
2019-05-03 19:53:19 +05:30
2020-03-13 15:44:24 +05:30
expect(subject.request_deadline).to be_nil
2019-05-03 19:53:19 +05:30
end
2020-03-13 15:44:24 +05:30
end
2019-05-03 19:53:19 +05:30
2020-03-13 15:44:24 +05:30
describe '#ensure_request_deadline_not_exceeded!' do
it 'does not raise an error when there was no deadline' do
expect(subject).to receive(:request_deadline).and_return(nil)
expect { subject.ensure_deadline_not_exceeded! }.not_to raise_error
end
2017-08-17 22:00:37 +05:30
2020-03-13 15:44:24 +05:30
it 'does not raise an error if the deadline is in the future' do
allow(subject).to receive(:request_deadline).and_return(Gitlab::Metrics::System.real_time + 10)
2017-08-17 22:00:37 +05:30
2020-03-13 15:44:24 +05:30
expect { subject.ensure_deadline_not_exceeded! }.not_to raise_error
end
2017-08-17 22:00:37 +05:30
2020-03-13 15:44:24 +05:30
it 'raises an error when the deadline is in the past' do
allow(subject).to receive(:request_deadline).and_return(Gitlab::Metrics::System.real_time - 10)
2017-08-17 22:00:37 +05:30
2020-03-13 15:44:24 +05:30
expect { subject.ensure_deadline_not_exceeded! }.to raise_error(described_class::RequestDeadlineExceeded)
2017-08-17 22:00:37 +05:30
end
end
end