aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--app/models/compliance_check_message.rb4
-rw-r--r--spec/factories/compliance_check_messages.rb1
2 files changed, 5 insertions, 0 deletions
diff --git a/app/models/compliance_check_message.rb b/app/models/compliance_check_message.rb
index 86899eb15..562c09c0e 100644
--- a/app/models/compliance_check_message.rb
+++ b/app/models/compliance_check_message.rb
@@ -1,4 +1,8 @@
class ComplianceCheckMessage < ActiveRecord::Base
+ extend Enumerize
+
belongs_to :compliance_check
belongs_to :compliance_check_resource
+
+ enumerize :status, in: %i(OK ERROR WARNING IGNORED), scope: true
end
diff --git a/spec/factories/compliance_check_messages.rb b/spec/factories/compliance_check_messages.rb
index 1a047a242..a0d45eb50 100644
--- a/spec/factories/compliance_check_messages.rb
+++ b/spec/factories/compliance_check_messages.rb
@@ -2,6 +2,7 @@ FactoryGirl.define do
factory :compliance_check_message do
association :compliance_check
association :compliance_check_resource
+ status 'OK'
message_key "message_key"
end
end