1 files changed, 47 insertions, 0 deletions
diff --git a/test/rexml/parse/test_document_type_declaration.rb b/test/rexml/parse/test_document_type_declaration.rb
new file mode 100644
index 0000000000..7d527a5fe8
--- /dev/null
+++ b/test/rexml/parse/test_document_type_declaration.rb
@@ -0,0 +1,47 @@
+require "test/unit"
+require "rexml/document"
+
+class TestParseDocumentTypeDeclaration < Test::Unit::TestCase
+  private
+  def xml(internal_subset)
+    <<-XML
+<!DOCTYPE r SYSTEM "urn:x-rexml:test" [
+#{internal_subset}
+]>
+<r/>
+    XML
+  end
+
+  def parse(internal_subset)
+    REXML::Document.new(xml(internal_subset)).doctype
+  end
+
+  class TestMixed < self
+    def test_entity_element
+      doctype = parse(<<-INTERNAL_SUBSET)
+<!ENTITY entity-name "entity content">
+<!ELEMENT element-name EMPTY>
+      INTERNAL_SUBSET
+      assert_equal([REXML::Entity, REXML::ElementDecl],
+                   doctype.children.collect(&:class))
+    end
+
+    def test_attlist_entity
+      doctype = parse(<<-INTERNAL_SUBSET)
+<!ATTLIST attribute-list-name attribute-name CDATA #REQUIRED>
+<!ENTITY entity-name "entity content">
+      INTERNAL_SUBSET
+      assert_equal([REXML::AttlistDecl, REXML::Entity],
+                   doctype.children.collect(&:class))
+    end
+
+    def test_notation_attlist
+      doctype = parse(<<-INTERNAL_SUBSET)
+<!NOTATION notation-name SYSTEM "system-literal">
+<!ATTLIST attribute-list-name attribute-name CDATA #REQUIRED>
+      INTERNAL_SUBSET
+      assert_equal([REXML::NotationDecl, REXML::AttlistDecl],
+                   doctype.children.collect(&:class))
+    end
+  end
+end