warning should be true only in env run

Signed-off-by: nikhil2611 <[email protected]>

Author: nikhil2611

lib/chef-cli/command/env.rb

@@ -91,9 +91,9 @@ def gem_environment
         h = {}
         if habitat_install?
           # Habitat-specific environment variables
-          h["GEM ROOT"] = habitat_env["GEM_ROOT"]
-          h["GEM HOME"] = habitat_env["GEM_HOME"]
-          h["GEM PATHS"] = habitat_env["GEM_PATH"].split(File::PATH_SEPARATOR)
+          h["GEM ROOT"] = habitat_env(show_warning: true)["GEM_ROOT"]
+          h["GEM HOME"] = habitat_env(show_warning: true)["GEM_HOME"]
+          h["GEM PATHS"] = habitat_env(show_warning: true)["GEM_PATH"].split(File::PATH_SEPARATOR)
         elsif omnibus_install?
           # Omnibus-specific environment variables
           h["GEM ROOT"] = omnibus_env["GEM_ROOT"]
@@ -109,7 +109,7 @@ def gem_environment
       end
 
       def paths
-        env = habitat_install? ? habitat_env : omnibus_env
+        env = habitat_install? ? habitat_env(show_warning: true) : omnibus_env
         env["PATH"].split(File::PATH_SEPARATOR)
       rescue OmnibusInstallNotFound
         ENV["PATH"].split(File::PATH_SEPARATOR)

lib/chef-cli/helpers.rb

@@ -154,15 +154,15 @@ def git_windows_bin_dir
     #
     # environment vars for habitat
     #
-    def habitat_env
+    def habitat_env(show_warning: false)
       @habitat_env ||=
       begin
         if habitat_chef_dke?
           bin_pkg_prefix = get_pkg_prefix(ChefCLI::Dist::CHEF_DKE_PKG_NAME)
         end
         versioned_pkg_prefix = fetch_chef_cli_version_pkg if ENV["CHEF_CLI_VERSION"]
 
-        if ENV["CHEF_CLI_VERSION"] && !versioned_pkg_prefix
+        if show_warning && ENV["CHEF_CLI_VERSION"] && !versioned_pkg_prefix
           ChefCLI::UI.new.msg("Warning: Habitat package '#{ChefCLI::Dist::HAB_PKG_NAME}' with version '#{ENV["CHEF_CLI_VERSION"]}' not found.")
         end
         # Use the first available package for bin_pkg_prefix