2016-01-14 18:37:52 +05:30
|
|
|
require 'spec_helper'
|
|
|
|
|
|
|
|
describe Gitlab::Metrics::Subscribers::ActiveRecord do
|
2018-03-17 18:26:18 +05:30
|
|
|
let(:env) { {} }
|
|
|
|
let(:transaction) { Gitlab::Metrics::WebTransaction.new(env) }
|
2016-01-14 18:37:52 +05:30
|
|
|
let(:subscriber) { described_class.new }
|
|
|
|
|
|
|
|
let(:event) do
|
2018-03-17 18:26:18 +05:30
|
|
|
double(:event, duration: 2,
|
2016-01-14 18:37:52 +05:30
|
|
|
payload: { sql: 'SELECT * FROM users WHERE id = 10' })
|
|
|
|
end
|
|
|
|
|
|
|
|
describe '#sql' do
|
|
|
|
describe 'without a current transaction' do
|
|
|
|
it 'simply returns' do
|
2017-09-10 17:25:29 +05:30
|
|
|
expect_any_instance_of(Gitlab::Metrics::Transaction)
|
|
|
|
.not_to receive(:increment)
|
2016-01-14 18:37:52 +05:30
|
|
|
|
|
|
|
subscriber.sql(event)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
describe 'with a current transaction' do
|
2018-03-17 18:26:18 +05:30
|
|
|
it 'observes sql_duration metric' do
|
|
|
|
expect(subscriber).to receive(:current_transaction)
|
|
|
|
.at_least(:once)
|
|
|
|
.and_return(transaction)
|
|
|
|
expect(described_class.send(:gitlab_sql_duration_seconds)).to receive(:observe).with({}, 0.002)
|
|
|
|
subscriber.sql(event)
|
|
|
|
end
|
|
|
|
|
2016-01-14 18:37:52 +05:30
|
|
|
it 'increments the :sql_duration value' do
|
2017-09-10 17:25:29 +05:30
|
|
|
expect(subscriber).to receive(:current_transaction)
|
|
|
|
.at_least(:once)
|
|
|
|
.and_return(transaction)
|
2016-01-14 18:37:52 +05:30
|
|
|
|
2017-09-10 17:25:29 +05:30
|
|
|
expect(transaction).to receive(:increment)
|
2018-03-17 18:26:18 +05:30
|
|
|
.with(:sql_duration, 2, false)
|
2016-01-14 18:37:52 +05:30
|
|
|
|
2017-09-10 17:25:29 +05:30
|
|
|
expect(transaction).to receive(:increment)
|
2018-03-17 18:26:18 +05:30
|
|
|
.with(:sql_count, 1, false)
|
2016-06-02 11:05:42 +05:30
|
|
|
|
2016-01-14 18:37:52 +05:30
|
|
|
subscriber.sql(event)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|