gitlab-org--gitlab-foss/spec/uploaders/personal_file_uploader_spec.rb

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe PersonalFileUploader do
  let(:model) { create(:personal_snippet) }
  let(:uploader) { described_class.new(model) }
  let(:upload) { create(:upload, :personal_snippet_upload) }
  let(:secret) { SecureRandom.hex }

  subject { uploader }

  shared_examples '#base_dir' do
    before do
      subject.instance_variable_set(:@secret, secret)
    end

    it 'is prefixed with uploads/-/system' do
      allow(uploader).to receive(:file).and_return(double(extension: 'txt', filename: 'file_name'))

      expect(described_class.base_dir(model)).to eq("uploads/-/system/personal_snippet/#{model.id}")
    end
  end

  shared_examples '#to_h' do
    before do
      subject.instance_variable_set(:@secret, secret)
    end

    it 'is correct' do
      allow(uploader).to receive(:file).and_return(double(extension: 'txt', filename: 'file_name'))
      expected_url = "/uploads/-/system/personal_snippet/#{model.id}/#{secret}/file_name"

      expect(uploader.to_h).to eq(
        alt: 'file_name',
        url: expected_url,
        markdown: "[file_name](#{expected_url})"
      )
    end
  end

  describe '#upload_paths' do
    it 'builds correct paths for both local and remote storage' do
      paths = uploader.upload_paths('test.jpg')

      expect(paths.first).to match(%r[\h+\/test.jpg])
      expect(paths.second).to match(%r[^personal_snippet\/\d+\/\h+\/test.jpg])
    end
  end

  context 'object_store is LOCAL' do
    it_behaves_like 'builds correct paths',
                    store_dir: %r[uploads/-/system/personal_snippet/\d+/\h+],
                    upload_path: %r[\h+/\S+],
                    absolute_path: %r[#{CarrierWave.root}/uploads/-/system/personal_snippet\/\d+\/\h+\/\S+$]

    it_behaves_like '#base_dir'
    it_behaves_like '#to_h'
  end

  context "object_store is REMOTE" do
    before do
      stub_uploads_object_storage
    end

    include_context 'with storage', described_class::Store::REMOTE

    it_behaves_like 'builds correct paths',
                    store_dir: %r[\d+/\h+],
                    upload_path: %r[^personal_snippet\/\d+\/\h+\/<filename>]

    it_behaves_like '#base_dir'
    it_behaves_like '#to_h'
  end

  describe "#migrate!" do
    before do
      uploader.store!(fixture_file_upload('spec/fixtures/doc_sample.txt'))
      stub_uploads_object_storage
    end

    it_behaves_like "migrates", to_store: described_class::Store::REMOTE
    it_behaves_like "migrates", from_store: described_class::Store::REMOTE, to_store: described_class::Store::LOCAL
  end
end