gitlab-org--gitlab-foss/spec/models/pages/virtual_domain_spec.rb

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Pages::VirtualDomain do
  describe '#certificate and #key pair' do
    let(:domain) { nil }
    let(:project) { instance_double(Project) }

    subject(:virtual_domain) { described_class.new(projects: [project], domain: domain) }

    it 'returns nil if there is no domain provided' do
      expect(virtual_domain.certificate).to be_nil
      expect(virtual_domain.key).to be_nil
    end

    context 'when Pages domain is provided' do
      let(:domain) { instance_double(PagesDomain, certificate: 'certificate', key: 'key') }

      it 'returns certificate and key from the provided domain' do
        expect(virtual_domain.certificate).to eq('certificate')
        expect(virtual_domain.key).to eq('key')
      end
    end
  end

  describe '#lookup_paths' do
    let(:project_a) { instance_double(Project) }
    let(:project_b) { instance_double(Project) }
    let(:project_c) { instance_double(Project) }
    let(:pages_lookup_path_a) { instance_double(Pages::LookupPath, prefix: 'aaa', source: { type: 'zip', path: 'https://example.com' }) }
    let(:pages_lookup_path_b) { instance_double(Pages::LookupPath, prefix: 'bbb', source: { type: 'zip', path: 'https://example.com' }) }
    let(:pages_lookup_path_without_source) { instance_double(Pages::LookupPath, prefix: 'ccc', source: nil) }

    context 'when there is pages domain provided' do
      let(:domain) { instance_double(PagesDomain) }

      subject(:virtual_domain) { described_class.new(projects: [project_a, project_b, project_c], domain: domain) }

      it 'returns collection of projects pages lookup paths sorted by prefix in reverse' do
        expect(project_a).to receive(:pages_lookup_path).with(domain: domain, trim_prefix: nil).and_return(pages_lookup_path_a)
        expect(project_b).to receive(:pages_lookup_path).with(domain: domain, trim_prefix: nil).and_return(pages_lookup_path_b)
        expect(project_c).to receive(:pages_lookup_path).with(domain: domain, trim_prefix: nil).and_return(pages_lookup_path_without_source)

        expect(virtual_domain.lookup_paths).to eq([pages_lookup_path_b, pages_lookup_path_a])
      end
    end

    context 'when there is trim_prefix provided' do
      subject(:virtual_domain) { described_class.new(projects: [project_a, project_b], trim_prefix: 'group/') }

      it 'returns collection of projects pages lookup paths sorted by prefix in reverse' do
        expect(project_a).to receive(:pages_lookup_path).with(trim_prefix: 'group/', domain: nil).and_return(pages_lookup_path_a)
        expect(project_b).to receive(:pages_lookup_path).with(trim_prefix: 'group/', domain: nil).and_return(pages_lookup_path_b)

        expect(virtual_domain.lookup_paths).to eq([pages_lookup_path_b, pages_lookup_path_a])
      end
    end
  end

  describe '#cache_key' do
    it 'returns the cache key based in the given cache_control' do
      cache_control = instance_double(::Gitlab::Pages::CacheControl, cache_key: 'cache_key')
      virtual_domain = described_class.new(projects: [instance_double(Project)], cache: cache_control)

      expect(virtual_domain.cache_key).to eq('cache_key')
    end

    it 'returns nil when no cache_control is given' do
      virtual_domain = described_class.new(projects: [instance_double(Project)])

      expect(virtual_domain.cache_key).to be_nil
    end
  end
end