spec/services/pages/legacy_storage_lease_spec.rb


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

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe ::Pages::LegacyStorageLease do
  let(:project) { create(:project) }

  let(:implementation) do
    Class.new do
      include ::Pages::LegacyStorageLease

      attr_reader :project

      def initialize(project)
        @project = project
      end

      def execute
        try_obtain_lease do
          execute_unsafe
        end
      end

      def execute_unsafe
        true
      end
    end
  end

  let(:service) { implementation.new(project) }

  it 'allows method to be executed' do
    expect(service).to receive(:execute_unsafe).and_call_original

    expect(service.execute).to eq(true)
  end

  context 'when another service holds the lease for the same project' do
    around do |example|
      implementation.new(project).try_obtain_lease do
        example.run
      end
    end

    it 'does not run guarded method' do
      expect(service).not_to receive(:execute_unsafe)

      expect(service.execute).to eq(nil)
    end

    it 'runs guarded method if feature flag is disabled' do
      stub_feature_flags(pages_use_legacy_storage_lease: false)

      expect(service).to receive(:execute_unsafe).and_call_original

      expect(service.execute).to eq(true)
    end
  end

  context 'when another service holds the lease for the different project' do
    around do |example|
      implementation.new(create(:project)).try_obtain_lease do
        example.run
      end
    end

    it 'allows method to be executed' do
      expect(service).to receive(:execute_unsafe).and_call_original

      expect(service.execute).to eq(true)
    end
  end
end