Simplifying the initialization interface for References

You previously had to call new(nil, "Foo[bar]") if you
just had the resource reference as a string.  Now
you can call new("Foo[bar]"), but the old behaviour
works, too.

Signed-off-by: Luke Kanies <luke@madstop.com>

1 files changed, 10 insertions, 5 deletions

diff --git a/lib/puppet/resource/reference.rb b/lib/puppet/resource/reference.rb
index d254254aa..d17b3e558 100644
--- a/lib/puppet/resource/reference.rb
+++ b/lib/puppet/resource/reference.rb
@@ -21,12 +21,17 @@ class Puppet::Resource::Reference
         builtin_type ? true : false
     end
 
-    def initialize(type, title)
-        # This will set @type if it looks like a resource reference.
-        self.title = title
+    def initialize(argtype, argtitle = nil)
+        if argtitle.nil?
+            self.title = argtype
+            raise ArgumentError, "No title provided and title '%s' is not a valid resource reference" % argtype if self.title == argtype
+        else
+            # This will set @type if it looks like a resource reference.
+            self.title = argtitle
 
-        # Don't override whatever was done by setting the title.
-        self.type = type if self.type.nil?
+            # Don't override whatever was done by setting the title.
+            self.type ||= argtype
+        end
         
         @builtin_type = nil
     end