gitlab-org--gitlab-foss/spec/factories/container_repositories.rb

103 lines
2.7 KiB
Ruby

# frozen_string_literal: true
FactoryBot.define do
factory :container_repository do
sequence(:name) { |n| "test_image_#{n}" }
project
transient do
tags { [] }
end
trait :root do
name { '' }
end
trait :status_delete_scheduled do
status { :delete_scheduled }
end
trait :status_delete_failed do
status { :delete_failed }
end
trait :cleanup_scheduled do
expiration_policy_cleanup_status { :cleanup_scheduled }
end
trait :cleanup_unfinished do
expiration_policy_cleanup_status { :cleanup_unfinished }
end
trait :cleanup_ongoing do
expiration_policy_cleanup_status { :cleanup_ongoing }
end
trait :default do
migration_state { 'default' }
end
trait :pre_importing do
migration_state { 'pre_importing' }
migration_pre_import_started_at { Time.zone.now }
end
trait :pre_import_done do
migration_state { 'pre_import_done' }
migration_pre_import_started_at { Time.zone.now }
migration_pre_import_done_at { Time.zone.now }
end
trait :importing do
migration_state { 'importing' }
migration_pre_import_started_at { Time.zone.now }
migration_pre_import_done_at { Time.zone.now }
migration_import_started_at { Time.zone.now }
end
trait :import_done do
migration_state { 'import_done' }
migration_pre_import_started_at { Time.zone.now }
migration_pre_import_done_at { Time.zone.now }
migration_import_started_at { Time.zone.now }
migration_import_done_at { Time.zone.now }
end
trait :import_aborted do
migration_state { 'import_aborted' }
migration_pre_import_started_at { Time.zone.now }
migration_pre_import_done_at { Time.zone.now }
migration_import_started_at { Time.zone.now }
migration_aborted_at { Time.zone.now }
migration_aborted_in_state { 'importing' }
migration_retries_count { 1 }
end
trait :import_skipped do
migration_state { 'import_skipped' }
migration_skipped_at { Time.zone.now }
migration_skipped_reason { :too_many_tags }
end
after(:build) do |repository, evaluator|
next if evaluator.tags.to_a.none?
tags = evaluator.tags
# convert Array into Hash
tags = tags.product(['sha256:4c8e63ca4cb663ce6c688cb06f1c372b088dac5b6d7ad7d49cd620d85cf72a15']).to_h unless tags.is_a?(Hash)
stub_method(repository.client, :repository_tags) do |*args|
{
'name' => repository.path,
'tags' => tags.keys
}
end
tags.each_pair do |tag, digest|
allow(repository.client)
.to receive(:repository_tag_digest)
.with(repository.path, tag)
.and_return(digest)
end
end
end
end