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

160 lines
5 KiB
Ruby
Raw Normal View History

require 'rake_helper'
describe 'gitlab:cleanup rake tasks' do
before do
Rake.application.rake_require 'tasks/gitlab/cleanup'
end
2018-07-26 17:23:33 -04:00
describe 'cleanup namespaces and repos' do
let(:gitlab_shell) { Gitlab::Shell.new }
let(:storage) { storages.keys.first }
let(:storages) do
{
'default' => Gitlab::GitalyClient::StorageSettings.new(@default_storage_hash.merge('path' => 'tmp/tests/default_storage'))
}
end
before(:all) do
@default_storage_hash = Gitlab.config.repositories.storages.default.to_h
end
before do
allow(Gitlab.config.repositories).to receive(:storages).and_return(storages)
end
after do
Gitlab::GitalyClient::StorageService.new(storage).delete_all_repositories
end
describe 'cleanup:repos' do
before do
gitlab_shell.add_namespace(storage, 'broken/project.git')
gitlab_shell.add_namespace(storage, '@hashed/12/34/5678.git')
end
it 'moves it to an orphaned path' do
now = Time.now
Timecop.freeze(now) do
run_rake_task('gitlab:cleanup:repos')
repo_list = Gitlab::GitalyClient::StorageService.new(storage).list_directories(depth: 0)
expect(repo_list.last).to include("broken+orphaned+#{now.to_i}")
end
end
it 'ignores @hashed repos' do
run_rake_task('gitlab:cleanup:repos')
expect(gitlab_shell.exists?(storage, '@hashed/12/34/5678.git')).to be(true)
end
end
describe 'cleanup:dirs' do
it 'removes missing namespaces' do
gitlab_shell.add_namespace(storage, "namespace_1/project.git")
gitlab_shell.add_namespace(storage, "namespace_2/project.git")
allow(Namespace).to receive(:pluck).and_return(['namespace_1'])
stub_env('REMOVE', 'true')
run_rake_task('gitlab:cleanup:dirs')
expect(gitlab_shell.exists?(storage, 'namespace_1')).to be(true)
expect(gitlab_shell.exists?(storage, 'namespace_2')).to be(false)
end
it 'ignores @hashed directory' do
gitlab_shell.add_namespace(storage, '@hashed/12/34/5678.git')
run_rake_task('gitlab:cleanup:dirs')
expect(gitlab_shell.exists?(storage, '@hashed/12/34/5678.git')).to be(true)
end
end
end
2018-07-26 17:23:33 -04:00
# A single integration test that is redundant with one part of the
# Gitlab::Cleanup::ProjectUploads spec.
#
# Additionally, this tests DRY_RUN env var values, and the extra line of
# output that says you can disable DRY_RUN if it's enabled.
2018-07-26 17:23:33 -04:00
describe 'cleanup:project_uploads' do
let!(:logger) { double(:logger) }
2018-07-26 17:23:33 -04:00
before do
expect(main_object).to receive(:logger).and_return(logger).at_least(1).times
2018-07-26 17:23:33 -04:00
allow(logger).to receive(:info).at_least(1).times
allow(logger).to receive(:debug).at_least(1).times
end
2018-07-26 17:23:33 -04:00
context 'with a fixable orphaned project upload file' do
let(:orphaned) { create(:upload, :issuable_upload, :with_file, model: build(:project, :legacy_storage)) }
let(:new_path) { orphaned.absolute_path }
let(:path) { File.join(FileUploader.root, 'some', 'wrong', 'location', orphaned.path) }
2018-07-26 17:23:33 -04:00
before do
FileUtils.mkdir_p(File.dirname(path))
FileUtils.mv(new_path, path)
2018-07-26 17:23:33 -04:00
end
context 'with DRY_RUN disabled' do
before do
stub_env('DRY_RUN', 'false')
2018-07-26 17:23:33 -04:00
end
it 'moves the file to its proper location' do
run_rake_task('gitlab:cleanup:project_uploads')
2018-07-26 17:23:33 -04:00
expect(File.exist?(path)).to be_falsey
expect(File.exist?(new_path)).to be_truthy
2018-07-26 17:23:33 -04:00
end
it 'logs action as done' do
expect(logger).to receive(:info).with("Looking for orphaned project uploads to clean up...")
expect(logger).to receive(:info).with("Did fix #{path} -> #{new_path}")
2018-07-26 17:23:33 -04:00
run_rake_task('gitlab:cleanup:project_uploads')
end
end
shared_examples_for 'does not move the file' do
it 'does not move the file' do
run_rake_task('gitlab:cleanup:project_uploads')
2018-07-26 17:23:33 -04:00
expect(File.exist?(path)).to be_truthy
expect(File.exist?(new_path)).to be_falsey
2018-07-26 17:23:33 -04:00
end
it 'logs action as able to be done' do
expect(logger).to receive(:info).with("Looking for orphaned project uploads to clean up. Dry run...")
expect(logger).to receive(:info).with("Can fix #{path} -> #{new_path}")
expect(logger).to receive(:info).with(/To clean up these files run this command with DRY_RUN=false/)
2018-07-26 17:23:33 -04:00
run_rake_task('gitlab:cleanup:project_uploads')
end
2018-07-26 17:23:33 -04:00
end
context 'with DRY_RUN explicitly enabled' do
2018-07-26 17:23:33 -04:00
before do
stub_env('DRY_RUN', 'true')
end
it_behaves_like 'does not move the file'
2018-07-26 17:23:33 -04:00
end
context 'with DRY_RUN set to an unknown value' do
2018-07-26 17:23:33 -04:00
before do
stub_env('DRY_RUN', 'foo')
end
it_behaves_like 'does not move the file'
2018-07-26 17:23:33 -04:00
end
context 'with DRY_RUN unset' do
it_behaves_like 'does not move the file'
2018-07-26 17:23:33 -04:00
end
end
end
end