85 lines
2.8 KiB
Ruby
85 lines
2.8 KiB
Ruby
|
# 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
|