Merge pull request #8772 from MsysTechnologiesllc/dh/MSYS-1057_fix_knife_node_environment

fix knife node environment set output

2 files changed, 6 insertions, 2 deletions

diff --git a/lib/chef/knife/node_environment_set.rb b/lib/chef/knife/node_environment_set.rb
index 95aab72625..644b6138b6 100644
--- a/lib/chef/knife/node_environment_set.rb
+++ b/lib/chef/knife/node_environment_set.rb
@@ -44,8 +44,7 @@ class Chef
 
         node.save
 
-        config[:attribute] = "chef_environment"
-
+        config[:environment] = @environment
         output(format_for_display(node))
       end
 
diff --git a/spec/unit/knife/node_environment_set_spec.rb b/spec/unit/knife/node_environment_set_spec.rb
index 5b704c4219..6a6d48cc2f 100644
--- a/spec/unit/knife/node_environment_set_spec.rb
+++ b/spec/unit/knife/node_environment_set_spec.rb
@@ -47,6 +47,11 @@ describe Chef::Knife::NodeEnvironmentSet do
       @knife.run
     end
 
+    it "sets the environment to config for display" do
+      @knife.run
+      expect(@knife.config[:environment]).to eq("bar")
+    end
+
     it "should print the environment" do
       expect(@knife).to receive(:output).and_return(true)
       @knife.run