From d663e4c59476d315cc976d7d7922b4392a29ec3d Mon Sep 17 00:00:00 2001
From: kikito <kikito@gmail.com>
Date: Mon, 23 May 2016 19:50:11 +0200
Subject: [PATCH] Adds/improves factories

---
 spec/factories.rb | 32 +++++++++++++++++++++++---------
 1 file changed, 23 insertions(+), 9 deletions(-)

diff --git a/spec/factories.rb b/spec/factories.rb
index 0ae660e9a..85047b511 100644
--- a/spec/factories.rb
+++ b/spec/factories.rb
@@ -1,5 +1,4 @@
 FactoryGirl.define do
-
   sequence(:document_number) { |n| "#{n.to_s.rjust(8, '0')}X" }
 
   factory :user do
@@ -192,6 +191,19 @@ FactoryGirl.define do
   factory :budget do
     sequence(:name) { |n| "Budget #{n}" }
     currency_symbol "€"
+    phase 'on_hold'
+
+    trait :selecting do
+      phase 'selecting'
+    end
+
+    trait :balloting do
+      phase 'balloting'
+    end
+
+    trait :finished do
+      phase 'finished'
+    end
   end
 
   factory :budget_heading, class: 'Budget::Heading' do
@@ -200,13 +212,18 @@ FactoryGirl.define do
     price 1000000
   end
 
-  factory :budget_investment, class: 'Budget::Investment' do
-    sequence(:title)     { |n| "Investment #{n} title" }
+  factory :budget_investment, class: Budget::Investment do
+    sequence(:title)     { |n| "Budget Investment #{n} title" }
+    association :budget
+    association :author, factory: :user
     description          'Spend money on this'
-    price                1000
+    unfeasibility_explanation ''
     external_url         'http://external_documention.org'
     terms_of_service     '1'
-    association :author, factory: :user
+
+    trait :with_confidence_score do
+      before(:save) { |i| i.calculate_confidence_score }
+    end
 
     trait :feasible do
       feasibility "feasible"
@@ -359,8 +376,5 @@ FactoryGirl.define do
     census_code { '01' }
   end
 
-  factory :budget do
-    sequence(:name) {|n| "Budget #{n}" }
-    phase "on_hold"
-  end
+
 end