Refs #4626 Add spec to validate presence of compliance_check_set for ComplianceCheckResource

2 files changed, 4 insertions, 0 deletions

diff --git a/app/models/compliance_check_resource.rb b/app/models/compliance_check_resource.rb
index 7826f00c3..c0cf46c38 100644
--- a/app/models/compliance_check_resource.rb
+++ b/app/models/compliance_check_resource.rb
@@ -2,4 +2,6 @@ class ComplianceCheckResource < ActiveRecord::Base
   extend Enumerize
 
   enumerize :status, in: %w[new successful warning failed]
+
+  validates_presence_of :compliance_check_set
 end
diff --git a/spec/models/compliance_check_resource_spec.rb b/spec/models/compliance_check_resource_spec.rb
index a9366bea2..549cbdd27 100644
--- a/spec/models/compliance_check_resource_spec.rb
+++ b/spec/models/compliance_check_resource_spec.rb
@@ -4,4 +4,6 @@ RSpec.describe ComplianceCheckResource, type: :model do
   it 'should have a valid factory' do
     expect(FactoryGirl.build(:compliance_check_resource)).to be_valid
   end
+
+  it { is_expected.to validate_presence_of(:compliance_check_set) }
 end