From 33b565a0d76b07f354e2e145e7e03e5a5db69e48 Mon Sep 17 00:00:00 2001
From: Markus Roberts <Markus@reality.com>
Date: Thu, 25 Mar 2010 15:54:33 -0700
Subject: Fix for #3424 and tests to prove it.

The original pure ruby yaml patch missed some edge cases; specifically, classes
that were modified by the syck version to directly call it and thus never
reached the pure ruby version.  This adds monkey patches to all of those case
which we might reasonably care about (omitting, for example, calls within the
syck version to itself) and tests which show that the monkey patch works.
---
 lib/puppet/util/monkey_patches.rb | 10 ++++++----
 spec/unit/util/zaml.rb            | 38 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 44 insertions(+), 4 deletions(-)
 create mode 100644 spec/unit/util/zaml.rb

diff --git a/lib/puppet/util/monkey_patches.rb b/lib/puppet/util/monkey_patches.rb
index 05d319356..53ae1b9c0 100644
--- a/lib/puppet/util/monkey_patches.rb
+++ b/lib/puppet/util/monkey_patches.rb
@@ -20,11 +20,13 @@ class Symbol
     end
 end
 
-class Object
-    def to_yaml
-        ZAML.dump(self)
+[Object, Exception, Integer, Struct, Date, Time, Range, Regexp, Hash, Array, Float, String, FalseClass, TrueClass, Symbol, NilClass, Class].each { |cls|
+    cls.class_eval do
+        def to_yaml
+            ZAML.dump(self)
+        end
     end
-end
+}
 
 def YAML.dump(*args)
     ZAML.dump(*args)
diff --git a/spec/unit/util/zaml.rb b/spec/unit/util/zaml.rb
new file mode 100644
index 000000000..1f21c4e29
--- /dev/null
+++ b/spec/unit/util/zaml.rb
@@ -0,0 +1,38 @@
+#!/usr/bin/env ruby
+
+Dir.chdir(File.dirname(__FILE__)) { (s = lambda { |f| File.exist?(f) ? require(f) : Dir.chdir("..") { s.call(f) } }).call("spec/spec_helper.rb") }
+
+require 'puppet/util/monkey_patches'
+
+describe "Pure ruby yaml implementation" do
+    {
+    7            => "--- 7",
+    3.14159      => "--- 3.14159",
+    'test'       => "--- test",
+    []           => "--- []",
+    :symbol      => "--- !ruby/sym symbol",
+    {:a => "A"}  => "--- \n  !ruby/sym a: A"
+    }.each { |o,y| 
+        it "should convert the #{o.class} #{o.inspect} to yaml" do
+            o.to_yaml.should == y
+        end
+        it "should produce yaml for the #{o.class} #{o.inspect} that can be reconstituted" do
+            YAML.load(o.to_yaml).should == o
+        end
+    }
+    #
+    # Can't test for equality on raw objects
+    {
+    Object.new                   => "--- !ruby/object {}",
+    [Object.new]                 => "--- \n  - !ruby/object {}",
+    {Object.new => Object.new}   => "--- \n  ? !ruby/object {}\n  : !ruby/object {}"
+    }.each { |o,y| 
+        it "should convert the #{o.class} #{o.inspect} to yaml" do
+            o.to_yaml.should == y
+        end
+        it "should produce yaml for the #{o.class} #{o.inspect} that can be reconstituted" do
+            lambda { YAML.load(o.to_yaml) }.should_not raise_error
+        end
+    }
+end
+
-- 
cgit