summaryrefslogtreecommitdiff
path: root/spec/requests/request_profiler_spec.rb
blob: 7f9999bf3d20612a2432aa4988985d6457e5fe58 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
# frozen_string_literal: true

require 'spec_helper'

RSpec.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