78 lines
1.9 KiB
Ruby
78 lines
1.9 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
require 'spec_helper'
|
||
|
|
|
||
|
|
describe Gitlab::Cluster::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 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 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
|
||
|
|
|
||
|
|
context 'when request is being processed' do
|
||
|
|
let(:endpoint) { double }
|
||
|
|
let(:env) do
|
||
|
|
{
|
||
|
|
::Rack::Timeout::ENV_INFO_KEY => double(state: :active),
|
||
|
|
Grape::Env::API_ENDPOINT => endpoint
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
subject { described_class.new }
|
||
|
|
|
||
|
|
it 'does not increment counter' do
|
||
|
|
allow(endpoint).to receive_message_chain('route.pattern.origin') { 'foobar' }
|
||
|
|
expect(counter).not_to receive(:increment)
|
||
|
|
|
||
|
|
subject.callback.call(env)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|