gitlab-org--gitlab-foss/spec/factories/milestones.rb
2017-08-02 17:47:31 -04:00

37 lines
800 B
Ruby

FactoryGirl.define do
factory :milestone do
title
transient do
project nil
group nil
project_id nil
group_id nil
end
trait :active do
state "active"
end
trait :closed do
state "closed"
end
after(:build, :stub) do |milestone, evaluator|
if evaluator.group
milestone.group = evaluator.group
elsif evaluator.group_id
milestone.group_id = evaluator.group_id
elsif evaluator.project
milestone.project = evaluator.project
elsif evaluator.project_id
milestone.project_id = evaluator.project_id
else
milestone.project = create(:project)
end
end
factory :active_milestone, traits: [:active]
factory :closed_milestone, traits: [:closed]
end
end