6d6bae66df
It assures that requests are aborted after 60 seconds, otherwise an exception is raised. This exception is logged by Sentry, also there is a Prometheus counter for measuring number of requests in each state.
58 lines
1.4 KiB
Ruby
58 lines
1.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
describe Gitlab::RackTimeoutObserver do
|
|
let(:counter) { Gitlab::Metrics::NullMetric.instance }
|
|
|
|
before do
|
|
allow(Gitlab::Metrics).to receive(:counter)
|
|
.with(any_args)
|
|
.and_return(counter)
|
|
end
|
|
|
|
describe '#callback' do
|
|
context 'when request times out' do
|
|
let(:env) do
|
|
{
|
|
::Rack::Timeout::ENV_INFO_KEY => double(state: :timed_out),
|
|
'action_dispatch.request.parameters' => {
|
|
'controller' => 'foo',
|
|
'action' => 'bar'
|
|
}
|
|
}
|
|
end
|
|
|
|
subject { described_class.new }
|
|
|
|
it 'increments timeout counter' do
|
|
expect(counter)
|
|
.to receive(:increment)
|
|
.with({ controller: 'foo', action: 'bar', route: nil, state: :timed_out })
|
|
|
|
subject.callback.call(env)
|
|
end
|
|
end
|
|
|
|
context 'when request expires' do
|
|
let(:endpoint) { double }
|
|
let(:env) do
|
|
{
|
|
::Rack::Timeout::ENV_INFO_KEY => double(state: :expired),
|
|
Grape::Env::API_ENDPOINT => endpoint
|
|
}
|
|
end
|
|
|
|
subject { described_class.new }
|
|
|
|
it 'increments timeout counter' do
|
|
allow(endpoint).to receive_message_chain('route.pattern.origin') { 'foobar' }
|
|
expect(counter)
|
|
.to receive(:increment)
|
|
.with({ controller: nil, action: nil, route: 'foobar', state: :expired })
|
|
|
|
subject.callback.call(env)
|
|
end
|
|
end
|
|
end
|
|
end
|