1 files changed, 42 insertions, 11 deletions
diff --git a/spec/models/compliance_control_spec.rb b/spec/models/compliance_control_spec.rb
index 641209d4f..7c58eaf3e 100644
--- a/spec/models/compliance_control_spec.rb
+++ b/spec/models/compliance_control_spec.rb
@@ -1,20 +1,51 @@
 RSpec.describe ComplianceControl, type: :model do
 
-  let(:compliance_control) { create :compliance_control }
+  context 'standard validation' do
 
-  it 'should have a valid factory' do
-    expect(compliance_control).to be_valid
-  end
+    let(:compliance_control) { build_stubbed :compliance_control }
+
+    it 'should have a valid factory' do
+      expect(compliance_control).to be_valid
+    end
 
-  it { should belong_to :compliance_control_set }
-  it { should belong_to :compliance_control_block }
+    it { should belong_to :compliance_control_set }
+    it { should belong_to :compliance_control_block }
 
 
-  it { should validate_presence_of :criticity }
-  it 'should validate_presence_of :name' do
-    expect( build :compliance_control, name: '' ).to_not be_valid 
+    it { should validate_presence_of :criticity }
+    it 'should validate_presence_of :name' do
+      expect( build :compliance_control, name: '' ).to_not be_valid
+    end
+    it { should validate_presence_of :code }
+    it { should validate_presence_of :origin_code }
+
   end
-  it { should validate_presence_of :code }
-  it { should validate_presence_of :origin_code }
 
+  context 'validates that direct and indirect (via control_block) control_set are not different instances' do
+
+    it 'not attached to control_block -> valid' do
+      compliance_control = create :compliance_control, compliance_control_block_id: nil
+      expect(compliance_control).to be_valid
+    end
+
+    it 'attached to a control_block belonging to the same control_set -> valid' do
+      compliance_control_block = create :compliance_control_block
+      compliance_control = create :compliance_control,
+        compliance_control_block_id: compliance_control_block.id,
+        compliance_control_set_id: compliance_control_block.compliance_control_set.id # DO NOT change the last . to _
+                                                                                      # We need to be sure that is is not nil
+      expect(compliance_control).to be_valid
+    end
+
+    it 'attached to a control_block **not** belonging to the same control_set -> invalid' do
+      compliance_control_block = create :compliance_control_block
+      compliance_control = build :compliance_control,
+        compliance_control_block_id: compliance_control_block.id,
+        compliance_control_set_id: create( :compliance_control_set ).id
+      expect(compliance_control).to_not be_valid
+      selected_error_message =
+        compliance_control.errors.messages[:coherent_control_set].grep(%r{ControlSet associé})
+      expect( selected_error_message ).to_not be_empty
+    end
+  end
 end