Add customizeProperty

fixture-monkey-api/src/main/java/com/navercorp/fixturemonkey/api/experimental/TypedExpressionGenerator.java

@@ -0,0 +1,36 @@
+/*
+ * Fixture Monkey
+ *
+ * Copyright (c) 2021-present NAVER Corp.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.navercorp.fixturemonkey.api.experimental;
+
+import org.apiguardian.api.API;
+import org.apiguardian.api.API.Status;
+
+import com.navercorp.fixturemonkey.api.expression.ExpressionGenerator;
+
+@API(since = "1.0.6", status = Status.EXPERIMENTAL)
+@FunctionalInterface
+public interface TypedExpressionGenerator<T> extends TypedPropertySelector<T>, ExpressionGenerator {
+	static <U> TypedExpressionGenerator<U> typedRoot() {
+		return propertyNameResolver -> "$";
+	}
+
+	static <U> TypedExpressionGenerator<U> typedString(String expression) {
+		return propertyNameResolver -> expression;
+	}
+}

fixture-monkey-api/src/main/java/com/navercorp/fixturemonkey/api/experimental/TypedPropertySelector.java

@@ -0,0 +1,31 @@
+/*
+ * Fixture Monkey
+ *
+ * Copyright (c) 2021-present NAVER Corp.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.navercorp.fixturemonkey.api.experimental;
+
+import org.apiguardian.api.API;
+import org.apiguardian.api.API.Status;
+
+import com.navercorp.fixturemonkey.api.property.PropertySelector;
+
+/**
+ * It is a marker interface for selecting a specific property with type.
+ */
+@API(since = "1.0.6", status = Status.EXPERIMENTAL)
+public interface TypedPropertySelector<T> extends PropertySelector {
+}

fixture-monkey-tests/kotlin-tests/src/test/kotlin/com/navercorp/fixturemonkey/tests/kotlin/KotlinTest.kt

@@ -20,6 +20,7 @@ package com.navercorp.fixturemonkey.tests.kotlin
 
 import com.navercorp.fixturemonkey.FixtureMonkey
 import com.navercorp.fixturemonkey.api.experimental.JavaGetterMethodPropertySelector.javaGetter
+import com.navercorp.fixturemonkey.api.experimental.TypedExpressionGenerator.typedRoot
 import com.navercorp.fixturemonkey.api.introspector.ConstructorPropertiesArbitraryIntrospector
 import com.navercorp.fixturemonkey.api.introspector.FactoryMethodArbitraryIntrospector
 import com.navercorp.fixturemonkey.api.introspector.FailoverIntrospector
@@ -36,12 +37,16 @@ import com.navercorp.fixturemonkey.kotlin.setExp
 import com.navercorp.fixturemonkey.kotlin.setExpGetter
 import com.navercorp.fixturemonkey.kotlin.sizeExp
 import com.navercorp.fixturemonkey.kotlin.sizeExpGetter
+import com.navercorp.fixturemonkey.tests.TestEnvironment.TEST_COUNT
 import com.navercorp.fixturemonkey.tests.kotlin.ImmutableJavaTestSpecs.ArrayObject
 import com.navercorp.fixturemonkey.tests.kotlin.ImmutableJavaTestSpecs.NestedArrayObject
 import com.navercorp.fixturemonkey.tests.kotlin.JavaConstructorTestSpecs.JavaTypeObject
 import org.assertj.core.api.BDDAssertions.then
 import org.assertj.core.api.BDDAssertions.thenThrownBy
+import org.junit.jupiter.api.RepeatedTest
 import org.junit.jupiter.api.Test
+import java.time.Instant
+import java.time.temporal.ChronoUnit
 import java.util.UUID
 import kotlin.reflect.jvm.javaMethod
 
@@ -199,6 +204,33 @@ class KotlinTest {
         then(actual).isNotNull
     }
 
