new `setOrUpdateField` operation to udpate `optionalFields`

1 files changed, 7 insertions, 0 deletions

diff --git a/src/test/scala/cc/spray/json/lenses/JsonLensesSpec.scala b/src/test/scala/cc/spray/json/lenses/JsonLensesSpec.scala
index a6a7697..3f7b0ff 100644
--- a/src/test/scala/cc/spray/json/lenses/JsonLensesSpec.scala
+++ b/src/test/scala/cc/spray/json/lenses/JsonLensesSpec.scala
@@ -124,6 +124,13 @@ class JsonLensesSpec extends Specification with SpecHelpers {
         "create" in {
           """[{"b": 4}, {"c": 5}]""".update((* / 'b.?) ! set(38)) must be_json("""[{"b": 38}, {"c": 5, "b": 38}]""")
         }
+        "set or update with default" in {
+          """[{"b": 4}, {"c": 5}]""".update((* / 'b.?) ! setOrUpdateField(38)(1 + )) must be_json("""[{"b": 5}, {"c": 5, "b": 38}]""")
+        }
+        "set or update with function" in {
+          """[{"b": 4}, {"c": 5}]""".update((* / 'b.?) ! setOrUpdateField[Int]((_: Option[Int]).map(1+).getOrElse(38))) must be_json("""[{"b": 5}, {"c": 5, "b": 38}]""")
+        }
+
         "create nested (current behavior)" in {
           // One could think that nested `optionalField`s and `set` would create intermediate
           // objects as well. However, this is currently not possible, since