2019-03-30 07:15:48 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2017-09-06 05:16:26 +00:00
|
|
|
require 'spec_helper'
|
|
|
|
|
2019-01-23 02:31:57 +00:00
|
|
|
describe HashedStorage::ProjectMigrateWorker, :clean_gitlab_redis_shared_state do
|
2018-06-28 18:51:36 +00:00
|
|
|
include ExclusiveLeaseHelpers
|
|
|
|
|
2017-09-06 05:16:26 +00:00
|
|
|
describe '#perform' do
|
2019-01-17 01:53:50 +00:00
|
|
|
let(:project) { create(:project, :empty_repo, :legacy_storage) }
|
2018-06-28 18:51:36 +00:00
|
|
|
let(:lease_key) { "project_migrate_hashed_storage_worker:#{project.id}" }
|
2019-01-17 01:53:50 +00:00
|
|
|
let(:lease_timeout) { described_class::LEASE_TIMEOUT }
|
|
|
|
let(:migration_service) { ::Projects::HashedStorage::MigrationService }
|
2018-06-28 18:51:36 +00:00
|
|
|
|
|
|
|
it 'skips when project no longer exists' do
|
2019-01-17 01:53:50 +00:00
|
|
|
expect(migration_service).not_to receive(:new)
|
2018-06-28 18:51:36 +00:00
|
|
|
|
|
|
|
subject.perform(-1)
|
|
|
|
end
|
2017-09-06 05:16:26 +00:00
|
|
|
|
2018-06-28 18:51:36 +00:00
|
|
|
it 'skips when project is pending delete' do
|
|
|
|
pending_delete_project = create(:project, :empty_repo, pending_delete: true)
|
2017-11-22 15:29:03 +00:00
|
|
|
|
2019-01-17 01:53:50 +00:00
|
|
|
expect(migration_service).not_to receive(:new)
|
2017-09-06 05:16:26 +00:00
|
|
|
|
2018-06-28 18:51:36 +00:00
|
|
|
subject.perform(pending_delete_project.id)
|
|
|
|
end
|
2017-11-22 05:35:53 +00:00
|
|
|
|
2019-01-17 01:53:50 +00:00
|
|
|
it 'delegates migration to service class when we have exclusive lease' do
|
2018-06-28 18:51:36 +00:00
|
|
|
stub_exclusive_lease(lease_key, 'uuid', timeout: lease_timeout)
|
2017-11-22 05:35:53 +00:00
|
|
|
|
2019-01-17 01:53:50 +00:00
|
|
|
service_spy = spy
|
2017-09-06 05:16:26 +00:00
|
|
|
|
2019-01-17 01:53:50 +00:00
|
|
|
allow(migration_service)
|
2018-08-03 14:34:28 +00:00
|
|
|
.to receive(:new).with(project, project.full_path, logger: subject.logger)
|
2019-01-17 01:53:50 +00:00
|
|
|
.and_return(service_spy)
|
2017-09-06 05:16:26 +00:00
|
|
|
|
2018-06-28 18:51:36 +00:00
|
|
|
subject.perform(project.id)
|
2017-11-22 05:35:53 +00:00
|
|
|
|
2019-01-17 01:53:50 +00:00
|
|
|
expect(service_spy).to have_received(:execute)
|
2017-09-06 05:16:26 +00:00
|
|
|
end
|
|
|
|
|
2019-01-17 01:53:50 +00:00
|
|
|
it 'skips when it cant acquire the exclusive lease' do
|
2018-06-28 18:51:36 +00:00
|
|
|
stub_exclusive_lease_taken(lease_key, timeout: lease_timeout)
|
2017-11-22 05:35:53 +00:00
|
|
|
|
2019-01-17 01:53:50 +00:00
|
|
|
expect(migration_service).not_to receive(:new)
|
2017-09-06 05:16:26 +00:00
|
|
|
|
2018-06-28 18:51:36 +00:00
|
|
|
subject.perform(project.id)
|
2017-09-06 05:16:26 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|