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