Add milestones to proposals

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

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