Model ComplianceControl

Refs #4385

3 files changed, 26 insertions, 1 deletions

diff --git a/spec/factories/compliance_controls.rb b/spec/factories/compliance_controls.rb
new file mode 100644
index 000000000..865bfe376
--- /dev/null
+++ b/spec/factories/compliance_controls.rb
@@ -0,0 +1,11 @@
+FactoryGirl.define do
+  factory :compliance_control do
+    sequence(:name) { |n| "Compliance control #{n}" }
+    type "Type"
+    criticity :info
+    code "code"
+    comment "Text"
+    compliance_control_set
+    compliance_control_block
+  end
+end
diff --git a/spec/models/compliance_control_block_spec.rb b/spec/models/compliance_control_block_spec.rb
index 836551a7c..248049b0c 100644
--- a/spec/models/compliance_control_block_spec.rb
+++ b/spec/models/compliance_control_block_spec.rb
@@ -2,7 +2,7 @@ require 'rails_helper'
 
 RSpec.describe ComplianceControlBlock, type: :model do
   it 'should have a valid factory' do
-    expect(FactoryGirl.build(:compliance_control_set)).to be_valid
+    expect(FactoryGirl.build(:compliance_control_block)).to be_valid
   end
 
   it { should belong_to :compliance_control_set }
diff --git a/spec/models/compliance_control_spec.rb b/spec/models/compliance_control_spec.rb
new file mode 100644
index 000000000..b00ff4c5a
--- /dev/null
+++ b/spec/models/compliance_control_spec.rb
@@ -0,0 +1,14 @@
+require 'rails_helper'
+
+RSpec.describe ComplianceControl, type: :model do
+  it 'should have a valid factory' do
+    expect(FactoryGirl.build(:compliance_control)).to be_valid
+  end
+
+  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 }
+  it { should validate_presence_of :code }
+end