BigW Consortium Gitlab

gitaly_rake_spec.rb 4.02 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
require 'rake_helper'

describe 'gitlab:gitaly namespace rake task' do
  before :all do
    Rake.application.rake_require 'tasks/gitlab/gitaly'
  end

  describe 'install' do
    let(:repo) { 'https://gitlab.com/gitlab-org/gitaly.git' }
    let(:clone_path) { Rails.root.join('tmp/tests/gitaly').to_s }
11
    let(:version) { File.read(Rails.root.join(Gitlab::GitalyClient::SERVER_VERSION_FILE)).chomp }
12 13 14 15 16 17 18 19 20 21 22 23

    context 'no dir given' do
      it 'aborts and display a help message' do
        # avoid writing task output to spec progress
        allow($stderr).to receive :write
        expect { run_rake_task('gitlab:gitaly:install') }.to raise_error /Please specify the directory where you want to install gitaly/
      end
    end

    context 'when an underlying Git command fail' do
      it 'aborts and display a help message' do
        expect_any_instance_of(Object).
24
          to receive(:checkout_or_clone_version).and_raise 'Git error'
25 26 27 28 29 30 31 32 33 34

        expect { run_rake_task('gitlab:gitaly:install', clone_path) }.to raise_error 'Git error'
      end
    end

    describe 'checkout or clone' do
      before do
        expect(Dir).to receive(:chdir).with(clone_path)
      end

35
      it 'calls checkout_or_clone_version with the right arguments' do
36
        expect_any_instance_of(Object).
37
          to receive(:checkout_or_clone_version).with(version: version, repo: repo, target_dir: clone_path)
38 39 40 41 42 43 44 45 46 47 48 49 50

        run_rake_task('gitlab:gitaly:install', clone_path)
      end
    end

    describe 'gmake/make' do
      before do
        FileUtils.mkdir_p(clone_path)
        expect(Dir).to receive(:chdir).with(clone_path).and_call_original
      end

      context 'gmake is available' do
        before do
51
          expect_any_instance_of(Object).to receive(:checkout_or_clone_version)
52 53 54 55 56 57 58 59 60 61 62 63 64
          allow_any_instance_of(Object).to receive(:run_command!).with(['gmake']).and_return(true)
        end

        it 'calls gmake in the gitaly directory' do
          expect(Gitlab::Popen).to receive(:popen).with(%w[which gmake]).and_return(['/usr/bin/gmake', 0])
          expect_any_instance_of(Object).to receive(:run_command!).with(['gmake']).and_return(true)

          run_rake_task('gitlab:gitaly:install', clone_path)
        end
      end

      context 'gmake is not available' do
        before do
65
          expect_any_instance_of(Object).to receive(:checkout_or_clone_version)
66 67 68 69 70 71 72 73 74 75 76 77
          allow_any_instance_of(Object).to receive(:run_command!).with(['make']).and_return(true)
        end

        it 'calls make in the gitaly directory' do
          expect(Gitlab::Popen).to receive(:popen).with(%w[which gmake]).and_return(['', 42])
          expect_any_instance_of(Object).to receive(:run_command!).with(['make']).and_return(true)

          run_rake_task('gitlab:gitaly:install', clone_path)
        end
      end
    end
  end
78 79 80 81

  describe 'storage_config' do
    it 'prints storage configuration in a TOML format' do
      config = {
82 83 84 85 86 87 88 89
        'default' => {
          'path' => '/path/to/default',
          'gitaly_address' => 'unix:/path/to/my.socket'
        },
        'nfs_01' => {
          'path' => '/path/to/nfs_01',
          'gitaly_address' => 'unix:/path/to/my.socket'
        }
90 91 92
      }
      allow(Gitlab.config.repositories).to receive(:storages).and_return(config)

93
      expected_output = ''
94
      Timecop.freeze do
95
        expected_output = <<~TOML
96 97
          # Gitaly storage configuration generated from #{Gitlab.config.source} on #{Time.current.to_s(:long)}
          # This is in TOML format suitable for use in Gitaly's config.toml file.
98
          socket_path = "/path/to/my.socket"
99 100 101 102 103 104
          [[storage]]
          name = "default"
          path = "/path/to/default"
          [[storage]]
          name = "nfs_01"
          path = "/path/to/nfs_01"
105 106 107
        TOML
      end

108 109
      expect { run_rake_task('gitlab:gitaly:storage_config')}.
        to output(expected_output).to_stdout
110

111
      parsed_output = TOML.parse(expected_output)
112 113 114 115 116
      config.each do |name, params|
        expect(parsed_output['storage']).to include({ 'name' => name, 'path' => params['path'] })
      end
    end
  end
117
end