Commit 94dc43f3 authored by Stan Hu's avatar Stan Hu

Clean up backup/restore temporary directory

parent 405c4cae
...@@ -28,6 +28,23 @@ describe Backup::Repository do ...@@ -28,6 +28,23 @@ describe Backup::Repository do
end end
describe '#restore' do describe '#restore' do
subject { described_class.new }
let(:timestamp) { Time.utc(2017, 3, 22) }
let(:temp_dirs) do
Gitlab.config.repositories.storages.map do |name, storage|
File.join(storage['path'], '..', 'repositories.old.' + timestamp.to_i.to_s)
end
end
around do |example|
Timecop.freeze(timestamp) { example.run }
end
after do
temp_dirs.each { |path| FileUtils.rm_rf(path) }
end
describe 'command failure' do describe 'command failure' do
before do before do
allow(Gitlab::Popen).to receive(:popen).and_return(['error', 1]) allow(Gitlab::Popen).to receive(:popen).and_return(['error', 1])
...@@ -35,7 +52,7 @@ describe Backup::Repository do ...@@ -35,7 +52,7 @@ describe Backup::Repository do
context 'hashed storage' do context 'hashed storage' do
it 'shows the appropriate error' do it 'shows the appropriate error' do
described_class.new.restore subject.restore
expect(progress).to have_received(:puts).with("Ignoring error on #{project.full_path} (#{project.disk_path}) - error") expect(progress).to have_received(:puts).with("Ignoring error on #{project.full_path} (#{project.disk_path}) - error")
end end
...@@ -45,7 +62,7 @@ describe Backup::Repository do ...@@ -45,7 +62,7 @@ describe Backup::Repository do
let!(:project) { create(:project, :legacy_storage) } let!(:project) { create(:project, :legacy_storage) }
it 'shows the appropriate error' do it 'shows the appropriate error' do
described_class.new.restore subject.restore
expect(progress).to have_received(:puts).with("Ignoring error on #{project.full_path} - error") expect(progress).to have_received(:puts).with("Ignoring error on #{project.full_path} - error")
end end
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment