spec/uploaders/object_storage/cdn/google_ip_cache_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
74
75
76
77
78
79
80
81
82
83
84

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe ObjectStorage::CDN::GoogleIpCache,
  :use_clean_rails_memory_store_caching, :use_clean_rails_redis_caching do
  include StubRequests

  let(:subnets) { [IPAddr.new("34.80.0.0/15"), IPAddr.new("2600:1900:4180::/44")] }
  let(:public_ip) { '18.245.0.42' }

  describe '.update!' do
    it 'caches to both L1 and L2 caches' do
      expect(Gitlab::ProcessMemoryCache.cache_backend.exist?(described_class::GOOGLE_CDN_LIST_KEY)).to be false
      expect(Rails.cache.exist?(described_class::GOOGLE_CDN_LIST_KEY)).to be false

      described_class.update!(subnets)

      expect(Gitlab::ProcessMemoryCache.cache_backend.fetch(described_class::GOOGLE_CDN_LIST_KEY)).to eq(subnets)
      expect(Rails.cache.fetch(described_class::GOOGLE_CDN_LIST_KEY)).to eq(subnets)
    end
  end

  describe '.ready?' do
    it 'returns false' do
      expect(described_class.ready?).to be false
    end

    it 'returns true' do
      described_class.update!(subnets)

      expect(described_class.ready?).to be true
    end
  end

  describe '.google_ip?' do
    using RSpec::Parameterized::TableSyntax

    where(:ip_address, :expected) do
      '34.80.0.1'                               | true
      '18.245.0.42'                             | false
      '2500:1900:4180:0000:0000:0000:0000:0000' | false
      '2600:1900:4180:0000:0000:0000:0000:0000' | true
      '10.10.1.5'                               | false
      'fc00:0000:0000:0000:0000:0000:0000:0000' | false
    end

    before do
      described_class.update!(subnets)
    end

    with_them do
      it { expect(described_class.google_ip?(ip_address)).to eq(expected) }
    end

    it 'uses the L2 cache and updates the L1 cache when L1 is missing' do
      Gitlab::ProcessMemoryCache.cache_backend.delete(described_class::GOOGLE_CDN_LIST_KEY)
      expect(Rails.cache.fetch(described_class::GOOGLE_CDN_LIST_KEY)).to eq(subnets)

      expect(described_class.google_ip?(public_ip)).to be false

      expect(Gitlab::ProcessMemoryCache.cache_backend.fetch(described_class::GOOGLE_CDN_LIST_KEY)).to eq(subnets)
      expect(Rails.cache.fetch(described_class::GOOGLE_CDN_LIST_KEY)).to eq(subnets)
    end

    it 'avoids populating L1 cache if L2 is missing' do
      Gitlab::ProcessMemoryCache.cache_backend.delete(described_class::GOOGLE_CDN_LIST_KEY)
      Rails.cache.delete(described_class::GOOGLE_CDN_LIST_KEY)

      expect(described_class.google_ip?(public_ip)).to be false

      expect(Gitlab::ProcessMemoryCache.cache_backend.exist?(described_class::GOOGLE_CDN_LIST_KEY)).to be false
      expect(Rails.cache.exist?(described_class::GOOGLE_CDN_LIST_KEY)).to be false
    end
  end

  describe '.async_refresh' do
    it 'schedules the worker' do
      expect(::GoogleCloud::FetchGoogleIpListWorker).to receive(:perform_async)

      described_class.async_refresh
    end
  end
end