From c65c0f2f403798d3284e22aec0f4e733c9889538 Mon Sep 17 00:00:00 2001
From: Yoonchae <bluegreenmagick@gmail.com>
Date: Mon, 25 Oct 2021 15:12:14 +0900
Subject: [PATCH] skip initial whitespaces after DocType sample_4 test is not
 enabled yet because of CData

---
 src/events/mod.rs                  |  2 +-
 src/reader.rs                      |  7 ++++++-
 src/writer.rs                      |  2 +-
 tests/documents/html5.txt          |  2 +-
 tests/documents/sample_4_full.txt  |  1 +
 tests/documents/sample_4_short.txt | 11 ++++++-----
 tests/serde-migrated.rs            |  1 -
 7 files changed, 16 insertions(+), 10 deletions(-)

diff --git a/src/events/mod.rs b/src/events/mod.rs
index eb6516ef..c0b9a924 100644
--- a/src/events/mod.rs
+++ b/src/events/mod.rs
@@ -885,7 +885,7 @@ pub enum Event<'a> {
     Decl(BytesDecl<'a>),
     /// Processing instruction `<?...?>`.
     PI(BytesText<'a>),
-    /// Doctype `<!DOCTYPE...>`.
+    /// Doctype `<!DOCTYPE ...>`.
     DocType(BytesText<'a>),
     /// End of XML document.
     Eof,
diff --git a/src/reader.rs b/src/reader.rs
index 961c9be2..cf050133 100644
--- a/src/reader.rs
+++ b/src/reader.rs
@@ -371,7 +371,12 @@ impl<R: BufRead> Reader<R> {
             Ok(Event::CData(BytesText::from_plain(&buf[8..buf.len() - 2])))
         } else if uncased_starts_with(buf, b"!DOCTYPE") {
             debug_assert!(len >= 8, "Minimum length guaranteed by read_bang_elem");
-            Ok(Event::DocType(BytesText::from_escaped(&buf[8..])))
+            let start = buf[8..]
+                .iter()
+                .position(|b| !is_whitespace(*b))
+                .unwrap_or_else(|| buf.len() - 8);
+            debug_assert!(start < buf.len() - 8, "DocType must have a name");
+            Ok(Event::DocType(BytesText::from_escaped(&buf[8 + start..])))
         } else {
             unreachable!("Proper bang start guaranteed by read_bang_elem");
         }
diff --git a/src/writer.rs b/src/writer.rs
index 04ed699d..73a7dc2e 100644
--- a/src/writer.rs
+++ b/src/writer.rs
@@ -119,7 +119,7 @@ impl<W: Write> Writer<W> {
             }
             Event::Decl(ref e) => self.write_wrapped(b"<?", e, b"?>"),
             Event::PI(ref e) => self.write_wrapped(b"<?", e, b"?>"),
-            Event::DocType(ref e) => self.write_wrapped(b"<!DOCTYPE", e, b">"),
+            Event::DocType(ref e) => self.write_wrapped(b"<!DOCTYPE ", e, b">"),
             Event::Eof => Ok(()),
         };
         if let Some(i) = self.indent.as_mut() {
diff --git a/tests/documents/html5.txt b/tests/documents/html5.txt
index 1ad39c09..f1bc908b 100644
--- a/tests/documents/html5.txt
+++ b/tests/documents/html5.txt
@@ -1,4 +1,4 @@
-DocType( html)
+DocType(html)
 Characters(
 )
 StartElement(a, attr-error: error while parsing attribute at position 7: Attribute value must start with a quote.)
diff --git a/tests/documents/sample_4_full.txt b/tests/documents/sample_4_full.txt
index 4bdadfb7..82ef6ef3 100644
--- a/tests/documents/sample_4_full.txt
+++ b/tests/documents/sample_4_full.txt
@@ -1,4 +1,5 @@
 StartDocument(1.0, utf-8)
+DocType(data SYSTEM "abcd.dtd")
 StartElement({urn:x}p:data [z=">"])
 Whitespace("\n    ")
 Comment(" abcd &lt; &gt; &amp; ")
diff --git a/tests/documents/sample_4_short.txt b/tests/documents/sample_4_short.txt
index 33035902..7262b033 100644
--- a/tests/documents/sample_4_short.txt
+++ b/tests/documents/sample_4_short.txt
@@ -1,15 +1,16 @@
 StartDocument(1.0, utf-8)
-DocType(DOCTYPE data SYSTEM "abcd.dtd")
+DocType(data SYSTEM "abcd.dtd")
 StartElement({urn:x}p:data [z=">"])
+Comment( abcd &lt; &gt; &amp; )
 StartElement(a)
-Characters("test")
+Characters(test)
 EndElement(a)
 StartElement(b)
-Characters("kkss\" = ddd\' >")
+Characters(kkss" = ddd' >)
 EndElement(b)
-Characters("<a>ddddd</b>!e3--><!-- ddckx")
+CData(<a>ddddd</b>!e3--><!-- ddckx)
 StartElement(c)
 EndElement(c)
-Characters("<![CDATA[zzzz]]>")
+CData(<![CDATA[zzzz]]>)
 EndElement({urn:x}p:data)
 EndDocument
diff --git a/tests/serde-migrated.rs b/tests/serde-migrated.rs
index 62b1a384..de5af898 100644
--- a/tests/serde-migrated.rs
+++ b/tests/serde-migrated.rs
@@ -98,7 +98,6 @@ fn test_namespaces() {
 }
 
 #[test]
-#[ignore] // FIXME
 fn test_doctype() {
     #[derive(PartialEq, Serialize, Deserialize, Debug)]
     struct Envelope {