Fix #1691 - Realize fails with array of Resource References

The following snippet:
realize( File["/tmp/a","/tmp/b"] )

is parsed into:
AST::Function @name=realize @arguments=
    AST::ASTArray @children = [
       AST::ResourceReference @title= AST::ASTArray @children = [
          String(/tmp/a), String(/tmp/b)
       ]
    ]

When evaluated:
ResourceReference gives -> [ File[/tmp/a], File[/tmp/b] ]

which means the function arguments are:
[[File[/tmp/a], File[/tmp/b]]
after evaluating the @arguments ASTArray of AST::Functions.

Then the collector complains that it can't find the resource
because it is not supposed to work on non-flattened resource
array.

The fix is to flatten in the realize function (although it can
be done more generally in the AST::Function evaluation) before
the resources are given to the Collector.

Signed-off-by: Brice Figureau <brice-puppet@daysofwonder.com>

1 files changed, 51 insertions, 0 deletions

diff --git a/spec/unit/parser/functions/realize.rb b/spec/unit/parser/functions/realize.rb
new file mode 100755
index 000000000..d9c94b143
--- /dev/null
+++ b/spec/unit/parser/functions/realize.rb
@@ -0,0 +1,51 @@
+#!/usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../../../spec_helper'
+
+describe "the realize function" do
+
+    before :each do
+        @collector = stub_everything 'collector'
+        @scope = Puppet::Parser::Scope.new()
+        @compiler = stub 'compiler'
+        @compiler.stubs(:add_collection).with(@collector)
+        @scope.stubs(:compiler).returns(@compiler)
+    end
+
+    it "should exist" do
+        Puppet::Parser::Functions.function("realize").should == "function_realize"
+    end
+
+    it "should create a Collector when called" do
+
+        Puppet::Parser::Collector.expects(:new).returns(@collector)
+
+        @scope.function_realize("test")
+    end
+
+    it "should assign the passed-in resources to the collector" do
+        Puppet::Parser::Collector.stubs(:new).returns(@collector)
+
+        @collector.expects(:resources=).with(["test"])
+
+        @scope.function_realize("test")
+    end
+
+    it "should flatten the resources assigned to the collector" do
+        Puppet::Parser::Collector.stubs(:new).returns(@collector)
+
+        @collector.expects(:resources=).with(["test"])
+
+        @scope.function_realize([["test"]])
+    end
+
+    it "should let the compiler know this collector" do
+        Puppet::Parser::Collector.stubs(:new).returns(@collector)
+        @collector.stubs(:resources=).with(["test"])
+
+        @compiler.expects(:add_collection).with(@collector)
+
+        @scope.function_realize("test")
+    end
+
+end