From 7891efee326d3289a0818ffc31ad459b33c27ec5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Javi=20Mart=C3=ADn?= <javim@elretirao.net>
Date: Thu, 15 Nov 2018 13:04:20 +0100
Subject: [PATCH] Add milestones to proposals

---
 app/models/proposal.rb     |  1 +
 db/dev_seeds/milestones.rb | 25 ++++++++++++++-----------
 2 files changed, 15 insertions(+), 11 deletions(-)

diff --git a/app/models/proposal.rb b/app/models/proposal.rb
index fff616626..aa54827c6 100644
--- a/app/models/proposal.rb
+++ b/app/models/proposal.rb
@@ -20,6 +20,7 @@ class Proposal < ActiveRecord::Base
                accepted_content_types: [ "application/pdf" ]
   include EmbedVideosHelper
   include Relationable
+  include Milestoneable
 
   acts_as_votable
   acts_as_paranoid column: :hidden_at
diff --git a/db/dev_seeds/milestones.rb b/db/dev_seeds/milestones.rb
index d3e2ced68..41b3f40ab 100644
--- a/db/dev_seeds/milestones.rb
+++ b/db/dev_seeds/milestones.rb
@@ -6,17 +6,20 @@ section "Creating default Milestone Statuses" do
 end
 
 section "Creating investment milestones" do
-  Budget::Investment.find_each do |investment|
-    rand(1..5).times do
-      milestone = investment.milestones.build(
-        publication_date: rand(Date.tomorrow..(Date.current + 3.weeks)),
-        status_id: Milestone::Status.all.sample
-      )
-      I18n.available_locales.map do |locale|
-        Globalize.with_locale(locale) do
-          milestone.description = "Description for locale #{locale}"
-          milestone.title = I18n.l(Time.current, format: :datetime)
-          milestone.save!
+  [Budget::Investment, Proposal].each do |model|
+    model.find_each do |record|
+      rand(1..5).times do
+        milestone = record.milestones.build(
+          publication_date: Date.tomorrow,
+          status_id: Milestone::Status.all.sample
+        )
+
+        I18n.available_locales.map do |locale|
+          Globalize.with_locale(locale) do
+            milestone.description = "Description for locale #{locale}"
+            milestone.title = I18n.l(Time.current, format: :datetime)
+            milestone.save!
+          end
         end
       end
     end