Merge pull request #999 from transitland/entity-attribute-validate

Entity attribute validate
transitland · Mar 1, 2017 · aa96a95 · aa96a95
2 parents 79b32f7 + b70aa41
commit aa96a95
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/app/models/entity_with_issues.rb b/app/models/entity_with_issues.rb
@@ -18,4 +18,12 @@
 class EntityWithIssues < ActiveRecord::Base
   belongs_to :issue, dependent: :destroy
   belongs_to :entity, polymorphic: true
+
+  validate :entity_attribute_exists?
+
+  def entity_attribute_exists?
+    unless entity.attributes.include?(entity_attribute) || entity_attribute.nil?
+      errors.add(:entity_attribute, "#{entity_attribute} does not exist")
+    end
+  end
 end