to_yaml function added.

lib/puppet/parser/functions/to_yaml.rb

@@ -0,0 +1,21 @@
+#
+# to_yaml.rb
+#
+
+module Puppet::Parser::Functions
+  newfunction(:to_yaml, :type => :rvalue, :doc => <<-EOS
+This function accepts a puppet data structure (I.E hash/array/string or nested combination of these) and converts it into a yaml format string, so that YAML files can be written using the content param in the file provider.
+    EOS
+  ) do |arguments|
+
+    if (arguments.size != 1) then
+      raise(Puppet::ParseError, "to_yaml(): Wrong number of arguments "+
+        "given #{arguments.size} for 1")
+    end
+
+    ZAML.dump(arguments[0])
+
+  end
+end
+
+# vim: set ts=2 sw=2 et :

spec/functions/to_yaml_spec.rb

@@ -0,0 +1,28 @@
+#! /usr/bin/env ruby -S rspec
+require 'spec_helper'
+
+describe "the to_yaml function" do
+  let(:scope) { PuppetlabsSpec::PuppetInternals.scope }
+
+  it "should exist" do
+    Puppet::Parser::Functions.function("to_yaml").should == "function_to_yaml"
+  end
+
+  it "should raise a ParseError if there is less than 1 arguments" do
+    lambda { scope.function_to_yaml([]) }.should( raise_error(Puppet::ParseError))
+  end
+
+  it "should convert a data structure to YAML" do
+    yaml_data = {
+      'string' => 'a string',
+      'array' => ['aaa', 'bbb', 'ccc' ],
+      'bool' => true,
+      'hash' => {
+        'string' => 'indeed it is',
+        'another_string' => 'quite so'
+      },
+    }
+    result = scope.function_to_yaml([yaml_data])
+    result.should(eq("#{yaml_data.to_yaml}"))
+  end
+end