+    @RepeatedTest(TEST_COUNT)
+    fun customizePropertySet() {
+        val actual = SUT.giveMeExperimentalBuilder<String>()
+            .customizeProperty(typedRoot<String>()) {
+                it.filter { str -> str.length > 5 }
+                    .map { str -> str.substring(0..3) }
+            }
+            .sample()
+
+        then(actual).hasSizeLessThanOrEqualTo(4)
+    }
+
+    @RepeatedTest(TEST_COUNT)
+    fun customizePropertyFilter() {
+        val now = Instant.now()
+        val min = now.minus(365, ChronoUnit.DAYS)
+        val max = now.plus(365, ChronoUnit.DAYS)
+
+        val actual = SUT.giveMeExperimentalBuilder<Instant>()
+            .customizeProperty(typedRoot<Instant>()) {
+                it.filter { instant -> instant.isAfter(min) && instant.isBefore(max) }
+            }
+            .sample()
+
+        then(actual).isBetween(min, max)
+    }
+
     companion object {
         private val SUT: FixtureMonkey = FixtureMonkey.builder()
             .plugin(KotlinPlugin())

fixture-monkey/src/main/java/com/navercorp/fixturemonkey/customizer/MonkeyManipulatorFactory.java

@@ -24,6 +24,7 @@
 import java.util.Map;
 import java.util.Map.Entry;
 import java.util.concurrent.atomic.AtomicInteger;
+import java.util.function.Function;
 import java.util.function.Predicate;
 import java.util.function.Supplier;
 import java.util.stream.Collectors;
@@ -36,6 +37,7 @@
 import net.jqwik.api.Arbitrary;
 
 import com.navercorp.fixturemonkey.ArbitraryBuilder;
+import com.navercorp.fixturemonkey.api.arbitrary.CombinableArbitrary;
 import com.navercorp.fixturemonkey.api.container.DecomposedContainerValueFactory;
 import com.navercorp.fixturemonkey.api.generator.ArbitraryContainerInfo;
 import com.navercorp.fixturemonkey.api.lazy.LazyArbitrary;
@@ -107,6 +109,16 @@ public <T> ArbitraryManipulator newArbitraryManipulator(
 		);
 	}
 
