summaryrefslogtreecommitdiff
path: root/spec/lib/gitlab/usage/metrics/instrumentations/generic_metric_spec.rb
blob: cc4df696b376c725a1b1343ec97c0de0c1c4596c (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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Gitlab::Usage::Metrics::Instrumentations::GenericMetric do
  shared_examples 'custom fallback' do |custom_fallback|
    subject do
      Class.new(described_class) do
        fallback(custom_fallback)
        value { ApplicationRecord.database.version }
      end.new(time_frame: 'none')
    end

    describe '#value' do
      it 'gives the correct value' do
        expect(subject.value).to eq(ApplicationRecord.database.version)
      end

      context 'when raising an exception' do
        before do
          allow(Gitlab::ErrorTracking).to receive(:should_raise_for_dev?).and_return(should_raise_for_dev)
          expect(ApplicationRecord.database).to receive(:version).and_raise('Error')
        end

        context 'with should_raise_for_dev? false' do
          let(:should_raise_for_dev) { false }

          it 'return the custom fallback' do
            expect(subject.value).to eq(custom_fallback)
          end
        end

        context 'with should_raise_for_dev? true' do
          let(:should_raise_for_dev) { true }

          it 'raises an error' do
            expect { subject.value }.to raise_error('Error')
          end
        end
      end
    end
  end

  context 'with default fallback' do
    subject do
      Class.new(described_class) do
        value { ApplicationRecord.database.version }
      end.new(time_frame: 'none')
    end

    describe '#value' do
      it 'gives the correct value' do
        expect(subject.value).to eq(ApplicationRecord.database.version )
      end

      context 'when raising an exception' do
        before do
          allow(Gitlab::ErrorTracking).to receive(:should_raise_for_dev?).and_return(should_raise_for_dev)
          expect(ApplicationRecord.database).to receive(:version).and_raise('Error')
        end

        context 'with should_raise_for_dev? false' do
          let(:should_raise_for_dev) { false }

          it 'return the default fallback' do
            expect(subject.value).to eq(described_class::FALLBACK)
          end
        end

        context 'with should_raise_for_dev? true' do
          let(:should_raise_for_dev) { true }

          it 'raises an error' do
            expect { subject.value }.to raise_error('Error')
          end
        end
      end
    end
  end

  context 'with custom fallback -2' do
    it_behaves_like 'custom fallback', -2
  end

  context 'with custom fallback nil' do
    it_behaves_like 'custom fallback', nil
  end

  context 'with custom fallback false' do
    it_behaves_like 'custom fallback', false
  end

  context 'with custom fallback true' do
    it_behaves_like 'custom fallback', true
  end

  context 'with custom fallback []' do
    it_behaves_like 'custom fallback', []
  end

  context 'with custom fallback { major: -1 }' do
    it_behaves_like 'custom fallback', { major: -1 }
  end
end