testing implcit document ends

1 files changed, 11 insertions, 2 deletions

diff --git a/lib/psych/tree_builder.rb b/lib/psych/tree_builder.rb
index efbeaf9..089ee96 100644
--- a/lib/psych/tree_builder.rb
+++ b/lib/psych/tree_builder.rb
@@ -17,7 +17,6 @@ module Psych
     end
 
     %w{
-      Document
       Sequence
       Mapping
     }.each do |node|
@@ -28,12 +27,22 @@ module Psych
           @stack.push n
         end
 
-        def end_#{node.downcase}(*args)
+        def end_#{node.downcase}
           @stack.pop
         end
       }
     end
 
+    def start_document(*args)
+      n = Nodes::Document.new(*args)
+      @stack.last.children << n
+      @stack.push n
+    end
+
+    def end_document implicit_end
+      @stack.pop.implicit_end = implicit_end
+    end
+
     def start_stream encoding
       @stack.push Nodes::Stream.new(encoding)
     end