More tests

Imangazaliev · Oct 16, 2016 · 2157726 · 2157726
1 parent 6d125ba
commit 2157726
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/DiDom/Document.php b/src/DiDom/Document.php
@@ -489,7 +489,11 @@ public function getElement()
      */
     public function toElement()
     {
-        return new Element($this->getElement());
+        if ($this->document->documentElement === null) {
+            throw new RuntimeException('Cannot convert empty document to Element');
+        }
+
+        return new Element($this->document->documentElement);
     }
 
     /**

diff --git a/tests/DiDom/DocumentTest.php b/tests/DiDom/DocumentTest.php
@@ -477,6 +477,16 @@ public function testGetElement()
         $this->assertInstanceOf('DOMElement', $document->getElement());
     }
 
+    /**
+     * @expectedException RuntimeException
+     */
+    public function testEmptyDocumentToElement()
+    {
+        $document = new Document();
+
+        $document->toElement();
+    }
+
     public function testToElement()
     {
         $html = $this->loadFixture('posts.html');