add helper to generate entity ids

aboutbits · Jan 22, 2025 · 52cde3d · 52cde3d
1 parent f165ea4
commit 52cde3d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/main/java/it/aboutbits/springboot/testing/testdata/base/FakerExtended.java b/src/main/java/it/aboutbits/springboot/testing/testdata/base/FakerExtended.java
@@ -1,12 +1,14 @@
 package it.aboutbits.springboot.testing.testdata.base;
 
+import it.aboutbits.springboot.toolbox.type.identity.EntityId;
 import net.datafaker.Faker;
 import net.datafaker.service.FakeValuesService;
 import net.datafaker.service.FakerContext;
 import net.datafaker.service.RandomService;
 
 import java.util.Locale;
 import java.util.Random;
+import java.util.function.LongFunction;
 
 public class FakerExtended extends Faker {
     public FakerExtended() {
@@ -40,4 +42,10 @@ public <T extends Enum<?>> T randomEnumValue(Class<T> enumClass) {
         }
         return values[this.random().nextInt(values.length)];
     }
+
+    public <T extends EntityId<Long>> T randomEntityId(LongFunction<T> constructor) {
+        return constructor.apply(
+                super.random().nextInt(99999)
+        );
+    }
 }