introduce YAMLMapping.withReplaced()

nresare · Aug 23, 2022 · af9dc9c · af9dc9c
1 parent bb1713f
commit af9dc9c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/main/java/com/resare/nryaml/YAMLMapping.java b/src/main/java/com/resare/nryaml/YAMLMapping.java
@@ -83,4 +83,10 @@ public YAMLMapping combine(YAMLMapping other) {
         result.putAll(other);
         return new YAMLMapping(result);
     }
+
+    public YAMLMapping withReplaced(String key, YAMLValue value) {
+        var result = new HashMap<>(this);
+        result.put(key, value);
+        return new YAMLMapping(result);
+    }
 }
diff --git a/src/test/java/com/resare/nryaml/YAMLMappingTest.java b/src/test/java/com/resare/nryaml/YAMLMappingTest.java
@@ -59,4 +59,12 @@ void testCombine() {
                 mapping.toBareObject()
         );
     }
+
+    @Test
+    void testWithReplaced() {
+        var mapping = YAMLMapping.of("key", YAMLValue.of("first"));
+        var another = mapping.withReplaced("key", YAMLValue.of("second"));
+        assertEquals(YAMLMapping.of("key", YAMLValue.of("first")), mapping);
+        assertEquals(YAMLMapping.of("key", YAMLValue.of("second")), another);
+    }
 }