4 files changed, 5 insertions, 1 deletions
diff --git a/app/models/compliance_check_resource.rb b/app/models/compliance_check_resource.rb
index c0cf46c38..a5c0c7d27 100644
--- a/app/models/compliance_check_resource.rb
+++ b/app/models/compliance_check_resource.rb
@@ -1,5 +1,6 @@
 class ComplianceCheckResource < ActiveRecord::Base
   extend Enumerize
+  belongs_to :compliance_check_set
 
   enumerize :status, in: %w[new successful warning failed]
 
diff --git a/app/models/compliance_check_set.rb b/app/models/compliance_check_set.rb
index 3c62221f0..f1f4a9f94 100644
--- a/app/models/compliance_check_set.rb
+++ b/app/models/compliance_check_set.rb
@@ -6,6 +6,8 @@ class ComplianceCheckSet < ActiveRecord::Base
   belongs_to :workbench
   belongs_to :parent, polymorphic: true
 
+  has_many :compliance_check_set
+
   enumerize :status, in: %w[new pending successful warning failed running aborted canceled]
 
   scope :where_created_at_between, ->(start_date, end_date) do
diff --git a/spec/factories/compliance_check_resources.rb b/spec/factories/compliance_check_resources.rb
index 813153be2..95be48de4 100644
--- a/spec/factories/compliance_check_resources.rb
+++ b/spec/factories/compliance_check_resources.rb
@@ -2,5 +2,6 @@ FactoryGirl.define do
   factory :compliance_check_resource do
     status :new
     sequence(:name) { |n| "Compliance check resource #{n}" }
+    association :compliance_check_set
   end
 end
diff --git a/spec/models/compliance_check_resource_spec.rb b/spec/models/compliance_check_resource_spec.rb
index 549cbdd27..40e209db0 100644
--- a/spec/models/compliance_check_resource_spec.rb
+++ b/spec/models/compliance_check_resource_spec.rb
@@ -5,5 +5,5 @@ RSpec.describe ComplianceCheckResource, type: :model do
     expect(FactoryGirl.build(:compliance_check_resource)).to be_valid
   end
 
-  it { is_expected.to validate_presence_of(:compliance_check_set) }
+  it { should validate_presence_of(:compliance_check_set) }
 end