add union update and delete

wangjild · wangjild · commit 5d653b2d0153 · 2016-12-21T10:43:55.000+08:00
diff --git a/jsonpath_rw/jsonpath.py b/jsonpath_rw/jsonpath.py
@@ -425,6 +425,16 @@ def is_singular(self):
     def find(self, data):
         return self.left.find(data) + self.right.find(data)
 
+    def update(self, data, val):
+        self.left.update(data, val)
+        self.right.update(data, val)
+        return data
+
+    def delete(self, data):
+        self.left.delete(data)
+        self.right.delete(data)
+        return data
+
 class Intersect(JSONPath):
     """
     JSONPath for bits that match *both* patterns.
diff --git a/tests/test_jsonpath.py b/tests/test_jsonpath.py
@@ -162,6 +162,11 @@ def test_descendants_value(self):
             ('foo..baz', {'foo': [{'baz': 1}, {'baz': 2}]}, [1, 2] ), 
         ])
 
+    def test_union_value(self):
+        self.check_cases([
+            ('foo | bar', {'foo': 1, 'bar': 2}, [1, 2])
+        ])
+
     def test_parent_value(self):
         self.check_cases([('foo.baz.`parent`', {'foo': {'baz': 3}}, [{'baz': 3}]),
                           ('foo.`parent`.foo.baz.`parent`.baz.bizzle', {'foo': {'baz': {'bizzle': 5}}}, [5])])
@@ -328,6 +333,11 @@ def test_update_child(self):
             ({'foo': {'bar': 1}}, 'foo.bar', 'baz', {'foo': {'bar': 'baz'}})
         ])
 
+    def test_update_union(self):
+        self.check_update_cases([
+            ({'foo': 1, 'bar': 2}, 'foo | bar', 3, {'foo': 3, 'bar': 3})
+        ])
+
     def test_update_where(self):
         self.check_update_cases([
             ({'foo': {'bar': {'baz': 1}}, 'bar': {'baz': 2}},
@@ -426,6 +436,12 @@ def test_delete_descendants_where(self):
              {'foo': {'flag': 1}, 'baz': {'bar': 2}})
         ])
 
+    def test_delete_union(self):
+        self.check_delete_cases([
+            ({'foo': 1, 'bar': 2}, 'foo | bar', {}),
+            ({'foo': 1, 'bar': 2, 'baz': 3}, 'foo | bar', {'baz': 3}),
+        ])
+
     def test_delete_index(self):
         self.check_delete_cases([
             ([42], '[0]', []),
