add TestData base classes

Author: SirCotare

src/main/java/it/aboutbits/springboot/testing/testdata/base/ModifyableTestDataCreator.java

@@ -0,0 +1,95 @@
+package it.aboutbits.springboot.testing.testdata.base;
+
+import it.aboutbits.springboot.testing.spring.BeanAccessor;
+import jakarta.persistence.EntityManager;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.transaction.support.TransactionTemplate;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.function.BiFunction;
+import java.util.function.Consumer;
+import java.util.function.ObjIntConsumer;
+import java.util.function.UnaryOperator;
+
+@SuppressWarnings("unchecked")
+@Slf4j
+public abstract class ModifyableTestDataCreator<CREATOR extends ModifyableTestDataCreator<CREATOR, ITEM, PARAMETER>, ITEM, PARAMETER> extends TestDataCreator<ITEM> {
+    private boolean mutatorSet = false;
+    private boolean mutatorCalled = false;
+
+    protected BiFunction<PARAMETER, Integer, PARAMETER> parameterMutator = (parameter, index) -> {
+        mutatorCalled = true;
+        return parameter;
+    };
+    protected ObjIntConsumer<ITEM> resultMutator = (item, index) -> {
+    };
+
+    protected ModifyableTestDataCreator(int count) {
+        super(count);
+    }
+
+    public CREATOR modifyParameter(BiFunction<PARAMETER, Integer, PARAMETER> parameterMutator) {
+        this.parameterMutator = (parameter, index) -> {
+            mutatorCalled = true;
+            return parameterMutator.apply(parameter, index);
+        };
+        mutatorSet = true;
+        return (CREATOR) this;
+    }
+
+    public CREATOR modifyParameter(UnaryOperator<PARAMETER> parameterMutator) {
+        this.parameterMutator = (parameter, index) -> {
+            mutatorCalled = true;
+            return parameterMutator.apply(parameter);
+        };
+        mutatorSet = true;
+        return (CREATOR) this;
+    }
+
+    public CREATOR modifyResult(ObjIntConsumer<ITEM> resultMutator) {
+        this.resultMutator = resultMutator;
+        return (CREATOR) this;
+    }
+
+    public CREATOR modifyResult(Consumer<ITEM> resultMutator) {
+        this.resultMutator = (item, index) -> resultMutator.accept(item);
+        return (CREATOR) this;
+    }
+
+    @Override
+    protected List<ITEM> create() {
+        var result = new ArrayList<ITEM>();
+
+        for (var index = 0; index < numberOfItems; index++) {
+            var item = create(index);
+
+            resultMutator.accept(item, index);
+
+            result.add(
+                    saveMutation(item)
+            );
+        }
+
+        if (mutatorSet && !mutatorCalled) {
+            log.error("Parameter-mutation is defined but was never called.");
+            throw new IllegalStateException("Parameter-mutation is defined but was never called.");
+        }
+
+        return result;
+    }
+
+    protected ITEM saveMutation(ITEM item) {
+        var entityManager = BeanAccessor.getBean(EntityManager.class);
+        var transactionTemplate = BeanAccessor.getBean(TransactionTemplate.class);
+
+        return transactionTemplate.execute(status -> {
+            try {
+                return entityManager.merge(item);
+            } catch (Exception e) {
+                status.setRollbackOnly();
+                throw e;
+            }
+        });
+    }
+}

src/main/java/it/aboutbits/springboot/testing/testdata/base/TestDataCreator.java

@@ -0,0 +1,57 @@
+package it.aboutbits.springboot.testing.testdata.base;
+
+import java.util.ArrayList;
+import java.util.Comparator;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+import java.util.function.Function;
+
+public abstract class TestDataCreator<ITEM> {
+    protected final int numberOfItems;
+
+    protected TestDataCreator(int numberIfItems) {
+        this.numberOfItems = numberIfItems;
+    }
+
+    public void commit() {
+        create();
+    }
+
+    public ITEM returnFirst() {
+        return create().getFirst();
+    }
+
+    public List<ITEM> returnAll() {
+        return create();
+    }
+
+    public List<ITEM> returnSorted(Comparator<ITEM> comparator) {
+        return returnAll().stream().sorted(comparator).toList();
+    }
+
+    public <U extends Comparable<? super U>> List<ITEM> returnSorted(Function<ITEM, U> comparator) {
+        return returnAll().stream()
+                .sorted(Comparator.comparing(comparator))
+                .toList();
+    }
+
+    public Set<ITEM> returnSet() {
+        return new HashSet<>(create());
+    }
+
+    protected List<ITEM> create() {
+        var result = new ArrayList<ITEM>();
+
+        for (var index = 0; index < numberOfItems; index++) {
+            result.add(
+                    create(index)
+            );
+        }
+
+        return result;
+    }
+
+    protected abstract ITEM create(int index);
+
+}