inspec 4.7.24 → 4.10.4

data/lib/plugins/inspec-habitat/test/functional/inspec_habitat_test.rb

@@ -5,7 +5,6 @@ class ProfileCli < Minitest::Test
   include CorePluginFunctionalHelper
 
   def setup
-    skip_windows!
     @tmpdir = Dir.mktmpdir
     @habitat_profile = File.join(@tmpdir, "habitat-profile")
     run_inspec_process("init profile " + @habitat_profile)
@@ -18,22 +17,18 @@ class ProfileCli < Minitest::Test
   def test_setup_subcommand
     result = run_inspec_process("habitat profile setup " + @habitat_profile + " --log-level debug")
 
-    # Command runs without error
-    assert_empty result.stderr
-    assert_equal 0, result.exit_status
-
     # Command creates only expected files
     base_dir = File.join(@tmpdir, "habitat-profile", "habitat")
     files = %w{
-      default.toml
       plan.sh
-      config
-      config/inspec_exec_config.json
-      hooks
-      hooks/run
     }
     actual_files = Dir.glob(File.join(base_dir, "**/*"))
     expected_files = files.map { |x| File.join(base_dir, x) }
     assert_equal actual_files.sort, expected_files.sort
+
+    # Command runs without error
+    assert_empty result.stderr
+
+    assert_exit_code 0, result
   end
 end

data/lib/plugins/inspec-init/templates/plugins/inspec-plugin-template/test/functional/inspec_plugin_template_test.rb

@@ -35,7 +35,7 @@ describe 'inspec list-resources core' do
     # Some tests through here use minitest Expectations, which attach to all
     # Objects, and begin with 'must' (positive) or 'wont' (negative)
     # See http://docs.seattlerb.org/minitest/Minitest/Expectations.html
-    it("should exit successfully") { outcome.exit_status.must_equal(0) }
+    it("should exit successfully") { assert_exit_code 0, outcome }
     it("should be silent on stderr") { outcome.stderr.must_be_empty }
 
     # A selection of core resources, just spot checking.
@@ -59,7 +59,7 @@ describe 'inspec list-resources core' do
     let(:outcome) { run_inspec_process_with_this_plugin("listresources core user") }
 
     # Should be well-behaved...
-    it("should exit successfully") { outcome.exit_status.must_equal(0) }
+    it("should exit successfully") { assert_exit_code 0, outcome }
     it("should be silent on stderr") { outcome.stderr.must_be_empty }
 
     # Here, we want to know it DID match some things, and NOT some others.
@@ -79,7 +79,7 @@ describe 'inspec list-resources core' do
     let(:outcome) { run_inspec_process_with_this_plugin("listresources core autogyro") }
 
     # Should be well-behaved...
-    it("should exit successfully") { outcome.exit_status.must_equal(0) }
+    it("should exit successfully") { assert_exit_code 0, outcome }
     it("should be silent on stderr") { outcome.stderr.must_be_empty }
 
     # Output lines should be just two, for the summary.
@@ -99,7 +99,7 @@ describe 'inspec list-resources core' do
     let(:outcome) { run_inspec_process_with_this_plugin('listresources core --no-summary') }
 
     # Should be well-behaved...
-    it("should exit successfully") { outcome.exit_status.must_equal(0) }
+    it("should exit successfully") { assert_exit_code 0, outcome }
     it("should be silent on stderr") { outcome.stderr.must_be_empty }
 
     # Check for the summary

data/lib/plugins/inspec-init/test/functional/inspec_init_plugin_test.rb

@@ -3,18 +3,18 @@ require_relative "../../../shared/core_plugin_test_helper.rb"
 class InitPluginCli < Minitest::Test
   include CorePluginFunctionalHelper
 
-  def setup
-    skip_windows!
-  end
-
   def test_generating_inspec_plugin_correct_prefix_required
     Dir.mktmpdir do |dir|
       plugin = "wacky-name"
       run_result = run_inspec_process("init plugin --no-prompt #{plugin} ", prefix: "cd #{dir} &&")
-      assert_equal 1, run_result.exit_status
-      assert_empty run_result.stderr
+
+      skip_windows!
       assert_includes run_result.stdout, "ERROR"
       assert_includes run_result.stdout, "Plugin names must begin with"
