f7163afb8a
This backports to CE changes that allow the recording of the repository_type in the table lfs_objects_projects. This is in order to allow future pruning of unreferenced LFS objects, as we will need to know which repository to look in for the LFS pointer file. The EE MR that contains the original code and a full explanation of the changes is https://gitlab.com/gitlab-org/gitlab-ee/merge_requests/13894 EE Issue https://gitlab.com/gitlab-org/gitlab-ee/issues/9490 Note that there was a lot of CE code changed in the EE MR because we want to allow the wiki repository to also use LFS. See https://gitlab.com/gitlab-org/gitlab-ce/issues/43721. As the wiki is an unlicensed feature, a full backport is required to enable this.
119 lines
3.5 KiB
Ruby
119 lines
3.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
describe LfsObject do
|
|
it 'has a distinct has_many :projects relation through lfs_objects_projects' do
|
|
lfs_object = create(:lfs_object)
|
|
project = create(:project)
|
|
[:project, :design].each do |repository_type|
|
|
create(:lfs_objects_project, project: project,
|
|
lfs_object: lfs_object,
|
|
repository_type: repository_type)
|
|
end
|
|
|
|
expect(lfs_object.lfs_objects_projects.size).to eq(2)
|
|
expect(lfs_object.projects.size).to eq(1)
|
|
expect(lfs_object.projects.to_a).to eql([project])
|
|
end
|
|
|
|
describe '#local_store?' do
|
|
it 'returns true when file_store is equal to LfsObjectUploader::Store::LOCAL' do
|
|
subject.file_store = LfsObjectUploader::Store::LOCAL
|
|
|
|
expect(subject.local_store?).to eq true
|
|
end
|
|
|
|
it 'returns false when file_store is equal to LfsObjectUploader::Store::REMOTE' do
|
|
subject.file_store = LfsObjectUploader::Store::REMOTE
|
|
|
|
expect(subject.local_store?).to eq false
|
|
end
|
|
end
|
|
|
|
describe '#schedule_background_upload' do
|
|
before do
|
|
stub_lfs_setting(enabled: true)
|
|
end
|
|
|
|
subject { create(:lfs_object, :with_file) }
|
|
|
|
context 'when object storage is disabled' do
|
|
before do
|
|
stub_lfs_object_storage(enabled: false)
|
|
end
|
|
|
|
it 'does not schedule the migration' do
|
|
expect(ObjectStorage::BackgroundMoveWorker).not_to receive(:perform_async)
|
|
|
|
subject
|
|
end
|
|
end
|
|
|
|
context 'when object storage is enabled' do
|
|
context 'when background upload is enabled' do
|
|
context 'when is licensed' do
|
|
before do
|
|
stub_lfs_object_storage(background_upload: true)
|
|
end
|
|
|
|
it 'schedules the model for migration' do
|
|
expect(ObjectStorage::BackgroundMoveWorker)
|
|
.to receive(:perform_async)
|
|
.with('LfsObjectUploader', described_class.name, :file, kind_of(Numeric))
|
|
.once
|
|
|
|
subject
|
|
end
|
|
|
|
it 'schedules the model for migration once' do
|
|
expect(ObjectStorage::BackgroundMoveWorker)
|
|
.to receive(:perform_async)
|
|
.with('LfsObjectUploader', described_class.name, :file, kind_of(Numeric))
|
|
.once
|
|
|
|
create(:lfs_object, :with_file)
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'when background upload is disabled' do
|
|
before do
|
|
stub_lfs_object_storage(background_upload: false)
|
|
end
|
|
|
|
it 'schedules the model for migration' do
|
|
expect(ObjectStorage::BackgroundMoveWorker).not_to receive(:perform_async)
|
|
|
|
subject
|
|
end
|
|
end
|
|
end
|
|
|
|
describe 'file is being stored' do
|
|
let(:lfs_object) { create(:lfs_object, :with_file) }
|
|
|
|
context 'when existing object has local store' do
|
|
it 'is stored locally' do
|
|
expect(lfs_object.file_store).to be(ObjectStorage::Store::LOCAL)
|
|
expect(lfs_object.file).to be_file_storage
|
|
expect(lfs_object.file.object_store).to eq(ObjectStorage::Store::LOCAL)
|
|
end
|
|
end
|
|
|
|
context 'when direct upload is enabled' do
|
|
before do
|
|
stub_lfs_object_storage(direct_upload: true)
|
|
end
|
|
|
|
context 'when file is stored' do
|
|
it 'is stored remotely' do
|
|
expect(lfs_object.file_store).to eq(ObjectStorage::Store::REMOTE)
|
|
expect(lfs_object.file).not_to be_file_storage
|
|
expect(lfs_object.file.object_store).to eq(ObjectStorage::Store::REMOTE)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|