diff --git a/db/migrate/20221003144016_add_school_to_cohorts.rb b/db/migrate/20221003144016_add_school_to_cohorts.rb index 251b13f692be20895d4e49b6051809e2ef528ea4..c2fee99576c1dfe4af05755d55e5e52437ef4efc 100644 --- a/db/migrate/20221003144016_add_school_to_cohorts.rb +++ b/db/migrate/20221003144016_add_school_to_cohorts.rb @@ -2,8 +2,11 @@ class AddSchoolToCohorts < ActiveRecord::Migration[6.1] def change add_reference :education_cohorts, :school, foreign_key: {to_table: :education_schools}, type: :uuid Education::Cohort.all.each do |cohort| - cohort.school_id = cohort.program.schools.first.id - cohort.save + if cohort.program.schools.any? + cohort.update(school_id: cohort.program.schools.first.id) + else + cohort.destroy + end end change_column_null :education_cohorts, :school_id, false end