1 files changed, 84 insertions, 0 deletions
diff --git a/spec/uploaders/object_storage/cdn/google_ip_cache_spec.rb b/spec/uploaders/object_storage/cdn/google_ip_cache_spec.rb
new file mode 100644
index 00000000000..d6568636bc0
--- /dev/null
+++ b/spec/uploaders/object_storage/cdn/google_ip_cache_spec.rb
@@ -0,0 +1,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