lib/draftsman/model.rb in draftsman-0.3.3 vs lib/draftsman/model.rb in draftsman-0.3.4
- old
+ new
@@ -257,14 +257,15 @@
dependent_associations = self.class.reflect_on_all_associations(:has_one) + self.class.reflect_on_all_associations(:has_many)
dependent_associations.each do |association|
if association.klass.draftable? && association.options.has_key?(:dependent) && association.options[:dependent] == :destroy
- dependents = association.macro == :has_one ? [self.send(association.name)] : self.send(association.name)
+ dependents = self.send(association.name)
+ dependents = [dependents] if (dependents && association.macro == :has_one)
dependents.each do |dependent|
dependent.draft_destroy unless dependent.draft? && dependent.send(dependent.class.draft_association_name).destroy?
- end
+ end if dependents
end
end
end
end