diff options
| -rw-r--r-- | app/controllers/compliance_controls_controller.rb | 1 | ||||
| -rw-r--r-- | app/models/compliance_control.rb | 4 |
2 files changed, 5 insertions, 0 deletions
diff --git a/app/controllers/compliance_controls_controller.rb b/app/controllers/compliance_controls_controller.rb index 87dfcf15b..c5c76fd58 100644 --- a/app/controllers/compliance_controls_controller.rb +++ b/app/controllers/compliance_controls_controller.rb @@ -1,4 +1,5 @@ class ComplianceControlsController < BreadcrumbController + include PolicyChecker defaults resource_class: ComplianceControl belongs_to :compliance_control_set diff --git a/app/models/compliance_control.rb b/app/models/compliance_control.rb index 12ff4737a..46902f9b1 100644 --- a/app/models/compliance_control.rb +++ b/app/models/compliance_control.rb @@ -9,4 +9,8 @@ class ComplianceControl < ActiveRecord::Base validates :name, presence: true validates :code, presence: true validates :compliance_control_set, presence: true + + def self.policy_class + ComplianceControlPolicy + end end |