+	public <T> ArbitraryManipulator newArbitraryManipulator(
+		String expression,
+		Function<CombinableArbitrary<? extends T>, CombinableArbitrary<? extends T>> arbitraryCustomizer
+	) {
+		return new ArbitraryManipulator(
+			monkeyExpressionFactory.from(expression).toNodeResolver(),
+			new NodeCustomizerManipulator<>(arbitraryCustomizer)
+		);
+	}
+
 	public ContainerInfoManipulator newContainerInfoManipulator(
 		String expression,
 		int min,

fixture-monkey/src/main/java/com/navercorp/fixturemonkey/customizer/NodeCustomizerManipulator.java

@@ -0,0 +1,44 @@
+/*
+ * Fixture Monkey
+ *
+ * Copyright (c) 2021-present NAVER Corp.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.navercorp.fixturemonkey.customizer;
+
+import java.util.function.Function;
+
+import org.apiguardian.api.API;
+import org.apiguardian.api.API.Status;
+
+import com.navercorp.fixturemonkey.api.arbitrary.CombinableArbitrary;
+import com.navercorp.fixturemonkey.tree.ObjectNode;
+
+@API(since = "1.0.6", status = Status.EXPERIMENTAL)
+public final class NodeCustomizerManipulator<T> implements NodeManipulator {
+	private final Function<CombinableArbitrary<? extends T>, CombinableArbitrary<? extends T>> arbitraryCustomizer;
+
+	public NodeCustomizerManipulator(
+		Function<CombinableArbitrary<? extends T>, CombinableArbitrary<? extends T>> arbitraryCustomizer
+	) {
+		this.arbitraryCustomizer = arbitraryCustomizer;
+	}
+
+	@SuppressWarnings({"rawtypes", "unchecked"})
+	@Override
+	public void manipulate(ObjectNode objectNode) {
+		objectNode.addArbitraryCustomizer((Function)arbitraryCustomizer);
+	}
+}

fixture-monkey/src/main/java/com/navercorp/fixturemonkey/experimental/ExperimentalArbitraryBuilder.java

@@ -18,11 +18,20 @@
 
 package com.navercorp.fixturemonkey.experimental;
 
+import java.util.function.Function;
+
 import org.apiguardian.api.API;
 import org.apiguardian.api.API.Status;
 
 import com.navercorp.fixturemonkey.ArbitraryBuilder;
+import com.navercorp.fixturemonkey.api.arbitrary.CombinableArbitrary;
+import com.navercorp.fixturemonkey.api.experimental.TypedPropertySelector;
 
 @API(since = "0.6.12", status = Status.MAINTAINED)
 public interface ExperimentalArbitraryBuilder<T> extends ArbitraryBuilder<T> {
+	@API(since = "1.0.6", status = Status.EXPERIMENTAL)
+	<U> ArbitraryBuilder<T> customizeProperty(
+		TypedPropertySelector<U> propertySelector,
+		Function<CombinableArbitrary<? extends U>, CombinableArbitrary<? extends U>> combinableArbitraryCustomizer
+	);
 }

fixture-monkey/src/main/java/com/navercorp/fixturemonkey/resolver/DefaultArbitraryBuilder.java

@@ -51,6 +51,7 @@
 import com.navercorp.fixturemonkey.ArbitraryBuilder;
 import com.navercorp.fixturemonkey.api.arbitrary.CombinableArbitrary;
 import com.navercorp.fixturemonkey.api.context.MonkeyContext;
+import com.navercorp.fixturemonkey.api.experimental.TypedPropertySelector;
 import com.navercorp.fixturemonkey.api.expression.ExpressionGenerator;
 import com.navercorp.fixturemonkey.api.instantiator.Instantiator;
 import com.navercorp.fixturemonkey.api.instantiator.InstantiatorProcessResult;
@@ -462,6 +463,18 @@ public ExperimentalArbitraryBuilder<T> instantiate(
 		return this;
 	}
 
+	@Override
+	public <U> ArbitraryBuilder<T> customizeProperty(
+		TypedPropertySelector<U> propertySelector,
+		Function<CombinableArbitrary<? extends U>, CombinableArbitrary<? extends U>> combinableArbitraryCustomizer
+	) {
+		String expression = resolveExpression(toExpressionGenerator(propertySelector));
+		context.addManipulator(
+			monkeyManipulatorFactory.newArbitraryManipulator(expression, combinableArbitraryCustomizer)
+		);
+		return this;
+	}
+
 	@SuppressWarnings("unchecked")
 	@Override
 	public Arbitrary<T> build() {

fixture-monkey/src/main/java/com/navercorp/fixturemonkey/tree/ObjectNode.java

@@ -20,6 +20,7 @@
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.function.Function;
 import java.util.function.Predicate;
 
 import javax.annotation.Nullable;
@@ -57,6 +58,8 @@ public final class ObjectNode {
 
 	@SuppressWarnings("rawtypes")
 	private final List<Predicate> arbitraryFilters = new ArrayList<>();
+	private final List<Function<CombinableArbitrary<?>, CombinableArbitrary<?>>> arbitraryCustomizers =
+		new ArrayList<>();
 
 	private final LazyArbitrary<Boolean> childNotCacheable = LazyArbitrary.lazy(() -> {
 		for (ObjectNode child : children) {
@@ -143,6 +146,14 @@ public List<Predicate> getArbitraryFilters() {
 		return arbitraryFilters;
 	}
 
+	public void addArbitraryCustomizer(Function<CombinableArbitrary<?>, CombinableArbitrary<?>> arbitraryCustomizer) {
+		this.arbitraryCustomizers.add(arbitraryCustomizer);
+	}
+
+	public List<Function<CombinableArbitrary<?>, CombinableArbitrary<?>>> getArbitraryCustomizers() {
+		return arbitraryCustomizers;
+	}
+
 	public boolean manipulated() {
 		return !manipulators.isEmpty() || !containerInfoManipulators.isEmpty();
 	}

fixture-monkey/src/main/java/com/navercorp/fixturemonkey/tree/ObjectTree.java

@@ -23,6 +23,7 @@
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.function.Function;
 import java.util.function.Predicate;
 import java.util.stream.Collectors;
 
@@ -159,6 +160,13 @@ private CombinableArbitrary<?> generateIntrospected(
 			}
 		}
 
+		List<Function<CombinableArbitrary<?>, CombinableArbitrary<?>>> arbitraryCustomizers =
+			node.getArbitraryCustomizers();
+
+		for (Function<CombinableArbitrary<?>, CombinableArbitrary<?>> arbitraryCustomizer : arbitraryCustomizers) {
+			generated = arbitraryCustomizer.apply(generated);
+		}
+
 		List<Predicate> arbitraryFilters = node.getArbitraryFilters();
 		for (Predicate predicate : arbitraryFilters) {
 			generated = generated.filter(fixtureMonkeyOptions.getGenerateMaxTries(), predicate);