From b81cbd6de7944443f9a024acb5fa6cdfdfeaac63 Mon Sep 17 00:00:00 2001
From: cedricnjanga
Date: Mon, 13 Nov 2017 12:33:15 +0100
Subject: Refs #4952 Fix bug related to compliance control set action links
 display

---
 app/controllers/compliance_control_blocks_controller.rb | 1 +
 app/controllers/compliance_control_sets_controller.rb   | 1 +
 app/controllers/compliance_controls_controller.rb       | 1 +
 3 files changed, 3 insertions(+)

(limited to 'app/controllers')

diff --git a/app/controllers/compliance_control_blocks_controller.rb b/app/controllers/compliance_control_blocks_controller.rb
index 996f8a464..f53a1e04a 100644
--- a/app/controllers/compliance_control_blocks_controller.rb
+++ b/app/controllers/compliance_control_blocks_controller.rb
@@ -1,4 +1,5 @@
 class ComplianceControlBlocksController < InheritedResources::Base
+  include PolicyChecker
   defaults resource_class: ComplianceControlBlock
   belongs_to :compliance_control_set
   actions :all, :except => [:show, :index]
diff --git a/app/controllers/compliance_control_sets_controller.rb b/app/controllers/compliance_control_sets_controller.rb
index 956f26112..5c78d09e3 100644
--- a/app/controllers/compliance_control_sets_controller.rb
+++ b/app/controllers/compliance_control_sets_controller.rb
@@ -1,4 +1,5 @@
 class ComplianceControlSetsController < InheritedResources::Base
+  include PolicyChecker
   defaults resource_class: ComplianceControlSet
   include RansackDateFilter
   before_action only: [:index] { set_date_time_params("updated_at", DateTime) }
diff --git a/app/controllers/compliance_controls_controller.rb b/app/controllers/compliance_controls_controller.rb
index 816d8897b..5a9c0b671 100644
--- a/app/controllers/compliance_controls_controller.rb
+++ b/app/controllers/compliance_controls_controller.rb
@@ -1,4 +1,5 @@
 class ComplianceControlsController < InheritedResources::Base
+  include PolicyChecker
   defaults resource_class: ComplianceControl
   belongs_to :compliance_control_set
   actions :all, :except => [:index]
-- 
cgit v1.2.3