# frozen_string_literal: true

require 'spec_helper'

describe Gitlab::ImportExport::SnippetsRepoRestorer do
  include GitHelpers

  describe 'bundle a snippet Git repo' do
    let_it_be(:user) { create(:user) }
    let_it_be(:project) { create(:project, namespace: user.namespace) }

    let!(:snippet1) { create(:project_snippet, project: project, author: user) }
    let!(:snippet2) { create(:project_snippet, project: project, author: user) }
    let(:shared) { project.import_export_shared }
    let(:exporter) { Gitlab::ImportExport::SnippetsRepoSaver.new(current_user: user, project: project, shared: shared) }
    let(:bundle_dir) { ::Gitlab::ImportExport.snippets_repo_bundle_path(shared.export_path) }
    let(:service) { instance_double(Gitlab::ImportExport::SnippetRepoRestorer) }
    let(:restorer) do
      described_class.new(user: user,
                          shared: shared,
                          project: project)
    end

    after do
      FileUtils.rm_rf(shared.export_path)
    end

    shared_examples 'imports snippet repositories' do
      before do
        snippet1.snippet_repository&.delete
        snippet1.repository.remove

        snippet2.snippet_repository&.delete
        snippet2.repository.remove
      end

      specify do
        expect(snippet1.repository_exists?).to be false
        expect(snippet2.repository_exists?).to be false

        expect(Gitlab::ImportExport::SnippetRepoRestorer).to receive(:new).with(hash_including(snippet: snippet1, path_to_bundle: bundle_path(snippet1))).and_call_original
        expect(Gitlab::ImportExport::SnippetRepoRestorer).to receive(:new).with(hash_including(snippet: snippet2, path_to_bundle: bundle_path(snippet2))).and_call_original
        expect(restorer.restore).to be_truthy

        snippet1.repository.expire_exists_cache
        snippet2.repository.expire_exists_cache

        expect(snippet1.blobs).not_to be_empty
        expect(snippet2.blobs).not_to be_empty
      end
    end

    context 'when export has no snippet repository bundle' do
      before do
        expect(Dir.exist?(bundle_dir)).to be false
      end

      it_behaves_like 'imports snippet repositories'
    end

    context 'when export has snippet repository bundles and snippets without them' do
      let!(:snippet1) { create(:project_snippet, :repository, project: project, author: user) }
      let!(:snippet2) { create(:project_snippet, project: project, author: user) }

      before do
        exporter.save

        expect(File.exist?(bundle_path(snippet1))).to be true
        expect(File.exist?(bundle_path(snippet2))).to be false
      end

      it_behaves_like 'imports snippet repositories'
    end

    context 'when export has only snippet bundles' do
      let!(:snippet1) { create(:project_snippet, :repository, project: project, author: user) }
      let!(:snippet2) { create(:project_snippet, :repository, project: project, author: user) }

      before do
        exporter.save

        expect(File.exist?(bundle_path(snippet1))).to be true
        expect(File.exist?(bundle_path(snippet2))).to be true
      end

      it_behaves_like 'imports snippet repositories'
    end

    context 'when one snippet cannot be saved' do
      it 'returns false and do not process other snippets' do
        allow(Gitlab::ImportExport::SnippetRepoRestorer).to receive(:new).with(hash_including(snippet: snippet1)).and_return(service)
        allow(service).to receive(:restore).and_return(false)

        expect(Gitlab::ImportExport::SnippetRepoRestorer).not_to receive(:new).with(hash_including(snippet: snippet2))
        expect(restorer.restore).to be_falsey
      end
    end

    def bundle_path(snippet)
      File.join(bundle_dir, ::Gitlab::ImportExport.snippet_repo_bundle_filename_for(snippet))
    end
  end
end
