Merge pull request #314 from amateo/feature/loadyaml_check_file

Check if file exists before loading with loadyaml. If not, return nil

2 files changed, 12 insertions, 1 deletions

diff --git a/lib/puppet/parser/functions/loadyaml.rb b/lib/puppet/parser/functions/loadyaml.rb
index 10c4005..ca655f6 100644
--- a/lib/puppet/parser/functions/loadyaml.rb
+++ b/lib/puppet/parser/functions/loadyaml.rb
@@ -13,7 +13,12 @@ module Puppet::Parser::Functions
       raise Puppet::ParseError, ("loadyaml(): wrong number of arguments (#{args.length}; must be 1)")
     end
 
-    YAML.load_file(args[0])
+    if File.exists?(args[0]) then
+      YAML.load_file(args[0])
+    else
+      warning("Can't load " + args[0] + ". File does not exist!")
+      nil
+    end
 
   end
 
diff --git a/spec/functions/loadyaml_spec.rb b/spec/functions/loadyaml_spec.rb
index cdc3d6f..4b8d406 100755
--- a/spec/functions/loadyaml_spec.rb
+++ b/spec/functions/loadyaml_spec.rb
@@ -14,6 +14,12 @@ describe "the loadyaml function" do
     expect { scope.function_loadyaml([]) }.to raise_error(Puppet::ParseError)
   end
 
+  it "should return nil when file does not exist" do
+    yaml_file = tmpfilename ('yamlfile')
+    result = scope.function_loadyaml([yaml_file])
+    expect(result).to(eq(nil))
+  end
+
   it "should convert YAML file to a data structure" do
     yaml_file = tmpfilename ('yamlfile')
     File.open(yaml_file, 'w') do |fh|