[#4090] Change how RunMode instances are created so that an object for each RunMode is only created once instead of every time it's called

Got lots of unpredictable test failures, presumably because a new
RunMode was being created every time we accessed the RunMode.

2 files changed, 7 insertions, 1 deletions

diff --git a/lib/puppet/application.rb b/lib/puppet/application.rb
index c49f42fd7..8aa3708aa 100644
--- a/lib/puppet/application.rb
+++ b/lib/puppet/application.rb
@@ -228,7 +228,7 @@ class Application
             return @run_mode if @run_mode and not mode_name
 
             require 'puppet/util/run_mode'
-            @run_mode = Puppet::Util::RunMode.new( mode_name || :user )
+            @run_mode = Puppet::Util::RunMode[ mode_name || :user ]
         end
     end
 
diff --git a/lib/puppet/util/run_mode.rb b/lib/puppet/util/run_mode.rb
index 08f2c851f..029c1f921 100644
--- a/lib/puppet/util/run_mode.rb
+++ b/lib/puppet/util/run_mode.rb
@@ -5,8 +5,14 @@ module Puppet
                 @name = name.to_sym
             end
 
+            @@run_modes = Hash.new {|h, k| h[k] = RunMode.new(k)}
+
             attr :name
 
+            def self.[](name)
+                @@run_modes[name]
+            end
+
             def master?
                 name == :master
             end