diff --git a/lib/rexml/parsers/streamparser.rb b/lib/rexml/parsers/streamparser.rb index b271e6743e..f6a8bfa802 100644 --- a/lib/rexml/parsers/streamparser.rb +++ b/lib/rexml/parsers/streamparser.rb @@ -7,6 +7,7 @@ module REXML def initialize source, listener @listener = listener @parser = BaseParser.new( source ) + @tag_stack = [] end def add_listener( listener ) @@ -19,14 +20,21 @@ module REXML event = @parser.pull case event[0] when :end_document + unless @tag_stack.empty? + tag_path = "/" + @tag_stack.join("/") + raise ParseException.new("Missing end tag for '#{tag_path}'", + @parser.source) + end return when :start_element + @tag_stack << event[1] attrs = event[2].each do |n, v| event[2][n] = @parser.unnormalize( v ) end @listener.tag_start( event[1], attrs ) when :end_element @listener.tag_end( event[1] ) + @tag_stack.pop when :text normalized = @parser.unnormalize( event[1] ) @listener.text( normalized ) diff --git a/test/rexml/parser/test_stream.rb b/test/rexml/parser/test_stream.rb new file mode 100644 index 0000000000..c315833e4b --- /dev/null +++ b/test/rexml/parser/test_stream.rb @@ -0,0 +1,32 @@ +require "test/unit" +require "rexml/document" +require "rexml/streamlistener" + +module REXMLTests + class TestStreamParser < Test::Unit::TestCase + class NullListener + include REXML::StreamListener + end + + class TestInvalid < self + def test_no_end_tag + xml = "" + exception = assert_raise(REXML::ParseException) do + parse(xml) + end + assert_equal(<<-MESSAGE, exception.to_s) +Missing end tag for '/root/sub' +Line: 1 +Position: #{xml.bytesize} +Last 80 unconsumed characters: + MESSAGE + end + + private + def parse(xml, listener=nil) + listener ||= NullListener.new + REXML::Document.parse_stream(xml, listener) + end + end + end +end