diff options
author | Jan Provaznik <jprovaznik@gitlab.com> | 2019-06-05 20:20:37 +0300 |
---|---|---|
committer | Jan Provaznik <jprovaznik@gitlab.com> | 2019-06-12 13:09:04 +0300 |
commit | d3a7bdda986949ca76df3c4932ee8c973437a743 (patch) | |
tree | f5bef987fa8c1a6c4f9946eda47ebb6ffba82e63 /spec/lib/gitlab/cluster | |
parent | f18f3ed9265e777ab7ddcb87948d0e405fbbb0b0 (diff) |
Moved RackTimeout observer to a different location
Because there will be similar observer for PumaWorkerKiller,
it makes sense to keep both on better place.
Diffstat (limited to 'spec/lib/gitlab/cluster')
-rw-r--r-- | spec/lib/gitlab/cluster/rack_timeout_observer_spec.rb | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/spec/lib/gitlab/cluster/rack_timeout_observer_spec.rb b/spec/lib/gitlab/cluster/rack_timeout_observer_spec.rb new file mode 100644 index 00000000000..2adb1d7cc71 --- /dev/null +++ b/spec/lib/gitlab/cluster/rack_timeout_observer_spec.rb @@ -0,0 +1,58 @@ +# 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 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 |