Merge pull request #61 from goetas/issue_50

Handling empty documents
Masterminds · Oct 1, 2014 · f50e879 · f50e879
2 parents 496b803 + 75fe1de
commit f50e879
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/src/HTML5/Serializer/OutputRules.php b/src/HTML5/Serializer/OutputRules.php
@@ -115,8 +115,10 @@ public function setTraverser(\Masterminds\HTML5\Serializer\Traverser $traverser)
     public function document($dom)
     {
         $this->doctype();
-        $this->traverser->node($dom->documentElement);
-        $this->nl();
+        if ($dom->documentElement) {
+            $this->traverser->node($dom->documentElement);
+            $this->nl();
+        }
     }
 
     protected function doctype()

diff --git a/test/HTML5/Serializer/OutputRulesTest.php b/test/HTML5/Serializer/OutputRulesTest.php
@@ -76,6 +76,19 @@ public function testDocument()
         $this->assertEquals($expected, stream_get_contents($stream, - 1, 0));
     }
 
+    public function testEmptyDocument()
+    {
+    	$dom = $this->html5->loadHTML('');
+
+    	$stream = fopen('php://temp', 'w');
+    	$r = new OutputRules($stream, $this->html5->getOptions());
+    	$t = new Traverser($dom, $stream, $r, $this->html5->getOptions());
+
+    	$r->document($dom);
+    	$expected = '<!DOCTYPE html>' . PHP_EOL;
+    	$this->assertEquals($expected, stream_get_contents($stream, - 1, 0));
+    }
+
     public function testDoctype()
     {
         $dom = $this->html5->loadHTML('<!doctype html><html lang="en"><body>foo</body></html>');