debian-mirror-gitlab/spec/lib/gitlab/metrics/requests_rack_middleware_spec.rb

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

393 lines
15 KiB
Ruby
Raw Normal View History

2019-12-04 20:38:33 +05:30
# frozen_string_literal: true
2017-09-10 17:25:29 +05:30
require 'spec_helper'
2021-01-29 00:20:46 +05:30
RSpec.describe Gitlab::Metrics::RequestsRackMiddleware, :aggregate_failures do
2017-09-10 17:25:29 +05:30
let(:app) { double('app') }
2020-01-01 13:55:28 +05:30
2017-09-10 17:25:29 +05:30
subject { described_class.new(app) }
2021-09-04 01:27:46 +05:30
around do |example|
# Simulate application context middleware
# In fact, this middleware cleans up the contexts after a request lifecycle
::Gitlab::ApplicationContext.with_context({}) do
example.run
end
end
2017-09-10 17:25:29 +05:30
describe '#call' do
2021-09-04 01:27:46 +05:30
let(:status) { 200 }
2017-09-10 17:25:29 +05:30
let(:env) { { 'REQUEST_METHOD' => 'GET' } }
let(:stack_result) { [status, {}, 'body'] }
before do
allow(app).to receive(:call).and_return(stack_result)
end
context '@app.call succeeds with 200' do
before do
allow(app).to receive(:call).and_return([200, nil, nil])
end
2018-03-17 18:26:18 +05:30
RSpec::Matchers.define :a_positive_execution_time do
match { |actual| actual > 0 }
end
2017-09-10 17:25:29 +05:30
2021-01-29 00:20:46 +05:30
it 'tracks request count and duration' do
expect(described_class).to receive_message_chain(:http_requests_total, :increment).with(method: 'get', status: '200', feature_category: 'unknown')
2021-01-03 14:25:43 +05:30
expect(described_class).to receive_message_chain(:http_request_duration_seconds, :observe).with({ method: 'get' }, a_positive_execution_time)
2021-12-11 22:18:48 +05:30
expect(Gitlab::Metrics::RailsSlis.request_apdex).to receive(:increment)
.with(labels: { feature_category: 'unknown', endpoint_id: 'unknown', request_urgency: :default }, success: true)
2017-09-10 17:25:29 +05:30
2021-01-29 00:20:46 +05:30
subject.call(env)
2017-09-10 17:25:29 +05:30
end
2020-04-22 19:07:51 +05:30
context 'request is a health check endpoint' do
2021-01-03 14:25:43 +05:30
['/-/liveness', '/-/liveness/', '/-/%6D%65%74%72%69%63%73'].each do |path|
context "when path is #{path}" do
before do
env['PATH_INFO'] = path
end
2020-04-22 19:07:51 +05:30
2021-01-29 00:20:46 +05:30
it 'increments health endpoint counter rather than overall counter and does not record duration' do
2021-01-03 14:25:43 +05:30
expect(described_class).not_to receive(:http_request_duration_seconds)
2021-01-29 00:20:46 +05:30
expect(described_class).not_to receive(:http_requests_total)
expect(described_class).to receive_message_chain(:http_health_requests_total, :increment).with(method: 'get', status: '200')
2020-04-22 19:07:51 +05:30
2021-01-03 14:25:43 +05:30
subject.call(env)
end
2020-04-22 19:07:51 +05:30
end
end
end
context 'request is not a health check endpoint' do
2021-01-03 14:25:43 +05:30
['/-/ordinary-requests', '/-/', '/-/health/subpath'].each do |path|
context "when path is #{path}" do
before do
env['PATH_INFO'] = path
end
2021-01-29 00:20:46 +05:30
it 'increments regular counters and tracks duration' do
expect(described_class).to receive_message_chain(:http_requests_total, :increment).with(method: 'get', status: '200', feature_category: 'unknown')
2021-01-03 14:25:43 +05:30
expect(described_class).not_to receive(:http_health_requests_total)
expect(described_class)
.to receive_message_chain(:http_request_duration_seconds, :observe)
2021-11-18 22:05:49 +05:30
.with({ method: 'get' }, a_positive_execution_time)
2021-01-03 14:25:43 +05:30
subject.call(env)
end
2020-04-22 19:07:51 +05:30
end
end
end
2017-09-10 17:25:29 +05:30
end
2021-09-04 01:27:46 +05:30
context '@app.call returns an error code' do
let(:status) { '500' }
2021-11-18 22:05:49 +05:30
it 'tracks count but not duration or apdex' do
2021-09-04 01:27:46 +05:30
expect(described_class).to receive_message_chain(:http_requests_total, :increment).with(method: 'get', status: '500', feature_category: 'unknown')
expect(described_class).not_to receive(:http_request_duration_seconds)
2021-11-18 22:05:49 +05:30
expect(Gitlab::Metrics::RailsSlis).not_to receive(:request_apdex)
2021-09-04 01:27:46 +05:30
subject.call(env)
end
end
2017-09-10 17:25:29 +05:30
context '@app.call throws exception' do
let(:http_request_duration_seconds) { double('http_request_duration_seconds') }
2021-01-29 00:20:46 +05:30
let(:http_requests_total) { double('http_requests_total') }
2017-09-10 17:25:29 +05:30
before do
allow(app).to receive(:call).and_raise(StandardError)
allow(described_class).to receive(:http_request_duration_seconds).and_return(http_request_duration_seconds)
2021-01-29 00:20:46 +05:30
allow(described_class).to receive(:http_requests_total).and_return(http_requests_total)
2017-09-10 17:25:29 +05:30
end
2021-01-29 00:20:46 +05:30
it 'tracks the correct metrics' do
2017-09-10 17:25:29 +05:30
expect(described_class).to receive_message_chain(:rack_uncaught_errors_count, :increment)
2021-01-29 00:20:46 +05:30
expect(described_class).to receive_message_chain(:http_requests_total, :increment).with(method: 'get', status: 'undefined', feature_category: 'unknown')
expect(described_class.http_request_duration_seconds).not_to receive(:observe)
2021-11-18 22:05:49 +05:30
expect(Gitlab::Metrics::RailsSlis).not_to receive(:request_apdex)
2017-09-10 17:25:29 +05:30
expect { subject.call(env) }.to raise_error(StandardError)
end
2021-01-29 00:20:46 +05:30
end
2017-09-10 17:25:29 +05:30
2021-11-18 22:05:49 +05:30
context 'application context' do
context 'when a context is present' do
2021-01-29 00:20:46 +05:30
before do
2021-12-11 22:18:48 +05:30
::Gitlab::ApplicationContext.push(feature_category: 'team_planning', caller_id: 'IssuesController#show')
2021-01-29 00:20:46 +05:30
end
2017-09-10 17:25:29 +05:30
2021-11-18 22:05:49 +05:30
it 'adds the feature category to the labels for required metrics' do
2021-12-11 22:18:48 +05:30
expect(described_class).to receive_message_chain(:http_requests_total, :increment).with(method: 'get', status: '200', feature_category: 'team_planning')
2021-01-29 00:20:46 +05:30
expect(described_class).not_to receive(:http_health_requests_total)
2021-11-18 22:05:49 +05:30
expect(Gitlab::Metrics::RailsSlis.request_apdex)
2021-12-11 22:18:48 +05:30
.to receive(:increment).with(labels: { feature_category: 'team_planning', endpoint_id: 'IssuesController#show', request_urgency: :default }, success: true)
2017-09-10 17:25:29 +05:30
2021-01-29 00:20:46 +05:30
subject.call(env)
end
2019-12-21 20:55:43 +05:30
2021-01-29 00:20:46 +05:30
it 'does not record a feature category for health check endpoints' do
env['PATH_INFO'] = '/-/liveness'
2021-01-03 14:25:43 +05:30
2021-01-29 00:20:46 +05:30
expect(described_class).to receive_message_chain(:http_health_requests_total, :increment).with(method: 'get', status: '200')
expect(described_class).not_to receive(:http_requests_total)
2021-11-18 22:05:49 +05:30
expect(Gitlab::Metrics::RailsSlis).not_to receive(:request_apdex)
2021-01-03 14:25:43 +05:30
2021-01-29 00:20:46 +05:30
subject.call(env)
end
2021-01-03 14:25:43 +05:30
end
2021-09-04 01:27:46 +05:30
context 'when application raises an exception when the feature category context is present' do
2021-01-29 00:20:46 +05:30
before do
2021-12-11 22:18:48 +05:30
::Gitlab::ApplicationContext.push(feature_category: 'team_planning')
2021-09-04 01:27:46 +05:30
allow(app).to receive(:call).and_raise(StandardError)
end
it 'adds the feature category to the labels for http_requests_total' do
2021-12-11 22:18:48 +05:30
expect(described_class).to receive_message_chain(:http_requests_total, :increment).with(method: 'get', status: 'undefined', feature_category: 'team_planning')
2021-11-18 22:05:49 +05:30
expect(Gitlab::Metrics::RailsSlis).not_to receive(:request_apdex)
2021-09-04 01:27:46 +05:30
expect { subject.call(env) }.to raise_error(StandardError)
2021-01-29 00:20:46 +05:30
end
2021-09-04 01:27:46 +05:30
end
2021-01-03 14:25:43 +05:30
2021-11-18 22:05:49 +05:30
context 'when the context is not available' do
it 'sets the required labels to unknown' do
2021-01-29 00:20:46 +05:30
expect(described_class).to receive_message_chain(:http_requests_total, :increment).with(method: 'get', status: '200', feature_category: 'unknown')
expect(described_class).not_to receive(:http_health_requests_total)
2021-12-11 22:18:48 +05:30
expect(Gitlab::Metrics::RailsSlis.request_apdex).to receive(:increment)
.with(labels: { feature_category: 'unknown', endpoint_id: 'unknown', request_urgency: :default }, success: true)
2021-01-03 14:25:43 +05:30
2021-01-29 00:20:46 +05:30
subject.call(env)
end
2021-01-03 14:25:43 +05:30
end
2021-11-18 22:05:49 +05:30
context 'SLI satisfactory' do
where(:request_urgency_name, :duration, :success) do
[
[:high, 0.1, true],
[:high, 0.25, false],
[:high, 0.3, false],
[:medium, 0.3, true],
[:medium, 0.5, false],
[:medium, 0.6, false],
[:default, 0.6, true],
[:default, 1.0, false],
[:default, 1.2, false],
[:low, 4.5, true],
[:low, 5.0, false],
[:low, 6, false]
]
end
with_them do
context 'Grape API handler having expected duration setup' do
let(:api_handler) do
request_urgency = request_urgency_name
Class.new(::API::Base) do
feature_category :hello_world, ['/projects/:id/archive']
urgency request_urgency, ['/projects/:id/archive']
end
end
let(:endpoint) do
route = double(:route, request_method: 'GET', path: '/:version/projects/:id/archive(.:format)')
double(:endpoint, route: route,
options: { for: api_handler, path: [":id/archive"] },
namespace: "/projects")
end
let(:env) { { 'api.endpoint' => endpoint, 'REQUEST_METHOD' => 'GET' } }
before do
::Gitlab::ApplicationContext.push(feature_category: 'hello_world', caller_id: 'GET /projects/:id/archive')
allow(Gitlab::Metrics::System).to receive(:monotonic_time).and_return(100, 100 + duration)
end
it "captures SLI metrics" do
expect(Gitlab::Metrics::RailsSlis.request_apdex).to receive(:increment).with(
2021-12-11 22:18:48 +05:30
labels: {
feature_category: 'hello_world',
endpoint_id: 'GET /projects/:id/archive',
request_urgency: request_urgency_name
},
2021-11-18 22:05:49 +05:30
success: success
)
subject.call(env)
end
end
context 'Rails controller having expected duration setup' do
let(:controller) do
request_urgency = request_urgency_name
Class.new(ApplicationController) do
feature_category :hello_world, [:index, :show]
urgency request_urgency, [:index, :show]
end
end
let(:env) do
controller_instance = controller.new
controller_instance.action_name = :index
{ 'action_controller.instance' => controller_instance, 'REQUEST_METHOD' => 'GET' }
end
before do
::Gitlab::ApplicationContext.push(feature_category: 'hello_world', caller_id: 'AnonymousController#index')
allow(Gitlab::Metrics::System).to receive(:monotonic_time).and_return(100, 100 + duration)
end
it "captures SLI metrics" do
expect(Gitlab::Metrics::RailsSlis.request_apdex).to receive(:increment).with(
2021-12-11 22:18:48 +05:30
labels: {
feature_category: 'hello_world',
endpoint_id: 'AnonymousController#index',
request_urgency: request_urgency_name
},
2021-11-18 22:05:49 +05:30
success: success
)
subject.call(env)
end
end
end
context 'Grape API without expected duration' do
let(:endpoint) do
route = double(:route, request_method: 'GET', path: '/:version/projects/:id/archive(.:format)')
double(:endpoint, route: route,
options: { for: api_handler, path: [":id/archive"] },
namespace: "/projects")
end
let(:env) { { 'api.endpoint' => endpoint, 'REQUEST_METHOD' => 'GET' } }
let(:api_handler) { Class.new(::API::Base) }
2021-12-11 22:18:48 +05:30
it "falls back request's expectation to default (1 second)" do
2021-11-18 22:05:49 +05:30
allow(Gitlab::Metrics::System).to receive(:monotonic_time).and_return(100, 100.9)
expect(Gitlab::Metrics::RailsSlis.request_apdex).to receive(:increment).with(
2021-12-11 22:18:48 +05:30
labels: {
feature_category: 'unknown',
endpoint_id: 'unknown',
request_urgency: :default
},
2021-11-18 22:05:49 +05:30
success: true
)
subject.call(env)
allow(Gitlab::Metrics::System).to receive(:monotonic_time).and_return(100, 101)
expect(Gitlab::Metrics::RailsSlis.request_apdex).to receive(:increment).with(
2021-12-11 22:18:48 +05:30
labels: {
feature_category: 'unknown',
endpoint_id: 'unknown',
request_urgency: :default
},
2021-11-18 22:05:49 +05:30
success: false
)
subject.call(env)
end
end
context 'Rails controller without expected duration' do
let(:controller) { Class.new(ApplicationController) }
let(:env) do
controller_instance = controller.new
controller_instance.action_name = :index
{ 'action_controller.instance' => controller_instance, 'REQUEST_METHOD' => 'GET' }
end
2021-12-11 22:18:48 +05:30
it "falls back request's expectation to default (1 second)" do
2021-11-18 22:05:49 +05:30
allow(Gitlab::Metrics::System).to receive(:monotonic_time).and_return(100, 100.9)
expect(Gitlab::Metrics::RailsSlis.request_apdex).to receive(:increment).with(
2021-12-11 22:18:48 +05:30
labels: {
feature_category: 'unknown',
endpoint_id: 'unknown',
request_urgency: :default
},
2021-11-18 22:05:49 +05:30
success: true
)
subject.call(env)
allow(Gitlab::Metrics::System).to receive(:monotonic_time).and_return(100, 101)
expect(Gitlab::Metrics::RailsSlis.request_apdex).to receive(:increment).with(
2021-12-11 22:18:48 +05:30
labels: {
feature_category: 'unknown',
endpoint_id: 'unknown',
request_urgency: :default
},
2021-11-18 22:05:49 +05:30
success: false
)
subject.call(env)
end
end
context 'An unknown request' do
let(:env) do
{ 'REQUEST_METHOD' => 'GET' }
end
2021-12-11 22:18:48 +05:30
it "falls back request's expectation to default (1 second)" do
2021-11-18 22:05:49 +05:30
allow(Gitlab::Metrics::System).to receive(:monotonic_time).and_return(100, 100.9)
expect(Gitlab::Metrics::RailsSlis.request_apdex).to receive(:increment).with(
2021-12-11 22:18:48 +05:30
labels: {
feature_category: 'unknown',
endpoint_id: 'unknown',
request_urgency: :default
},
2021-11-18 22:05:49 +05:30
success: true
)
subject.call(env)
allow(Gitlab::Metrics::System).to receive(:monotonic_time).and_return(100, 101)
expect(Gitlab::Metrics::RailsSlis.request_apdex).to receive(:increment).with(
2021-12-11 22:18:48 +05:30
labels: {
feature_category: 'unknown',
endpoint_id: 'unknown',
request_urgency: :default
},
2021-11-18 22:05:49 +05:30
success: false
)
subject.call(env)
end
end
end
2021-01-03 14:25:43 +05:30
end
2021-01-29 00:20:46 +05:30
describe '.initialize_metrics', :prometheus do
it "sets labels for http_requests_total" do
2019-12-21 20:55:43 +05:30
expected_labels = []
2021-01-29 00:20:46 +05:30
described_class::HTTP_METHODS.each do |method, statuses|
statuses.each do |status|
described_class::FEATURE_CATEGORIES_TO_INITIALIZE.each do |feature_category|
expected_labels << { method: method.to_s, status: status.to_s, feature_category: feature_category.to_s }
end
end
2019-12-21 20:55:43 +05:30
end
2021-01-29 00:20:46 +05:30
described_class.initialize_metrics
expect(described_class.http_requests_total.values.keys).to contain_exactly(*expected_labels)
end
it 'sets labels for http_request_duration_seconds' do
expected_labels = described_class::HTTP_METHODS.keys.map { |method| { method: method } }
described_class.initialize_metrics
2019-12-21 20:55:43 +05:30
expect(described_class.http_request_duration_seconds.values.keys).to include(*expected_labels)
end
2021-01-29 00:20:46 +05:30
it 'has every label in config/feature_categories.yml' do
2021-11-18 22:05:49 +05:30
defaults = [::Gitlab::FeatureCategories::FEATURE_CATEGORY_DEFAULT, 'not_owned']
feature_categories = Gitlab::FeatureCategories.default.categories + defaults
2021-01-29 00:20:46 +05:30
expect(described_class::FEATURE_CATEGORIES_TO_INITIALIZE).to all(be_in(feature_categories))
end
2019-12-21 20:55:43 +05:30
end
2017-09-10 17:25:29 +05:30
end
end