Add: removal of keys from leaves

2 files changed, 10 insertions, 0 deletions

diff --git a/obnamlib/fmt_ga/leaf.py b/obnamlib/fmt_ga/leaf.py
index 6522a6a9..3e5c0b2b 100644
--- a/obnamlib/fmt_ga/leaf.py
+++ b/obnamlib/fmt_ga/leaf.py
@@ -36,6 +36,10 @@ class CowLeaf(object):
     def insert(self, key, value):
         self._dict[key] = value
 
+    def remove(self, key):
+        if key in self._dict:
+            del self._dict[key]
+
     def as_dict(self):
         return copy.deepcopy(self._dict)
 
diff --git a/obnamlib/fmt_ga/leaf_tests.py b/obnamlib/fmt_ga/leaf_tests.py
index 6c5f2903..bb035c9c 100644
--- a/obnamlib/fmt_ga/leaf_tests.py
+++ b/obnamlib/fmt_ga/leaf_tests.py
@@ -52,6 +52,12 @@ class CowLeafTests(unittest.TestCase):
         leaf.insert('foo', 'bar')
         self.assertEqual(leaf.keys(), ['foo'])
 
+    def test_removes_key(self):
+        leaf = obnamlib.CowLeaf()
+        leaf.insert('foo', 'bar')
+        leaf.remove('foo')
+        self.assertEqual(leaf.keys(), [])
+
     def test_dict_round_trip(self):
         leaf = obnamlib.CowLeaf()
         leaf.insert('foo', 'bar')