Add support for dist-upgrade & autoremove action

diff --git a/tasks/init.json b/tasks/init.json
index 202000593e6578551eac1a54d07f6def10765db1..91cf9514d84dcb5d31cb68add219dc9d5dac50c7 100644 (file)
--- a/tasks/init.json
+++ b/tasks/init.json
@@ -4,7 +4,7 @@
   "parameters": {
     "action": {
       "description": "Action to perform ",
-      "type": "Enum[update, upgrade]"
+      "type": "Enum[update, upgrade, dist-upgrade, autoremove]"
     }
   }
 }

diff --git a/tasks/init.rb b/tasks/init.rb
index b4cb0dc7622744be6334972eeb720ae1d89faf08..f8ca85e2f5d81126a49146e785622ba4551cb0b2 100755 (executable)
--- a/tasks/init.rb
+++ b/tasks/init.rb
@@ -5,7 +5,7 @@ require 'puppet'
 
 def apt_get(action)
   cmd = ['apt-get', action]
-  cmd << '-y' if action == 'upgrade'
+  cmd << '-y' if ['upgrade', 'dist-upgrade', 'autoremove'].include?(action)
   stdout, stderr, status = Open3.capture3(*cmd)
   raise Puppet::Error, stderr if status != 0 # rubocop:disable GetText/DecorateFunctionMessage
   { status: stdout.strip }