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} #{Gitlab::Git::BLANK_SHA}" } 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] } subject(:cleaner) { described_class.new(repository.raw) } shared_examples_for '#apply_bfg_object_map_stream' do before do (clean_refs + keep_refs).each { |ref| repository.create_ref(head_sha, ref) } end it 'removes internal references' do entries = [] cleaner.apply_bfg_object_map_stream(object_map) do |rsp| entries.concat(rsp.entries) end aggregate_failures do clean_refs.each { |ref| expect(repository.ref_exists?(ref)).to be(false) } keep_refs.each { |ref| expect(repository.ref_exists?(ref)).to be(true) } expect(entries).to contain_exactly( Gitaly::ApplyBfgObjectMapStreamResponse::Entry.new( type: :COMMIT, old_oid: head_sha, new_oid: Gitlab::Git::BLANK_SHA ) ) end end end describe '#apply_bfg_object_map_stream (from StringIO)' do let(:object_map) { StringIO.new(object_map_data) } include_examples '#apply_bfg_object_map_stream' end describe '#apply_bfg_object_map_stream (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 stub_remote_url_200(url, tempfile.path) example.run ensure tempfile.unlink end include_examples '#apply_bfg_object_map_stream' end end