spec/factories/milestones.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

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) 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(:empty_project)
      end
    end

    factory :active_milestone, traits: [:active]
    factory :closed_milestone, traits: [:closed]
  end
end