Merge pull request #277 from bwitt/bwitt/manage-service

Manage service

Author: kenyon

REFERENCE.md

@@ -59,6 +59,8 @@ The following parameters are available in the `unattended_upgrades` class:
 * [`whitelist_strict`](#-unattended_upgrades--whitelist_strict)
 * [`allow_downgrade`](#-unattended_upgrades--allow_downgrade)
 * [`dpkg_options`](#-unattended_upgrades--dpkg_options)
+* [`service_ensure`](#-unattended_upgrades--service_ensure)
+* [`service_enable`](#-unattended_upgrades--service_enable)
 
 ##### <a name="-unattended_upgrades--age"></a>`age`
 
@@ -312,6 +314,22 @@ Array of dpkg options.
 
 Default value: `[]`
 
+##### <a name="-unattended_upgrades--service_ensure"></a>`service_ensure`
+
+Data type: `Enum['running', 'stopped']`
+
+Specifies whether the service should be running.
+
+Default value: `'running'`
+
+##### <a name="-unattended_upgrades--service_enable"></a>`service_enable`
+
+Data type: `Boolean`
+
+Specifies whether the service should be enabled at boot.
+
+Default value: `true`
+
 ### <a name="unattended_upgrades--params"></a>`unattended_upgrades::params`
 
 This class provides default parameters for the unattended_upgrades module.

manifests/init.pp

@@ -68,6 +68,10 @@
 #   Allow downgrades.
 # @param dpkg_options
 #   Array of dpkg options.
+# @param service_ensure
+#   Specifies whether the service should be running.
+# @param service_enable
+#   Specifies whether the service should be enabled at boot.
 #
 class unattended_upgrades (
   Unattended_upgrades::Age                  $age                    = {},
@@ -101,6 +105,8 @@
   Optional[Boolean]                         $whitelist_strict       = undef,
   Optional[Boolean]                         $allow_downgrade        = undef,
   Array[String[1]]                          $dpkg_options           = [],
+  Enum['running', 'stopped']                $service_ensure         = 'running',
+  Boolean                                   $service_enable         = true,
 ) inherits unattended_upgrades::params {
   # apt::conf settings require the apt class to work
   include apt
@@ -149,4 +155,10 @@
   if $random_sleep != undef {
     warning('The parameter `random_sleep` is deprecated and will be removed in a future release.')
   }
+
+  service { 'unattended-upgrades':
+    ensure  => $service_ensure,
+    enable  => $service_enable,
+    require => Package['unattended-upgrades'],
+  }
 }

spec/classes/unattended_upgrades_spec.rb

@@ -196,6 +196,38 @@
       end
     end
 
+    context 'service management' do
+      let(:params) do
+        {
+          service_ensure: 'running',
+          service_enable: true
+        }
+      end
+
+      it do
+        is_expected.to contain_service('unattended-upgrades').with(
+          ensure: 'running',
+          enable: true
+        )
+      end
+    end
+
+    context 'service management with stopped service' do
+      let(:params) do
+        {
+          service_ensure: 'stopped',
+          service_enable: false
+        }
+      end
+
+      it do
+        is_expected.to contain_service('unattended-upgrades').with(
+          ensure: 'stopped',
+          enable: false
+        )
+      end
+    end
+
     describe 'validation tests' do
       context 'bad install_on_shutdown' do
         let :params do