gitlab-org--gitlab-foss/spec/lib/safe_zip/extract_spec.rb

# frozen_string_literal: true

require 'spec_helper'

describe SafeZip::Extract do
  let(:target_path) { Dir.mktmpdir('safe-zip') }
  let(:directories) { %w(public) }
  let(:object) { described_class.new(archive) }
  let(:archive) { Rails.root.join('spec', 'fixtures', 'safe_zip', archive_name) }

  after do
    FileUtils.remove_entry_secure(target_path)
  end

  describe '#extract' do
    subject { object.extract(directories: directories, to: target_path) }

    shared_examples 'extracts archive' do |param|
      before do
        stub_feature_flags(safezip_use_rubyzip: param)
      end

      it 'does extract archive' do
        subject

        expect(File.exist?(File.join(target_path, 'public', 'index.html'))).to eq(true)
        expect(File.exist?(File.join(target_path, 'source'))).to eq(false)
      end
    end

    shared_examples 'fails to extract archive' do |param|
      before do
        stub_feature_flags(safezip_use_rubyzip: param)
      end

      it 'does not extract archive' do
        expect { subject }.to raise_error(SafeZip::Extract::Error)
      end
    end

    %w(valid-simple.zip valid-symlinks-first.zip valid-non-writeable.zip).each do |name|
      context "when using #{name} archive" do
        let(:archive_name) { name }

        context 'for RubyZip' do
          it_behaves_like 'extracts archive', true
        end

        context 'for UnZip' do
          it_behaves_like 'extracts archive', false
        end
      end
    end

    %w(invalid-symlink-does-not-exist.zip invalid-symlinks-outside.zip).each do |name|
      context "when using #{name} archive" do
        let(:archive_name) { name }

        context 'for RubyZip' do
          it_behaves_like 'fails to extract archive', true
        end

        context 'for UnZip (UNSAFE)' do
          it_behaves_like 'extracts archive', false
        end
      end
    end

    context 'when no matching directories are found' do
      let(:archive_name) { 'valid-simple.zip' }
      let(:directories) { %w(non/existing) }

      context 'for RubyZip' do
        it_behaves_like 'fails to extract archive', true
      end

      context 'for UnZip' do
        it_behaves_like 'fails to extract archive', false
      end
    end
  end
end