1 files changed, 46 insertions, 0 deletions
diff --git a/test/psych/test_document.rb b/test/psych/test_document.rb
new file mode 100644
index 0000000000..e5ef00c746
--- /dev/null
+++ b/test/psych/test_document.rb
@@ -0,0 +1,46 @@
+require 'test/psych/helper'
+
+module Psych
+  class TestDocument < TestCase
+    def setup
+      super
+      @stream = Psych.parse_stream(<<-eoyml)
+%YAML 1.1
+%TAG ! tag:tenderlovemaking.com,2009:
+--- !fun
+      eoyml
+      @doc = @stream.children.first
+    end
+
+    def test_parse_tag
+      assert_equal([['!', 'tag:tenderlovemaking.com,2009:']],
+        @doc.tag_directives)
+    end
+
+    def test_emit_tag
+      assert_match('%TAG ! tag:tenderlovemaking.com,2009:', @stream.to_yaml)
+    end
+
+    def test_emit_multitag
+      @doc.tag_directives << ['!!', 'foo.com,2009:']
+      yaml = @stream.to_yaml
+      assert_match('%TAG ! tag:tenderlovemaking.com,2009:', yaml)
+      assert_match('%TAG !! foo.com,2009:', yaml)
+    end
+
+    def test_emit_bad_tag
+      assert_raises(RuntimeError) do
+        @doc.tag_directives = [['!']]
+        @stream.to_yaml
+      end
+    end
+
+    def test_parse_version
+      assert_equal([1,1], @doc.version)
+    end
+
+    def test_emit_version
+      assert_match('%YAML 1.1', @stream.to_yaml)
+    end
+  end
+end