Add utils to ease configuring ApplicationContext

Author: IlyaMuravjov

utbot-framework/src/main/kotlin/org/utbot/framework/context/utils/ApplicationContextUtils.kt

@@ -0,0 +1,22 @@
+package org.utbot.framework.context.utils
+
+import org.utbot.framework.context.ApplicationContext
+import org.utbot.framework.context.ConcreteExecutionContext
+import org.utbot.fuzzing.JavaValueProvider
+
+fun ApplicationContext.transformConcreteExecutionContext(
+    transformer: (ConcreteExecutionContext) -> ConcreteExecutionContext
+) = object : ApplicationContext by this {
+    override fun createConcreteExecutionContext(
+        fullClasspath: String,
+        classpathWithoutDependencies: String
+    ): ConcreteExecutionContext = transformer(
+        this@transformConcreteExecutionContext.createConcreteExecutionContext(
+            fullClasspath, classpathWithoutDependencies
+        )
+    )
+}
+
+fun ApplicationContext.transformValueProvider(
+    transformer: (JavaValueProvider) -> JavaValueProvider
+) = transformConcreteExecutionContext { it.transformValueProvider(transformer) }

utbot-framework/src/main/kotlin/org/utbot/framework/context/utils/ConcreteExecutionContextUtils.kt

@@ -0,0 +1,26 @@
+package org.utbot.framework.context.utils
+
+import org.utbot.framework.context.ConcreteExecutionContext
+import org.utbot.framework.context.JavaFuzzingContext
+import org.utbot.framework.plugin.api.ClassId
+import org.utbot.fuzzer.IdentityPreservingIdGenerator
+import org.utbot.fuzzing.JavaValueProvider
+import org.utbot.instrumentation.ConcreteExecutor
+import org.utbot.instrumentation.instrumentation.execution.UtConcreteExecutionResult
+import org.utbot.instrumentation.instrumentation.execution.UtExecutionInstrumentation
+
+fun ConcreteExecutionContext.transformJavaFuzzingContext(
+    transformer: (JavaFuzzingContext) -> JavaFuzzingContext
+) = object : ConcreteExecutionContext by this {
+    override fun tryCreateFuzzingContext(
+        concreteExecutor: ConcreteExecutor<UtConcreteExecutionResult, UtExecutionInstrumentation>,
+        classUnderTest: ClassId,
+        idGenerator: IdentityPreservingIdGenerator<Int>
+    ): JavaFuzzingContext = transformer(
+        this@transformJavaFuzzingContext.tryCreateFuzzingContext(concreteExecutor, classUnderTest, idGenerator)
+    )
+}
+
+fun ConcreteExecutionContext.transformValueProvider(
+    transformer: (JavaValueProvider) -> JavaValueProvider
+) = transformJavaFuzzingContext { it.transformValueProvider(transformer) }

utbot-framework/src/main/kotlin/org/utbot/framework/context/utils/JavaFuzzingContextUtils.kt

@@ -0,0 +1,15 @@
+package org.utbot.framework.context.utils
+
+import org.utbot.framework.context.JavaFuzzingContext
+import org.utbot.fuzzing.JavaValueProvider
+
+fun JavaFuzzingContext.transformValueProvider(
+    transformer: (JavaValueProvider) -> JavaValueProvider
+) = object : JavaFuzzingContext by this {
+    override val valueProvider: JavaValueProvider =
+        transformer(this@transformValueProvider.valueProvider)
+}
+
+fun JavaFuzzingContext.withValueProvider(
+    valueProvider: JavaValueProvider
+) = transformValueProvider { it.with(valueProvider) }