CHEF-5372: Support specific run_levels for RedHat service

1 files changed, 20 insertions, 4 deletions

diff --git a/lib/chef/provider/service/redhat.rb b/lib/chef/provider/service/redhat.rb
index 19cd2aa485..a471d4bbe4 100644
--- a/lib/chef/provider/service/redhat.rb
+++ b/lib/chef/provider/service/redhat.rb
@@ -1,6 +1,6 @@
 #
 # Author:: AJ Christensen (<aj@hjksolutions.com>)
-# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# Copyright:: Copyright (c) 2008-2015 Chef Software, Inc.
 # License:: Apache License, Version 2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -39,6 +39,7 @@ class Chef
           @init_command = "/sbin/service #{@new_resource.service_name}"
           @new_resource.supports[:status] = true
           @service_missing = false
+          @run_levels = @new_resource.run_levels
         end
 
         def define_resource_requirements
@@ -62,19 +63,34 @@ class Chef
 
           if ::File.exists?("/sbin/chkconfig")
             chkconfig = shell_out!("/sbin/chkconfig --list #{@current_resource.service_name}", :returns => [0,1])
-            @current_resource.enabled(!!(chkconfig.stdout =~ CHKCONFIG_ON))
+            unless @run_levels.nil? or @run_levels.empty?
+              chkconfig.split(/\s+/)[1..-1].each do |level|
+                index = level.split(':').first
+                status = level.split(':').last
+                is_enabled = false
+                is_enabled = true if @run_levels.include?(index) and status =~ CHKCONFIG_ON
+                @current_resource.enabled(is_enabled)
+              end
+            else
+              @current_resource.enabled(!!(chkconfig.stdout =~ CHKCONFIG_ON))
+            end
             @service_missing = !!(chkconfig.stderr =~ CHKCONFIG_MISSING)
           end
 
           @current_resource
         end
 
+        def levels
+          (@run_levels.nil? or @run_levels.empty?) ? levels = "" : levels = "--level #{@run_levels.join('')} "
+          levels
+        end
+
         def enable_service()
-          shell_out! "/sbin/chkconfig #{@new_resource.service_name} on"
+          shell_out! "/sbin/chkconfig #{levels}#{@new_resource.service_name} on"
         end
 
         def disable_service()
-          shell_out! "/sbin/chkconfig #{@new_resource.service_name} off"
+          shell_out! "/sbin/chkconfig #{levels}#{@new_resource.service_name} off"
         end
       end
     end