+
+      assert_empty run_result.stderr
+
+      assert_exit_code 1, run_result
     end
   end
 
@@ -25,12 +25,15 @@ class InitPluginCli < Minitest::Test
       module_name = plugin.sub(/^inspec\-/, "").split("-").map(&:capitalize).join("")
 
       run_result = run_inspec_process("init plugin --no-prompt #{plugin}", prefix: "cd #{dir} &&")
-      assert_empty run_result.stderr
 
-      assert_equal 0, run_result.exit_status
+      skip_windows!
       assert_includes run_result.stdout, "Creating new inspec plugin at"
       assert_includes run_result.stdout, plugin
 
+      assert_empty run_result.stderr
+
+      assert_exit_code 0, run_result
+
       # Check generated files and contents.
       # Each file must exist, and its contents must match each of the regexen given.
       {
@@ -128,11 +131,15 @@ class InitPluginCli < Minitest::Test
       opts += " --module_name FunPlugin"
 
       run_result = run_inspec_process("init plugin #{plugin} --no-prompt #{opts}", prefix: "cd #{dir} &&")
-      assert_empty run_result.stderr
-      assert_equal 0, run_result.exit_status
+
+      skip_windows!
       assert_includes run_result.stdout, "Creating new inspec plugin at"
       assert_includes run_result.stdout, plugin
 
+      assert_empty run_result.stderr
+
+      assert_exit_code 0, run_result
+
       # Check generated files and contents.
       # Each file must exist, and its contents must match each of the regexen given.
       {

data/lib/plugins/inspec-init/test/functional/inspec_init_profile_test.rb

@@ -4,19 +4,18 @@ require_relative "../../../shared/core_plugin_test_helper.rb"
 class InitCli < Minitest::Test
   include CorePluginFunctionalHelper
 
-  def setup
-    skip_windows!
-  end
-
   def test_generating_inspec_profile
     Dir.mktmpdir do |dir|
       profile = File.join(dir, "test-profile")
       out = run_inspec_process("init profile test-profile", prefix: "cd #{dir} &&")
-      assert_equal 0, out.exit_status
+
+      skip_windows!
       assert_includes out.stdout, "Creating new profile at"
       assert_includes out.stdout, profile
       assert_includes Dir.entries(profile).join, "inspec.yml"
       assert_includes Dir.entries(profile).join, "README.md"
+
+      assert_exit_code 0, out
     end
   end
 
@@ -24,20 +23,26 @@ class InitCli < Minitest::Test
     Dir.mktmpdir do |dir|
       profile = File.join(dir, "test-profile")
       out = run_inspec_process("init profile --platform os test-profile", prefix: "cd #{dir} &&")
-      assert_equal 0, out.exit_status
+
+      skip_windows!
       assert_includes out.stdout, "Creating new profile at"
       assert_includes out.stdout, profile
       assert_includes Dir.entries(profile).join, "inspec.yml"
       assert_includes Dir.entries(profile).join, "README.md"
+
+      assert_exit_code 0, out
     end
   end
 
   def test_generating_inspec_profile_with_bad_platform
     Dir.mktmpdir do |dir|
       out = run_inspec_process("init profile --platform nonesuch test-profile", prefix: "cd #{dir} &&")
-      assert_equal 1, out.exit_status
+
+      skip_windows!
       assert_includes out.stdout, "Unable to generate profile"
       assert_includes out.stdout, "No template available for platform 'nonesuch'"
+
+      assert_exit_code 1, out
     end
   end
 
@@ -45,10 +50,12 @@ class InitCli < Minitest::Test
     Dir.mktmpdir do |dir|
       profile = dir + "/test/deeper/profile"
       out = run_inspec_process("init profile test/deeper/profile", prefix: "cd #{dir} &&")
-      assert_equal 0, out.exit_status
+      skip_windows!
       assert_equal true, File.exist?(profile)
       profile = YAML.load_file("#{profile}/inspec.yml")
       assert_equal "profile", profile["name"]
+
+      assert_exit_code 0, out
     end
   end
 
@@ -56,11 +63,14 @@ class InitCli < Minitest::Test
     Dir.mktmpdir do |dir|
       profile = File.join(dir, "test-gcp-profile")
       out = run_inspec_process("init profile --platform gcp test-gcp-profile", prefix: "cd #{dir} &&")
-      assert_equal 0, out.exit_status
+
+      skip_windows!
       assert_includes out.stdout, "Creating new profile at"
       assert_includes out.stdout, profile
       assert_includes Dir.entries(profile).join, "inspec.yml"
       assert_includes Dir.entries(profile).join, "README.md"
+
+      assert_exit_code 0, out
     end
   end
 
@@ -68,11 +78,14 @@ class InitCli < Minitest::Test
     Dir.mktmpdir do |dir|
       profile = File.join(dir, "test-aws-profile")
       out = run_inspec_process("init profile --platform aws test-aws-profile", prefix: "cd #{dir} &&")
-      assert_equal 0, out.exit_status
+
+      skip_windows!
       assert_includes out.stdout, "Creating new profile at"
       assert_includes out.stdout, profile
       assert_includes Dir.entries(profile).join, "inspec.yml"
       assert_includes Dir.entries(profile).join, "README.md"
+
+      assert_exit_code 0, out
     end
   end
 
@@ -80,11 +93,14 @@ class InitCli < Minitest::Test
     Dir.mktmpdir do |dir|
       profile = File.join(dir, "test-azure-profile")
       out = run_inspec_process("init profile --platform azure test-azure-profile", prefix: "cd #{dir} &&")
-      assert_equal 0, out.exit_status
+
+      skip_windows!
       assert_includes out.stdout, "Creating new profile at"
       assert_includes out.stdout, profile
       assert_includes Dir.entries(profile).join, "inspec.yml"
       assert_includes Dir.entries(profile).join, "README.md"
+
+      assert_exit_code 0, out
     end
   end
 
@@ -92,11 +108,14 @@ class InitCli < Minitest::Test
     Dir.mktmpdir do |dir|
       profile = File.join(dir, "test-os-profile")
       out = run_inspec_process("init profile --platform os test-os-profile", prefix: "cd #{dir} &&")
-      assert_equal 0, out.exit_status
+
+      skip_windows!
       assert_includes out.stdout, "Creating new profile at"
       assert_includes out.stdout, profile
       assert_includes Dir.entries(profile).join, "inspec.yml"
       assert_includes Dir.entries(profile).join, "README.md"
+
+      assert_exit_code 0, out
     end
   end
 end

data/lib/plugins/inspec-plugin-manager-cli/lib/inspec-plugin-manager-cli/cli_command.rb

@@ -1,4 +1,3 @@
-require "term/ansicolor"
 require "pathname"
 require "inspec/plugin/v2"
 require "inspec/plugin/v2/installer"
@@ -7,7 +6,6 @@ require "inspec/dist"
 module InspecPlugins
   module PluginManager
     class CliCommand < Inspec.plugin(2, :cli_command)
-      include Term::ANSIColor
       include Inspec::Dist
 
       subcommand_desc "plugin SUBCOMMAND", "Manage #{PRODUCT_NAME} and Train plugins"
@@ -22,15 +20,17 @@ module InspecPlugins
         plugin_statuses = Inspec::Plugin::V2::Registry.instance.plugin_statuses
         plugin_statuses.reject! { |s| %i{core bundle}.include?(s.installation_type) } unless options[:all]
 
-        # TODO: ui object support
         puts
-        puts(bold { format(" %-30s%-10s%-8s%-6s", "Plugin Name", "Version", "Via", "ApiVer") })
-        puts "-" * 55
+        ui.bold(format(" %-30s%-10s%-8s%-6s", "Plugin Name", "Version", "Via", "ApiVer"))
+        ui.line
         plugin_statuses.sort_by(&:name).each do |status|
-          puts(format(" %-30s%-10s%-8s%-6s", status.name, make_pretty_version(status), status.installation_type, status.api_generation.to_s))
+          ui.plain(format(" %-30s%-10s%-8s%-6s", status.name,
+                          make_pretty_version(status),
+                          status.installation_type,
+                          status.api_generation.to_s))
         end
-        puts "-" * 55
-        puts(" #{plugin_statuses.count} plugin(s) total")
+        ui.line
+        ui.plain(" #{plugin_statuses.count} plugin(s) total")
         puts
       end
 
@@ -59,23 +59,22 @@ module InspecPlugins
           search_results.delete("train-test-fixture")
         end
 
-        # TODO: ui object support
         puts
-        puts(bold { format(" %-30s%-50s", "Plugin Name", "Versions Available") })
-        puts "-" * 55
+        ui.bold(format(" %-30s%-50s", "Plugin Name", "Versions Available"))
+        ui.line
         search_results.keys.sort.each do |plugin_name|
           versions = options[:all] ? search_results[plugin_name] : [search_results[plugin_name].first]
           versions = "(" + versions.join(", ") + ")"
-          puts(format(" %-30s%-50s", plugin_name, versions))
+          ui.plain(format(" %-30s%-50s", plugin_name, versions))
         end
-        puts "-" * 55
-        puts(" #{search_results.count} plugin(s) found")
+        ui.line
+        ui.plain(" #{search_results.count} plugin(s) found")
         puts
 
-        exit 2 if search_results.empty?
+        ui.exit Inspec::UI::EXIT_PLUGIN_ERROR if search_results.empty?
       rescue Inspec::Plugin::V2::SearchError => ex
         Inspec::Log.error ex.message
-        exit 1
+        ui.exit Inspec::UI::EXIT_USAGE_ERROR
       end
 
       #==================================================================#
@@ -119,13 +118,14 @@ module InspecPlugins
         begin
           installer.update(plugin_name)
         rescue Inspec::Plugin::V2::UpdateError => ex
-          puts(red { "Update error: " } + ex.message + " - update failed")
-          exit 1
+          ui.plain("#{ui.red('Update error:')} #{ex.message} - update failed")
+          ui.exit Inspec::UI::EXIT_USAGE_ERROR
         end
         post_update_versions = installer.list_installed_plugin_gems.select { |spec| spec.name == plugin_name }.map { |spec| spec.version.to_s }
         new_version = (post_update_versions - pre_update_versions).first
 
-        puts(bold { plugin_name } + " plugin, version #{old_version} -> #{new_version}, updated from rubygems.org")
+        ui.bold(plugin_name + " plugin, version #{old_version} -> " \
+                "#{new_version}, updated from rubygems.org")
       end
 
       #--------------------------
@@ -144,9 +144,9 @@ module InspecPlugins
       def uninstall(plugin_name)
         status = Inspec::Plugin::V2::Registry.instance[plugin_name.to_sym]
         unless status
-          puts(red { "No such plugin installed: " } + "#{plugin_name} is not installed - uninstall failed")
-
-          exit 1
+          ui.plain("#{ui.red('No such plugin installed:')} #{plugin_name} is not " \
+                 "installed - uninstall failed")
+          ui.exit Inspec::UI::EXIT_USAGE_ERROR
         end
         installer = Inspec::Plugin::V2::Installer.instance
 
@@ -156,11 +156,13 @@ module InspecPlugins
         installer.uninstall(plugin_name)
 
         if status.installation_type == :path
-          puts(bold { plugin_name } + " path-based plugin install has been uninstalled")
+          ui.bold(plugin_name + " path-based plugin install has been " \
+                  "uninstalled")
         else
-          puts(bold { plugin_name } + " plugin, version #{old_version}, has been uninstalled")
+          ui.bold(plugin_name + " plugin, version #{old_version}, has " \
+                  "been uninstalled")
         end
-        exit 0
+        ui.exit Inspec::UI::EXIT_NORMAL
       end
 
       private
@@ -172,8 +174,8 @@ module InspecPlugins
 
       def install_from_gemfile(gem_file)
         unless File.exist? gem_file
-          puts(red { "No such plugin gem file " } + gem_file + " - installation failed.")
-          exit 1
+          ui.red("No such plugin gem file #{gem_file} - installation failed.")
+          ui.exit Inspec::UI::EXIT_USAGE_ERROR
         end
 
         plugin_name_parts = File.basename(gem_file, ".gem").split("-")
@@ -183,14 +185,15 @@ module InspecPlugins
 
         installer.install(plugin_name, gem_file: gem_file)
 
-        puts(bold { plugin_name } + " plugin, version #{version}, installed from local .gem file")
-        exit 0
+        ui.bold("#{plugin_name} plugin, version #{version}, installed from " \
+                "local .gem file")
+        ui.exit Inspec::UI::EXIT_NORMAL
       end
 
       def install_from_path(path)
         unless File.exist? path
-          puts(red { "No such source code path " } + path + " - installation failed.")
-          exit 1
+          ui.red("No such source code path #{path} - installation failed.")
+          ui.exit Inspec::UI::EXIT_USAGE_ERROR
         end
 
         plugin_name = File.basename(path, ".rb")
@@ -204,8 +207,10 @@ module InspecPlugins
 
         # Already installed?
         if registry.known_plugin?(plugin_name.to_sym)
-          puts(red { "Plugin already installed" } + " - #{plugin_name} - Use '#{EXEC_NAME} plugin list' to see previously installed plugin - installation failed.")
-          exit 2
+          ui.red("Plugin already installed - #{plugin_name} - Use '#{EXEC_NAME} " \
+                 "plugin list' to see previously installed plugin - " \
+                 "installation failed.")
+          ui.exit Inspec::UI::EXIT_PLUGIN_ERROR
         end
 
         # Can we figure out how to load it?
@@ -217,8 +222,9 @@ module InspecPlugins
         # OK, install it!
         installer.install(plugin_name, path: entry_point)
 
-        puts(bold { plugin_name } + " plugin installed via source path reference, resolved to entry point " + entry_point)
-        exit 0
+        ui.bold("#{plugin_name} plugin installed via source path reference, " \
+                "resolved to entry point #{entry_point}")
+        ui.exit Inspec::UI::EXIT_NORMAL
       end
 
       # Rationale for rubocop variances: It's a heuristics method, and will be full of
@@ -229,6 +235,7 @@ module InspecPlugins
         given = given.expand_path # Resolve any relative paths
         name_regex = /^(inspec|train)-/
         versioned_regex = /^(inspec|train)-[a-z0-9\-\_]+-\d+\.\d+\.\d+$/
+        sha_ref_regex = /^(inspec|train)-[a-z0-9\-\_]+-[0-9a-f]{5,40}$/
 
         # What are the last four things like?
         parts = [
@@ -257,14 +264,14 @@ module InspecPlugins
         # In that case, we'll have a version on the plugin name in part 0
         # /home/you/.gems/2.4.0/gems/inspec-something-3.45.1/lib/inspec-something.rb
         #   parts index:                     ^0^             ^1^      ^2^         ^3^
-        if parts[0] =~ versioned_regex && parts[1] == "lib" && parts[0].start_with?(parts[2]) && parts[3] == ".rb"
+        if (parts[0] =~ versioned_regex || parts[0] =~ sha_ref_regex) && parts[1] == "lib" && parts[0].start_with?(parts[2]) && parts[3] == ".rb"
           return given.to_s
         end
 
         # Case 4: Like case 3, but missing the .rb
         # /home/you/.gems/2.4.0/gems/inspec-something-3.45.1/lib/inspec-something
         #   parts index:                     ^0^             ^1^      ^2^         ^3^ (empty)
-        if parts[0] =~ versioned_regex && parts[1] == "lib" && parts[0].start_with?(parts[2]) && parts[3].empty?
+        if (parts[0] =~ versioned_regex || parts[0] =~ sha_ref_regex) && parts[1] == "lib" && parts[0].start_with?(parts[2]) && parts[3].empty?
           return given.to_s + ".rb"
         end
 
@@ -279,8 +286,10 @@ module InspecPlugins
 
         # Well, if we got here, parts[2] matches an inspec/train prefix, but we have no idea about anything.
         # Give up.
-        puts(red { "Unrecognizable plugin structure" } + " - #{parts[2]} - When installing from a path, please provide the path of the entry point file - installation failed.")
-        exit 1
+        ui.red("Unrecognizable plugin structure - #{parts[2]} - When " \
+               "installing from a path, please provide the path of the " \
+               "entry point file - installation failed.")
+        ui.exit Inspec::UI::EXIT_USAGE_ERROR
       end
 
       def install_from_path__probe_load(entry_point, plugin_name)
@@ -288,9 +297,11 @@ module InspecPlugins
         begin
           require entry_point
         rescue LoadError => ex
-          puts(red { "Plugin contains errors" } + " - #{plugin_name} - Encountered errors while trying to test load the plugin entry point, resolved to #{entry_point} - installation failed")
-          puts ex.message
-          exit 1
+          ui.red("Plugin contains errors - #{plugin_name} - Encountered " \
+                 "errors while trying to test load the plugin entry point, " \
+                 "resolved to #{entry_point} - installation failed")
+          ui.plain ex.message
+          ui.exit Inspec::UI::EXIT_USAGE_ERROR
         end
 
         # OK, the wheels didn't fall off.  But is it a plugin?
@@ -299,13 +310,19 @@ module InspecPlugins
           # And the registry is keyed on Strings
           registry_key = plugin_name.to_s.sub(/^train-/, "")
           unless Train::Plugins.registry.key?(registry_key)
-            puts(red { "Does not appear to be a plugin" } + " - #{plugin_name} - After probe-loading the supposed plugin, it did not register itself to Train. Ensure something inherits from 'Train.plugin(1)' - installation failed.")
-            exit 1
+            ui.red("Does not appear to be a plugin - #{plugin_name} - After " \
+                   "probe-loading the supposed plugin, it did not register " \
+                   "itself to Train. Ensure something inherits from " \
+                   "'Train.plugin(1)' - installation failed.")
+            ui.exit Inspec::UI::EXIT_USAGE_ERROR
           end
         else
           unless registry.known_plugin?(plugin_name.to_sym)
-            puts(red { "Does not appear to be a plugin" } + " - #{plugin_name} - After probe-loading the supposed plugin, it did not register itself to InSpec. Ensure something inherits from 'Inspec.plugin(2)' - installation failed.")
-            exit 1
+            ui.red("Does not appear to be a plugin - #{plugin_name} - After " \
+                   "probe-loading the supposed plugin, it did not register " \
+                   "itself to InSpec. Ensure something inherits from " \
+                   "'Inspec.plugin(2)' - installation failed.")
+            ui.exit Inspec::UI::EXIT_USAGE_ERROR
           end
         end
       end
@@ -325,8 +342,9 @@ module InspecPlugins
         post_installed_versions = installer.list_installed_plugin_gems.select { |spec| spec.name == plugin_name }.map { |spec| spec.version.to_s }
         new_version = (post_installed_versions - pre_installed_versions).first
 
-        puts(bold { plugin_name } + " plugin, version #{new_version}, installed from rubygems.org")
-        exit 0
+        ui.bold("#{plugin_name} plugin, version #{new_version}, installed " \
+                "from rubygems.org")
+        ui.exit Inspec::UI::EXIT_NORMAL
       end
 
       def install_from_remote_gem_verson_preflight_check(plugin_name, requested_version, pre_installed_versions)
@@ -348,38 +366,50 @@ module InspecPlugins
         they_explicitly_asked_for_a_version = !options[:version].nil?
         what_we_would_install_is_already_installed = pre_installed_versions.include?(requested_version)
         if what_we_would_install_is_already_installed && they_explicitly_asked_for_a_version
-          puts(red { "Plugin already installed at requested version" } + " - plugin #{plugin_name} #{requested_version} - refusing to install.")
+          ui.red("Plugin already installed at requested version - plugin " \
+                 "#{plugin_name} #{requested_version} - refusing to install.")
         elsif what_we_would_install_is_already_installed && !they_explicitly_asked_for_a_version
-          puts(red { "Plugin already installed at latest version" } + " - plugin #{plugin_name} #{requested_version} - refusing to install.")
+          ui.red("Plugin already installed at latest version - plugin " \
+                 "#{plugin_name} #{requested_version} - refusing to install.")
         else
           # There are existing versions installed, but none of them are what was requested
-          puts(red { "Update required" } + " - plugin #{plugin_name}, requested #{requested_version}, have #{pre_installed_versions.join(", ")}; use `inspec plugin update` - refusing to install.")
+          ui.red("Update required - plugin #{plugin_name}, requested " \
+                 "#{requested_version}, have " \
+                 "#{pre_installed_versions.join(', ')}; use `inspec " \
+                 "plugin update` - refusing to install.")
         end
 
-        exit 2
+        ui.exit Inspec::UI::EXIT_PLUGIN_ERROR
       end
 
       # Rationale for RuboCop variance: This is a one-line method with heavy UX-focused error handling.
       def install_attempt_install(plugin_name) # rubocop: disable Metrics/AbcSize
         installer.install(plugin_name, version: options[:version])
       rescue Inspec::Plugin::V2::PluginExcludedError => ex
-        puts(red { "Plugin on Exclusion List" } + " - #{plugin_name} is listed as an incompatible gem - refusing to install.")
-        puts "Rationale: #{ex.details.rationale}"
-        puts "Exclusion list location: " + File.join(Inspec.src_root, "etc", "plugin_filters.json")
-        puts "If you disagree with this determination, please accept our apologies for the misunderstanding, and open an issue at https://github.com/inspec/inspec/issues/new"
-        exit 2
+        ui.red("Plugin on Exclusion List - #{plugin_name} is listed as an " \
+               "incompatible gem - refusing to install.")
+        ui.plain("Rationale: #{ex.details.rationale}")
+        ui.plain("Exclusion list location: " +
+                 File.join(Inspec.src_root, "etc", "plugin_filters.json"))
+        ui.plain("If you disagree with this determination, please accept " \
+                 "our apologies for the misunderstanding, and open an issue " \
+                 "at https://github.com/inspec/inspec/issues/new")
+        ui.exit Inspec::UI::EXIT_PLUGIN_ERROR
       rescue Inspec::Plugin::V2::InstallError
         raise if Inspec::Log.level == :debug
 
         results = installer.search(plugin_name, exact: true)
         if results.empty?
-          puts(red { "No such plugin gem " } + plugin_name + " could be found on rubygems.org - installation failed.")
+          ui.red("No such plugin gem #{plugin_name} could be found on " \
+                 "rubygems.org - installation failed.")
         elsif options[:version] && !results[plugin_name].include?(options[:version])
-          puts(red { "No such version" } + " - " + plugin_name + " exists, but no such version #{options[:version]} found on rubygems.org - installation failed.")
+          ui.red("No such version - #{plugin_name} exists, but no such " \
+                 "version #{options[:version]} found on rubygems.org - " \
+                 "installation failed.")
         else
-          puts(red { "Unknown error occured " } + " - installation failed.")
+          ui.red("Unknown error occured - installation failed.")
         end
-        exit 1
+        ui.exit Inspec::UI::EXIT_USAGE_ERROR
       end
 
       #==================================================================#
@@ -390,11 +420,14 @@ module InspecPlugins
           # Check for path install
           status = Inspec::Plugin::V2::Registry.instance[plugin_name.to_sym]
           if !status
-            puts(red { "No such plugin installed: " } + "#{plugin_name} - update failed")
-            exit 1
+            ui.plain("#{ui.red('No such plugin installed:')} #{plugin_name} - update failed")
+            ui.exit Inspec::UI::EXIT_USAGE_ERROR
           elsif status.installation_type == :path
-            puts(red { "Cannot update path-based install: " } + "#{plugin_name} is installed via path reference; use `inspec plugin uninstall` to remove - refusing to update")
-            exit 2
+            ui.plain("#{ui.red('Cannot update path-based install:')} " \
+                   "#{plugin_name} is installed via path reference; " \
+                   "use `inspec plugin uninstall` to remove - refusing to" \
+                   "update")
+            ui.exit Inspec::UI::EXIT_PLUGIN_ERROR
           end
         end
 
@@ -403,8 +436,10 @@ module InspecPlugins
         latest_version = latest_version[plugin_name]&.last
 
         if pre_update_versions.include?(latest_version)
-          puts(red { "Already installed at latest version: " } + "#{plugin_name} is at #{latest_version}, which the latest - refusing to update")
-          exit 2
+          ui.plain("#{ui.red('Already installed at latest version:')} " \
+                   "#{plugin_name} is at #{latest_version}, which the " \
+                   "latest - refusing to update")
+          ui.exit Inspec::UI::EXIT_PLUGIN_ERROR
         end
       end
 
@@ -421,8 +456,10 @@ module InspecPlugins
 
       def check_plugin_name(plugin_name, action)
         unless plugin_name =~ /^(inspec|train)-/
-          puts(red { "Invalid plugin name" } + " - #{plugin_name} - All inspec plugins must begin with either 'inspec-' or 'train-' - #{action} failed.")
-          exit 1
+          ui.red("Invalid plugin name - #{plugin_name} - All inspec " \
+                 "plugins must begin with either 'inspec-' or 'train-' " \
+                 "- #{action} failed.")
+          ui.exit Inspec::UI::EXIT_USAGE_ERROR
         end
       end