Modifying the "Resource#set" method to simplifying adding new parameters

git-svn-id: https://reductivelabs.com/svn/puppet/trunk@2013 980ebf18-57e1-0310-9a29-db15c13687c0

1 files changed, 8 insertions, 1 deletions

diff --git a/lib/puppet/parser/resource.rb b/lib/puppet/parser/resource.rb
index 64c4e1bb9..4a1b2d12a 100644
--- a/lib/puppet/parser/resource.rb
+++ b/lib/puppet/parser/resource.rb
@@ -206,7 +206,14 @@ class Puppet::Parser::Resource
     end
 
     # You have to pass a Resource::Param to this.
-    def set(param)
+    def set(param, value = nil, source = nil)
+        if value and source
+            param = Puppet::Parser::Resource::Param.new(
+                :name => param, :value => value, :source => source
+            )
+        elsif ! param.is_a?(Puppet::Parser::Resource::Param)
+            raise ArgumentError, "Must pass a parameter or all necessary values"
+        end
         # Because definitions are now parse-time, I can paramcheck immediately.
         paramcheck(param.name)