Adding Delete() func to properties Change-Id: Ifea9d67b6467f0373736d291a41d7d4685e697d7
diff --git a/properties.go b/properties.go index 6201f8b..e7cac17 100644 --- a/properties.go +++ b/properties.go
@@ -596,6 +596,15 @@ // ---------------------------------------------------------------------------- +// deletes the key from the map and its comments +// abides by the rules of the builtin delete() +func (p *Properties) Delete(key string) () { + delete(p.m, key) + delete(p.c, key) +} + +// ---------------------------------------------------------------------------- + // check expands all values and returns an error if a circular reference or // a malformed expression was found. func (p *Properties) check() error {
diff --git a/properties_test.go b/properties_test.go index 6ae0b42..2615f32 100644 --- a/properties_test.go +++ b/properties_test.go
@@ -797,6 +797,28 @@ c.Assert(func() { p.MustGetUint("max") }, PanicMatches, ".* out of range") } +func (s *TestSuite) TestDeleteKey(c *C) { + input := "#comments should also be gone\nkey=to-be-deleted" + p, err := parse(input) + c.Assert(err, IsNil) + c.Check(len(p.m), Equals, 1) + c.Check(len(p.c), Equals, 1) + p.Delete("key") + c.Check(len(p.m), Equals, 0) + c.Check(len(p.c), Equals, 0) +} + +func (s *TestSuite) TestDeleteUnknownKey(c *C) { + input := "#comments should also be gone\nkey=to-be-deleted" + p, err := parse(input) + c.Assert(err, IsNil) + c.Check(len(p.m), Equals, 1) + c.Check(len(p.c), Equals, 1) + p.Delete("wrong-key") + c.Check(len(p.m), Equals, 1) + c.Check(len(p.c), Equals, 1) +} + // ---------------------------------------------------------------------------- // tests all combinations of delimiters, leading and/or trailing whitespace and newlines.