summaryrefslogtreecommitdiff
path: root/spec/lib/peek/views/rugged_spec.rb
blob: d07d6b51a1fc7392059a1086cd3c523696467b3f (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
# frozen_string_literal: true

require 'spec_helper'

describe Peek::Views::Rugged, :request_store do
  subject { described_class.new }

  let(:project) { create(:project) }

  before do
    allow(Gitlab::RuggedInstrumentation).to receive(:peek_enabled?).and_return(true)
  end

  it 'returns no results' do
    expect(subject.results).to eq({})
  end

  it 'returns aggregated results' do
    ::Gitlab::RuggedInstrumentation.query_time += 1.234
    ::Gitlab::RuggedInstrumentation.increment_query_count
    ::Gitlab::RuggedInstrumentation.increment_query_count

    ::Gitlab::RuggedInstrumentation.add_call_details(feature: :rugged_test,
                                                     args: [project.repository.raw, 'HEAD'],
                                                     duration: 0.123)
    ::Gitlab::RuggedInstrumentation.add_call_details(feature: :rugged_test2,
                                                     args: [project.repository, 'refs/heads/master'],
                                                     duration: 0.456)

    results = subject.results
    expect(results[:calls]).to eq(2)
    expect(results[:duration]).to eq('1234.00ms')
    expect(results[:details].count).to eq(2)

    expected = [
      [project.repository.raw.to_s, "HEAD"],
      [project.repository.to_s, "refs/heads/master"]
    ]

    expect(results[:details].map { |data| data[:args] }).to match_array(expected)
  end
end