From af9a92086df8fcfbb572eb3d2bc11eee383ef9e4 Mon Sep 17 00:00:00 2001 From: Luke Kanies Date: Thu, 21 Jan 2010 22:45:43 -0800 Subject: Adding an environment helper Signed-off-by: Luke Kanies --- spec/unit/node/environment.rb | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) (limited to 'spec/unit/node') diff --git a/spec/unit/node/environment.rb b/spec/unit/node/environment.rb index 81981973d..8945f28df 100755 --- a/spec/unit/node/environment.rb +++ b/spec/unit/node/environment.rb @@ -216,4 +216,28 @@ describe Puppet::Node::Environment do end end end + + describe Puppet::Node::Environment::Helper do + before do + @helper = Object.new + @helper.extend(Puppet::Node::Environment::Helper) + end + + it "should be able to set and retrieve the environment" do + @helper.environment = :foo + @helper.environment.name.should == :foo + end + + it "should accept an environment directly" do + env = Puppet::Node::Environment.new :foo + @helper.environment = env + @helper.environment.name.should == :foo + end + + it "should accept an environment as a string" do + env = Puppet::Node::Environment.new "foo" + @helper.environment = env + @helper.environment.name.should == :foo + end + end end -- cgit