Unit tests for DOMTreeBuilder begun.

Minor bugs fixed, too.

1 files changed, 9 insertions, 2 deletions

diff --git a/src/HTML5/Parser/DOMTreeBuilder.php b/src/HTML5/Parser/DOMTreeBuilder.php
index be9fa23..2807790 100644
--- a/src/HTML5/Parser/DOMTreeBuilder.php
+++ b/src/HTML5/Parser/DOMTreeBuilder.php
@@ -30,7 +30,14 @@ class DOMTreeBuilder implements EventHandler {
     $this->doc = \DOMImplementation::createDocument(NULL, 'html', $dt);
     $this->doc->errors = array();
 
-    $this->current = $this->doc->documentElement();
+    $this->current = $this->doc->documentElement;
+  }
+
+  /**
+   * Get the document.
+   */
+  public function document() {
+    return $this->doc;
   }
 
   /**
@@ -71,7 +78,7 @@ class DOMTreeBuilder implements EventHandler {
 
   public function endTag($name) {
     $lname = $this->normalizeTagName($name);
-    if ($this->current->tagName() != $lname) {
+    if ($this->current->tagName != $lname) {
       return $this->quirksTreeResolver($lname);
     }