1 files changed, 76 insertions, 0 deletions
diff --git a/spec/unit/parser/ast/ifstatement_spec.rb b/spec/unit/parser/ast/ifstatement_spec.rb
new file mode 100755
index 000000000..83121cddd
--- /dev/null
+++ b/spec/unit/parser/ast/ifstatement_spec.rb
@@ -0,0 +1,76 @@
+#!/usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../../../spec_helper'
+
+describe Puppet::Parser::AST::IfStatement do
+    before :each do
+        @scope = Puppet::Parser::Scope.new()
+    end
+
+    describe "when evaluating" do
+
+        before :each do
+            @test = stub 'test'
+            @test.stubs(:safeevaluate).with(@scope)
+
+            @stmt = stub 'stmt'
+            @stmt.stubs(:safeevaluate).with(@scope)
+
+            @else = stub 'else'
+            @else.stubs(:safeevaluate).with(@scope)
+
+            @ifstmt = Puppet::Parser::AST::IfStatement.new :test => @test, :statements => @stmt
+            @ifelsestmt = Puppet::Parser::AST::IfStatement.new :test => @test, :statements => @stmt, :else => @else
+        end
+
+        it "should evaluate test" do
+            Puppet::Parser::Scope.stubs(:true?).returns(false)
+
+            @test.expects(:safeevaluate).with(@scope)
+
+            @ifstmt.evaluate(@scope)
+        end
+
+        it "should evaluate if statements if test is true" do
+            Puppet::Parser::Scope.stubs(:true?).returns(true)
+
+            @stmt.expects(:safeevaluate).with(@scope)
+
+            @ifstmt.evaluate(@scope)
+        end
+
+        it "should not evaluate if statements if test is false" do
+            Puppet::Parser::Scope.stubs(:true?).returns(false)
+
+            @stmt.expects(:safeevaluate).with(@scope).never
+
+            @ifstmt.evaluate(@scope)
+        end
+
+        it "should evaluate the else branch if test is false" do
+            Puppet::Parser::Scope.stubs(:true?).returns(false)
+
+            @else.expects(:safeevaluate).with(@scope)
+
+            @ifelsestmt.evaluate(@scope)
+        end
+
+        it "should not evaluate the else branch if test is true" do
+            Puppet::Parser::Scope.stubs(:true?).returns(true)
+
+            @else.expects(:safeevaluate).with(@scope).never
+
+            @ifelsestmt.evaluate(@scope)
+        end
+
+        it "should reset ephemeral statements after evaluation" do
+            @scope.expects(:ephemeral_level).returns(:level)
+            Puppet::Parser::Scope.stubs(:true?).returns(true)
+
+            @stmt.expects(:safeevaluate).with(@scope)
+            @scope.expects(:unset_ephemeral_var).with(:level)
+
+            @ifstmt.evaluate(@scope)
+        end
+    end
+end