aboutsummaryrefslogtreecommitdiffstats
path: root/app
diff options
context:
space:
mode:
Diffstat (limited to 'app')
-rw-r--r--app/models/compliance_check_message.rb3
-rw-r--r--app/models/compliance_check_set.rb3
2 files changed, 5 insertions, 1 deletions
diff --git a/app/models/compliance_check_message.rb b/app/models/compliance_check_message.rb
index 562c09c0e..738bd4a4b 100644
--- a/app/models/compliance_check_message.rb
+++ b/app/models/compliance_check_message.rb
@@ -1,8 +1,11 @@
class ComplianceCheckMessage < ActiveRecord::Base
extend Enumerize
+ belongs_to :compliance_check_set
belongs_to :compliance_check
belongs_to :compliance_check_resource
enumerize :status, in: %i(OK ERROR WARNING IGNORED), scope: true
+
+ validates_presence_of :compliance_check_set
end
diff --git a/app/models/compliance_check_set.rb b/app/models/compliance_check_set.rb
index f1f4a9f94..4c00c3516 100644
--- a/app/models/compliance_check_set.rb
+++ b/app/models/compliance_check_set.rb
@@ -6,7 +6,8 @@ class ComplianceCheckSet < ActiveRecord::Base
belongs_to :workbench
belongs_to :parent, polymorphic: true
- has_many :compliance_check_set
+ has_many :compliance_check_resources
+ has_many :compliance_check_messages
enumerize :status, in: %w[new pending successful warning failed running aborted canceled]