Add support for dist-upgrade & autoremove action

[puppet-modules/puppetlabs-apt.git] / tasks / init.rb

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 }