summaryrefslogtreecommitdiff
path: root/spec/lib/gitlab/process_memory_cache/helper_spec.rb
blob: 27d7fd0bdcfd49ba9c1fe9df038bde922d00d9f9 (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
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Gitlab::ProcessMemoryCache::Helper, :use_clean_rails_memory_store_caching do
  let(:minimal_test_class) do
    Class.new do
      include Gitlab::ProcessMemoryCache::Helper

      def cached_content
        fetch_memory_cache(:cached_content_instance_key) { expensive_computation }
      end

      def clear_cached_content
        invalidate_memory_cache(:cached_content_instance_key)
      end
    end
  end

  before do
    stub_const("MinimalTestClass", minimal_test_class)
  end

  subject { MinimalTestClass.new }

  describe '.fetch_memory_cache' do
    it 'memoizes the result' do
      is_expected.to receive(:expensive_computation).once.and_return(1)

      2.times do
        expect(subject.cached_content).to eq(1)
      end
    end

    it 'resets the cache when the shared key is missing', :aggregate_failures do
      expect(Rails.cache).to receive(:read).with(:cached_content_instance_key).twice.and_return(nil)
      is_expected.to receive(:expensive_computation).thrice.and_return(1, 2, 3)

      3.times do |index|
        expect(subject.cached_content).to eq(index + 1)
      end
    end
  end

  describe '.invalidate_memory_cache' do
    it 'invalidates the cache' do
      is_expected.to receive(:expensive_computation).twice.and_return(1, 2)

      expect { subject.clear_cached_content }.to change { subject.cached_content }
    end
  end
end