Bug fixses

Author: udan-jayanith

FUTURE-CHANGELOG.md

@@ -1,9 +1,6 @@
 ## v0.0.0-beta.3 <- current
 
 ## v0.0.0
-Delete deletes the node without connecting sibling nodes.
-* Delete
-
 QuerySelector takes attribute name and regexp for the value and returns the first node that matches the regexp.  
 * QuerySelector
 

node-tree.go

@@ -229,6 +229,7 @@ func (node *Node) RemoveNode() {
 
 	node.previousNode = nil
 	node.nextNode = nil
+	node.parentNode = nil
 
 	node.rwMutex.Unlock()
 
@@ -239,9 +240,13 @@ func (node *Node) RemoveNode() {
 		nextNode.SetPreviousNode(previousNode)
 	}
 
-	if nextNode != nil{
+	if nextNode != nil && previousNode == nil{
 		nextNode.setParentNode(parentNode)
 	}
+
+	if parentNode != nil{
+		parentNode.childNode = nextNode
+	}
 }
 
 // IsTextNode returns a boolean value indicating node is a text node or not.

node-tree_test.go

@@ -86,4 +86,27 @@ func TestAppendTextAndInnerText(t *testing.T){
 		t.Fatal(body.GetInnerText(), " != ", h1.GetChildNode().GetText() + p.GetChildNode().GetText())
 	}
 
+}
+
+func TestRemoveNode(t *testing.T){
+	article := GoHtml.CreateNode("article")
+	
+	h1 := GoHtml.CreateNode("h1")
+	h1.AppendText("This is a heading.")
+	article.AppendChild(h1)
+
+	article.AppendChild(GoHtml.CreateNode(GoHtml.Br))
+
+	p := GoHtml.CreateNode("p")
+	p.AppendText("this is a paragraph.")
+	article.AppendChild(p)
+
+	h1.RemoveNode()
+
+	if article.GetChildNode().GetTagName() != GoHtml.Br{
+		t.Fatal("Unexpected tag. ", article.GetChildNode().GetTagName())
+		return
+	}else if p.GetParent() != article {
+		t.Fatal("Unexpected parent.")
+	}
 }