Update _persistClass_Test.java.ejs

mshima · Oct 29, 2023 · e18e340 · e18e340
1 parent 22dde5e
commit e18e340
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...java/templates/src/test/java/_package_/_entityPackage_/domain/_persistClass_Test.java.ejs b/...java/templates/src/test/java/_package_/_entityPackage_/domain/_persistClass_Test.java.ejs
@@ -26,7 +26,7 @@ import static <%= otherEntity.entityAbsolutePackage %>.domain.<%= otherEntity.pe
 import <%= otherEntity.entityAbsolutePackage %>.domain.<%= otherEntity.persistClass %>;
 <%_ } _%>
 <%_ if (relationships.some(relationship => relationship.collection)) { _%>
-import java.util.Collections;
+import java.util.Set;
 <%_ } _%>
 import org.junit.jupiter.api.Test;
 import static org.assertj.core.api.Assertions.assertThat;
@@ -92,7 +92,7 @@ class <%= persistClass %>Test {
   <%_ } _%>
   <%_ if (relationship.collection) { _%>
 
-        <%= persistInstance %>.set<%- relationship.propertyNameCapitalized %>(Collections.singleton(<%= relationship.otherEntity.persistInstance %>Back));
+        <%= persistInstance %>.set<%- relationship.propertyNameCapitalized %>(Set.of(<%= relationship.otherEntity.persistInstance %>Back));
         assertThat(<%= persistInstance %>.get<%- relationship.propertyNameCapitalized %>()).containsOnly(<%= relationship.otherEntity.persistInstance %>Back);
     <%_ if (relationship.otherRelationship && !relationship.ownerSide && !relationship.otherEntity.embedded) { _%>
       <%_ if (relationship.otherRelationship.collection) { _%>