104 lines
2.7 KiB
Ruby
104 lines
2.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
FactoryBot.define do
|
|
factory :protected_branch do
|
|
sequence(:name) { |n| "protected_branch_#{n}" }
|
|
project
|
|
|
|
transient do
|
|
ee { false }
|
|
default_push_level { true }
|
|
default_merge_level { true }
|
|
default_access_level { true }
|
|
end
|
|
|
|
after(:create) do |protected_branch, evaluator|
|
|
# Do not use `break` because it will cause `LocalJumpError`
|
|
next unless protected_branch.project&.persisted?
|
|
|
|
ProtectedBranches::CacheService.new(protected_branch.project).refresh
|
|
end
|
|
|
|
after(:build) do |obj, ctx|
|
|
next if ctx.ee || !ctx.default_access_level
|
|
|
|
obj.push_access_levels.new(access_level: Gitlab::Access::MAINTAINER) if ctx.default_push_level
|
|
obj.merge_access_levels.new(access_level: Gitlab::Access::MAINTAINER) if ctx.default_merge_level
|
|
end
|
|
|
|
trait :create_branch_on_repository do
|
|
association :project, factory: [:project, :repository]
|
|
|
|
transient do
|
|
repository_branch_name { name }
|
|
end
|
|
|
|
after(:create) do |protected_branch, evaluator|
|
|
project = protected_branch.project
|
|
|
|
project.repository.create_branch(evaluator.repository_branch_name, project.default_branch_or_main)
|
|
end
|
|
end
|
|
|
|
trait :no_one_can_merge do
|
|
transient do
|
|
default_merge_level { false }
|
|
end
|
|
|
|
after(:build) do |protected_branch|
|
|
protected_branch.merge_access_levels.new(access_level: Gitlab::Access::NO_ACCESS)
|
|
end
|
|
end
|
|
|
|
trait :developers_can_merge do
|
|
transient do
|
|
default_merge_level { false }
|
|
end
|
|
|
|
after(:build) do |protected_branch|
|
|
protected_branch.merge_access_levels.new(access_level: Gitlab::Access::DEVELOPER)
|
|
end
|
|
end
|
|
|
|
trait :maintainers_can_merge do
|
|
transient do
|
|
default_merge_level { false }
|
|
end
|
|
|
|
after(:build) do |protected_branch|
|
|
protected_branch.merge_access_levels.new(access_level: Gitlab::Access::MAINTAINER)
|
|
end
|
|
end
|
|
|
|
trait :no_one_can_push do
|
|
transient do
|
|
default_push_level { false }
|
|
end
|
|
|
|
after(:build) do |protected_branch|
|
|
protected_branch.push_access_levels.new(access_level: Gitlab::Access::NO_ACCESS)
|
|
end
|
|
end
|
|
|
|
trait :developers_can_push do
|
|
transient do
|
|
default_push_level { false }
|
|
end
|
|
|
|
after(:build) do |protected_branch|
|
|
protected_branch.push_access_levels.new(access_level: Gitlab::Access::DEVELOPER)
|
|
end
|
|
end
|
|
|
|
trait :maintainers_can_push do
|
|
transient do
|
|
default_push_level { false }
|
|
end
|
|
|
|
after(:build) do |protected_branch|
|
|
protected_branch.push_access_levels.new(access_level: Gitlab::Access::MAINTAINER)
|
|
end
|
|
end
|
|
end
|
|
end
|