Fixing #2250 - Missing templates throw a helpful error

This changes the behaviour of template searching a bit -
we previously usually returned a file name, whether the template
existed or not.  Now we only return a path if it
exists.

Refactoring a few of the the tests for TemplateWrapper, also.

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

1 files changed, 5 insertions, 5 deletions

diff --git a/lib/puppet/parser/files.rb b/lib/puppet/parser/files.rb
index ca4fb4f10..749428e9f 100644
--- a/lib/puppet/parser/files.rb
+++ b/lib/puppet/parser/files.rb
@@ -41,7 +41,7 @@ module Puppet::Parser::Files
 
         if template_paths = templatepath(environment)
             # If we can find the template in :templatedir, we return that.
-            td_file = template_paths.collect { |path|
+            template_paths.collect { |path|
                 File::join(path, template)
             }.each do |f|
                 return f if FileTest.exist?(f)
@@ -49,11 +49,11 @@ module Puppet::Parser::Files
         end
 
         # check in the default template dir, if there is one
-        unless td_file = find_template_in_module(template, environment)
-            raise Puppet::Error, "No valid template directory found, please check templatedir settings" if template_paths.nil?
-            td_file = File::join(template_paths.first, template)
+        if td_file = find_template_in_module(template, environment)
+            return td_file
         end
-        td_file
+
+        return nil
     end
 
     def find_template_in_module(template, environment = nil)