gitlab-org--gitlab-foss/spec/tasks/gitlab/backup_rake_spec.rb

require 'spec_helper'
require 'rake'

describe 'gitlab:app namespace rake task' do
  before :all do
    Rake.application.rake_require "tasks/gitlab/task_helpers"
    Rake.application.rake_require "tasks/gitlab/backup"
    Rake.application.rake_require "tasks/gitlab/shell"
    # empty task as env is already loaded
    Rake::Task.define_task :environment
  end

  describe 'backup_restore' do
    before do
      # avoid writing task output to spec progress
      allow($stdout).to receive :write
    end

    let :run_rake_task do
      Rake::Task["gitlab:backup:restore"].reenable
      Rake.application.invoke_task "gitlab:backup:restore"
    end

    context 'gitlab version' do
      before do
        Dir.stub glob: []
        allow(Dir).to receive :chdir
        File.stub exists?: true
        Kernel.stub system: true
        FileUtils.stub cp_r: true
        FileUtils.stub mv: true
        Rake::Task["gitlab:shell:setup"].stub invoke: true
      end

      let(:gitlab_version) { Gitlab::VERSION }

      it 'should fail on mismatch' do
        YAML.stub load_file: {gitlab_version: "not #{gitlab_version}" }
        expect { run_rake_task }.to raise_error SystemExit
      end

      it 'should invoke restoration on mach' do
        YAML.stub load_file: {gitlab_version: gitlab_version}
        expect(Rake::Task["gitlab:backup:db:restore"]).to receive :invoke
        expect(Rake::Task["gitlab:backup:repo:restore"]).to receive :invoke
        expect(Rake::Task["gitlab:shell:setup"]).to receive :invoke
        expect { run_rake_task }.to_not raise_error
      end
    end

  end # backup_restore task
end # gitlab:app namespace