Merge pull request #522 from com-pas/feat/521-add-setstringprivate-in-privateutils

feat(#521): Add setStringPrivate in PrivateUtils

Author: massifben

sct-commons/src/main/java/org/lfenergy/compas/sct/commons/util/PrivateUtils.java

@@ -320,7 +320,6 @@ public static void copyCompasICDHeaderFromLNodePrivateIntoSTDPrivate(TPrivate st
         stdPrivate.getContent().add(objectFactory.createICDHeader(compasICDHeader));
     }
 
-
     public static Optional<String> extractStringPrivate(TBaseElement tBaseElement, String privateType) {
         return tBaseElement.getPrivate().stream()
                 .filter(tPrivate -> privateType.equals(tPrivate.getType()))
@@ -330,4 +329,9 @@ public static Optional<String> extractStringPrivate(TBaseElement tBaseElement, S
                 .filter(StringUtils::isNotBlank)
                 .findFirst();
     }
+
+    public static void setStringPrivate(TBaseElement tBaseElement, String privateType, String privateValue) {
+        tBaseElement.getPrivate().removeIf(tPrivate -> privateType.equals(tPrivate.getType()));
+        tBaseElement.getPrivate().add(createStringPrivate(privateType, privateValue));
+    }
 }

sct-commons/src/test/java/org/lfenergy/compas/sct/commons/util/PrivateUtilsTest.java

@@ -644,4 +644,34 @@ void extractStringPrivate_should_succeed() {
         // Then
         assertThat(result).hasValue("hello World");
     }
+
+    @Test
+    void setStringPrivate_when_private_missing_should_create_new_private() {
+        // Given
+        TIED tied = new TIED();
+        // When
+        PrivateUtils.setStringPrivate(tied, "privateType", "privateValue");
+        // Then
+        assertThat(tied.getPrivate())
+                .singleElement()
+                .extracting(TPrivate::getType, TPrivate::getContent)
+                .containsExactly("privateType", List.of("privateValue"));
+    }
+
+    @Test
+    void setStringPrivate_when_private_is_present_should_update_private() {
+        // Given
+        TIED tied = new TIED();
+        TPrivate tPrivate = new TPrivate();
+        tPrivate.setType("privateType");
+        tPrivate.getContent().add("oldValue");
+        tied.getPrivate().add(tPrivate);
+        // When
+        PrivateUtils.setStringPrivate(tied, "privateType", "newValue");
+        // Then
+        assertThat(tied.getPrivate())
+                .singleElement()
+                .extracting(TPrivate::getType, TPrivate::getContent)
+                .containsExactly("privateType", List.of("newValue"));
+    }
 }