48 lines
1.3 KiB
Ruby
48 lines
1.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'rake_helper'
|
|
|
|
RSpec.describe 'gitlab:usage data take tasks', :silence_stdout do
|
|
include UsageDataHelpers
|
|
|
|
let(:metrics_file) { Rails.root.join('tmp', 'test', 'sql_metrics_queries.json') }
|
|
|
|
before do
|
|
Rake.application.rake_require 'tasks/gitlab/usage_data'
|
|
|
|
# stub prometheus external http calls https://gitlab.com/gitlab-org/gitlab/-/issues/245277
|
|
stub_prometheus_queries
|
|
stub_database_flavor_check
|
|
end
|
|
|
|
after do
|
|
FileUtils.rm_rf(metrics_file)
|
|
end
|
|
|
|
describe 'dump_sql_in_yaml' do
|
|
it 'dumps SQL queries in yaml format' do
|
|
expect { run_rake_task('gitlab:usage_data:dump_sql_in_yaml') }.to output(/.*recorded_at:.*/).to_stdout
|
|
end
|
|
end
|
|
|
|
describe 'dump_sql_in_json' do
|
|
it 'dumps SQL queries in json format' do
|
|
expect { run_rake_task('gitlab:usage_data:dump_sql_in_json') }.to output(/.*"recorded_at":.*/).to_stdout
|
|
end
|
|
end
|
|
|
|
describe 'dump_non_sql_in_json' do
|
|
it 'dumps non SQL data in json format' do
|
|
expect { run_rake_task('gitlab:usage_data:dump_non_sql_in_json') }.to output(/.*"recorded_at":.*/).to_stdout
|
|
end
|
|
end
|
|
|
|
describe 'generate_sql_metrics_fixture' do
|
|
it 'generates fixture file correctly' do
|
|
run_rake_task('gitlab:usage_data:generate_sql_metrics_queries')
|
|
|
|
expect(Pathname.new(metrics_file)).to exist
|
|
end
|
|
end
|
|
end
|