summaryrefslogtreecommitdiff
path: root/spec/lib/gitlab/git/repository_cleaner_spec.rb
blob: 6602f22843fb802835ecbef0a7072821f24200fc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
require 'spec_helper'

describe Gitlab::Git::RepositoryCleaner do
  include HttpIOHelpers

  let(:project) { create(:project, :repository) }
  let(:repository) { project.repository }
  let(:head_sha) { repository.head_commit.id }
  let(:object_map_data) { "#{head_sha} #{'0' * 40}" }

  subject(:cleaner) { described_class.new(repository.raw) }

  describe '#apply_bfg_object_map' do
    let(:clean_refs) { %W[refs/environments/1 refs/merge-requests/1 refs/keep-around/#{head_sha}] }
    let(:keep_refs) { %w[refs/heads/_keep refs/tags/_keep] }

    before do
      (clean_refs + keep_refs).each { |ref| repository.create_ref(head_sha, ref) }
    end

    context 'from StringIO' do
      let(:object_map) { StringIO.new(object_map_data) }

      it 'removes internal references' do
        cleaner.apply_bfg_object_map(object_map)

        aggregate_failures do
          clean_refs.each { |ref| expect(repository.ref_exists?(ref)).to be_falsy }
          keep_refs.each { |ref| expect(repository.ref_exists?(ref)).to be_truthy }
        end
      end
    end

    context 'from Gitlab::HttpIO' do
      let(:url) { 'http://example.com/bfg_object_map.txt' }
      let(:tempfile) { Tempfile.new }
      let(:object_map) { Gitlab::HttpIO.new(url, object_map_data.size) }

      around do |example|
        tempfile.write(object_map_data)
        tempfile.close

        example.run
      ensure
        tempfile.unlink
      end

      it 'removes internal references' do
        stub_remote_url_200(url, tempfile.path)

        cleaner.apply_bfg_object_map(object_map)

        aggregate_failures do
          clean_refs.each { |ref| expect(repository.ref_exists?(ref)).to be_falsy }
          keep_refs.each { |ref| expect(repository.ref_exists?(ref)).to be_truthy }
        end
      end
    end
  end
end