136 lines
3.7 KiB
Ruby
136 lines
3.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.shared_context 'with storage' do |store, **stub_params|
|
|
before do
|
|
subject.object_store = store
|
|
end
|
|
end
|
|
|
|
RSpec.shared_examples "migrates" do |to_store:, from_store: nil|
|
|
let(:to) { to_store }
|
|
let(:from) { from_store || subject.object_store }
|
|
|
|
def migrate(to)
|
|
subject.migrate!(to)
|
|
end
|
|
|
|
def checksum
|
|
Digest::SHA256.hexdigest(subject.read)
|
|
end
|
|
|
|
before do
|
|
migrate(from)
|
|
end
|
|
|
|
it 'returns corresponding file type' do
|
|
expect(subject).to be_an(CarrierWave::Uploader::Base)
|
|
expect(subject).to be_a(ObjectStorage::Concern)
|
|
|
|
if from == described_class::Store::REMOTE
|
|
expect(subject.file).to be_a(CarrierWave::Storage::Fog::File)
|
|
elsif from == described_class::Store::LOCAL
|
|
expect(subject.file).to be_a(CarrierWave::SanitizedFile)
|
|
else
|
|
raise 'Unexpected file type'
|
|
end
|
|
end
|
|
|
|
it 'does nothing when migrating to the current store' do
|
|
expect { migrate(from) }.not_to change { subject.object_store }.from(from)
|
|
end
|
|
|
|
it 'migrate to the specified store' do
|
|
from_checksum = checksum
|
|
|
|
expect { migrate(to) }.to change { subject.object_store }.from(from).to(to)
|
|
expect(checksum).to eq(from_checksum)
|
|
end
|
|
|
|
it 'removes the original file after the migration' do
|
|
original_file = subject.file.path
|
|
migrate(to)
|
|
|
|
expect(File.exist?(original_file)).to be_falsey
|
|
end
|
|
|
|
it 'can access to the original file during migration' do
|
|
file = subject.file
|
|
|
|
allow(subject).to receive(:delete_migrated_file) {} # Remove as a callback of :migrate
|
|
allow(subject).to receive(:record_upload) {} # Remove as a callback of :store (:record_upload)
|
|
|
|
expect(file.exists?).to be_truthy
|
|
expect { migrate(to) }.not_to change { file.exists? }
|
|
end
|
|
|
|
context 'when migrate! is not occupied by another process' do
|
|
it 'executes migrate!' do
|
|
expect(subject).to receive(:object_store=).at_least(1)
|
|
|
|
migrate(to)
|
|
end
|
|
|
|
it 'executes use_file' do
|
|
expect(subject).to receive(:unsafe_use_file).once
|
|
|
|
subject.use_file
|
|
end
|
|
end
|
|
|
|
context 'when migrate! is occupied by another process' do
|
|
include ExclusiveLeaseHelpers
|
|
|
|
before do
|
|
stub_exclusive_lease_taken(subject.exclusive_lease_key, timeout: 1.hour.to_i)
|
|
end
|
|
|
|
it 'does not execute migrate!' do
|
|
expect(subject).not_to receive(:unsafe_migrate!)
|
|
|
|
expect { migrate(to) }.to raise_error(ObjectStorage::ExclusiveLeaseTaken)
|
|
end
|
|
|
|
it 'does not execute use_file' do
|
|
expect(subject).not_to receive(:unsafe_use_file)
|
|
|
|
expect { subject.use_file }.to raise_error(ObjectStorage::ExclusiveLeaseTaken)
|
|
end
|
|
end
|
|
|
|
context 'migration is unsuccessful' do
|
|
shared_examples "handles gracefully" do |error:|
|
|
it 'does not update the object_store' do
|
|
expect { migrate(to) }.to raise_error(error)
|
|
expect(subject.object_store).to eq(from)
|
|
end
|
|
|
|
it 'does not delete the original file' do
|
|
expect { migrate(to) }.to raise_error(error)
|
|
expect(subject.exists?).to be_truthy
|
|
end
|
|
end
|
|
|
|
context 'when the store is not supported' do
|
|
let(:to) { -1 } # not a valid store
|
|
|
|
include_examples "handles gracefully", error: ObjectStorage::UnknownStoreError
|
|
end
|
|
|
|
context 'upon a fog failure' do
|
|
before do
|
|
storage_class = subject.send(:storage_for, to).class
|
|
expect_any_instance_of(storage_class).to receive(:store!).and_raise("Store failure.")
|
|
end
|
|
|
|
include_examples "handles gracefully", error: "Store failure."
|
|
end
|
|
|
|
context 'upon a database failure' do
|
|
before do
|
|
expect(uploader).to receive(:persist_object_store!).and_raise("ActiveRecord failure.")
|
|
end
|
|
|
|
include_examples "handles gracefully", error: "ActiveRecord failure."
|
|
end
|
|
end
|
|
end
|