10e51ac5f7
Extend the support for "X-Profile-Token: <token>" to have an additional header that defines type of profile we are looking for, like: X-Profile-Mode: execution X-Profile-Mode: memory
54 lines
1.7 KiB
Ruby
54 lines
1.7 KiB
Ruby
require 'spec_helper'
|
|
|
|
describe 'Request Profiler' do
|
|
let(:user) { create(:user) }
|
|
|
|
shared_examples 'profiling a request' do |profile_type, extension|
|
|
before do
|
|
allow(Rails).to receive(:cache).and_return(ActiveSupport::Cache::MemoryStore.new)
|
|
allow(RubyProf::Profile).to receive(:profile) do |&blk|
|
|
blk.call
|
|
RubyProf::Profile.new
|
|
end
|
|
allow(MemoryProfiler).to receive(:report) do |&blk|
|
|
blk.call
|
|
MemoryProfiler.start
|
|
MemoryProfiler.stop
|
|
end
|
|
end
|
|
|
|
it 'creates a profile of the request' do
|
|
project = create(:project, namespace: user.namespace)
|
|
time = Time.now
|
|
path = "/#{project.full_path}"
|
|
|
|
Timecop.freeze(time) do
|
|
get path, params: {}, headers: { 'X-Profile-Token' => Gitlab::RequestProfiler.profile_token, 'X-Profile-Mode' => profile_type }
|
|
end
|
|
|
|
profile_type = 'execution' if profile_type.nil?
|
|
profile_path = "#{Gitlab.config.shared.path}/tmp/requests_profiles/#{path.tr('/', '|')}_#{time.to_i}_#{profile_type}.#{extension}"
|
|
expect(File.exist?(profile_path)).to be true
|
|
end
|
|
|
|
after do
|
|
Gitlab::RequestProfiler.remove_all_profiles
|
|
end
|
|
end
|
|
|
|
context "when user is logged-in" do
|
|
before do
|
|
login_as(user)
|
|
end
|
|
|
|
include_examples 'profiling a request', 'execution', 'html'
|
|
include_examples 'profiling a request', nil, 'html'
|
|
include_examples 'profiling a request', 'memory', 'txt'
|
|
end
|
|
|
|
context "when user is not logged-in" do
|
|
include_examples 'profiling a request', 'execution', 'html'
|
|
include_examples 'profiling a request', nil, 'html'
|
|
include_examples 'profiling a request', 'memory', 'txt'
|
|
end
|
|
end
|