Add function to remove a child from tree

Author: Galaco

keyvalue.go

@@ -116,6 +116,23 @@ func (node *KeyValue) AddChild(value *KeyValue) error {
 	return nil
 }
 
+func (node *KeyValue) RemoveChild(key string) error {
+	if !node.HasChildren() {
+		return errors.New("parent does not accept child keys")
+	}
+	ret,err := node.Find(key)
+	if err != nil {
+		return errors.New("key does not exist")
+	}
+	for idx,c := range node.value {
+		if c == ret {
+			node.value = append(node.value[:idx], node.value[idx+1:]...)
+			return nil
+		}
+	}
+	return nil
+}
+
 // Parent returns this node's parent.
 // Parent can be nil
 func (node *KeyValue) Parent() *KeyValue {

keyvalue_test.go

@@ -242,3 +242,73 @@ func TestKeyValue_MergeInto(t *testing.T) {
 		t.Error(err)
 	}
 }
+
+
+func TestKeyValue_RemoveChild(t *testing.T) {
+	a := &KeyValue{
+		key: "foo",
+		valueType: ValueArray,
+		value: []interface{}{
+			&KeyValue{
+				key: "bar",
+				valueType: ValueString,
+				value: []interface{}{
+					"bar",
+				},
+			},
+			&KeyValue{
+				key: "baz",
+				valueType: ValueString,
+				value: []interface{}{
+					"baz",
+				},
+			},
+			&KeyValue{
+				key: "bat",
+				valueType: ValueString,
+				value: []interface{}{
+					"bat",
+				},
+			},
+		},
+	}
+	b := &KeyValue{
+		key: "foo",
+		valueType: ValueArray,
+		value: []interface{}{
+			&KeyValue{
+				key: "bar",
+				valueType: ValueString,
+				value: []interface{}{
+					"cart",
+				},
+			},
+			&KeyValue{
+				key: "egg",
+				valueType: ValueString,
+				value: []interface{}{
+					"bat",
+				},
+			},
+		},
+	}
+
+	result,err := a.MergeInto(b)
+	if err != nil {
+		t.Error(err)
+	}
+
+	actual,err := result.Find("bar")
+	if actual == nil {
+		t.Error(err)
+	}
+	err = result.RemoveChild("bar")
+	if err != nil {
+		t.Error(err)
+	}
+	actual,err = result.Find("bar")
+	if actual != nil {
+		t.Error("found key that should have been removed")
+	}
+
+}