X-Git-Url: https://review.fuel-infra.org/gitweb?a=blobdiff_plain;f=tasks%2Finit.rb;h=43053128a25dbd1edad4a4f041424df26c17f338;hb=2c6b0c7cc6ce2f2ee7fcf7afd29943e55379a7e4;hp=d256468b64a739fdaa6446f49b1d5069bc46891d;hpb=cef56276873b534415210abefd28454c321038c3;p=puppet-modules%2Fpuppetlabs-apt.git

diff --git a/tasks/init.rb b/tasks/init.rb
index d256468..4305312 100755
--- a/tasks/init.rb
+++ b/tasks/init.rb
@@ -1,11 +1,13 @@
 #!/opt/puppetlabs/puppet/bin/ruby
+# frozen_string_literal: true
+
 require 'json'
 require 'open3'
 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
   { status: stdout.strip }