Small fix

Imangazaliev · Oct 14, 2016 · b766098 · b766098
1 parent e57fe9b
commit b766098
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/src/DiDom/Element.php b/src/DiDom/Element.php
@@ -262,6 +262,8 @@ public function innerHtml($options = LIBXML_NOEMPTYTAG)
      * Sets inner HTML.
      * 
      * @param string $html
+     * 
+     * @return Element
      */
     public function setInnerHtml($html)
     {
@@ -292,6 +294,8 @@ public function setInnerHtml($html)
 
             Errors::restore();
         }
+
+        return $this;
     }
 
     /**

diff --git a/tests/DiDom/ElementTest.php b/tests/DiDom/ElementTest.php
@@ -273,6 +273,21 @@ public function testInnerHtml()
         $this->assertTrue(is_string($document->find('body')[0]->innerHtml()));
     }
 
+    public function testSetInnerHtml()
+    {
+        $list = new Element('ul');
+
+        $html = '<li>One</li><li>Two</li><li>Three</li>';
+
+        $this->assertEquals($list, $list->setInnerHtml($html));
+        $this->assertEquals(['One', 'Two', 'Three'], $list->find('li::text'));
+
+        $span = new Element('span');
+
+        $this->assertEquals($span, $span->setInnerHtml('  Foo  '));
+        $this->assertEquals('<span>  Foo  </span>', $span->html());
+    }
+
     public function testHtmlWithOptions()
     {
         $html = '<html><body><span></span></body></html>';