fix avatar restorer

This commit is contained in:
James Lopez 2018-09-11 08:59:14 +02:00
parent 6e394c44d5
commit 994d91883b
No known key found for this signature in database
GPG key ID: 756BF8E9D7C0CF39
2 changed files with 24 additions and 11 deletions

View file

@ -19,7 +19,7 @@ module Gitlab
private
def avatar_export_file
@avatar_export_file ||= Dir["#{avatar_export_path}/**/*"].first
@avatar_export_file ||= Dir["#{avatar_export_path}/**/*"].reject { |f| File.directory?(f) }.first
end
def avatar_export_path

View file

@ -6,22 +6,35 @@ describe Gitlab::ImportExport::AvatarRestorer do
let(:shared) { project.import_export_shared }
let(:project) { create(:project) }
before do
allow_any_instance_of(described_class).to receive(:avatar_export_file)
.and_return(uploaded_image_temp_path)
end
after do
project.remove_avatar!
end
it 'restores a project avatar' do
expect(described_class.new(project: project, shared: shared).restore).to be true
context 'with avatar' do
before do
allow_any_instance_of(described_class).to receive(:avatar_export_file)
.and_return(uploaded_image_temp_path)
end
it 'restores a project avatar' do
expect(described_class.new(project: project, shared: shared).restore).to be true
end
it 'saves the avatar into the project' do
described_class.new(project: project, shared: shared).restore
expect(project.reload.avatar.file.exists?).to be true
end
end
it 'saves the avatar into the project' do
described_class.new(project: project, shared: shared).restore
it 'does not break if there is just a directory' do
Dir.mktmpdir do |tmpdir|
FileUtils.mkdir_p("#{tmpdir}/a/b")
expect(project.reload.avatar.file.exists?).to be true
allow_any_instance_of(described_class).to receive(:avatar_export_path)
.and_return("#{tmpdir}/a")
expect(described_class.new(project: project, shared: shared).restore).to be true
end
end
end