supporting add_domain_ype syntax

1 files changed, 16 insertions, 0 deletions

diff --git a/test/test_psych.rb b/test/test_psych.rb
index 690bf23..4bea1b3 100644
--- a/test/test_psych.rb
+++ b/test/test_psych.rb
@@ -18,4 +18,20 @@ class TestPsych < MiniTest::Unit::TestCase
     }
     assert_equal %w{ foo bar }, docs
   end
+
+  def test_domain_types
+    got = nil
+    Psych.add_domain_type 'foo.bar,2002', 'foo' do |type, val|
+      got = val
+    end
+
+    Psych.load('--- !foo.bar,2002/foo hello')
+    assert_equal 'hello', got
+
+    Psych.load("--- !foo.bar,2002/foo\n- hello\n- world")
+    assert_equal %w{ hello world }, got
+
+    Psych.load("--- !foo.bar,2002/foo\nhello: world")
+    assert_equal({ 'hello' => 'world' }, got)
+  end
 end