Fix compilation after rebase

UnitTestBot · Sep 19, 2023 · 98da383 · 98da383
1 parent 33c1c21
commit 98da383
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 5 deletions.
diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/context/JavaFuzzingContext.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/context/JavaFuzzingContext.kt
@@ -1,6 +1,5 @@
 package org.utbot.framework.context
 
-import org.utbot.engine.MockStrategy
 import org.utbot.framework.plugin.api.ClassId
 import org.utbot.framework.plugin.api.EnvironmentModels
 import org.utbot.framework.plugin.api.ExecutableId
@@ -12,7 +11,6 @@ import org.utbot.instrumentation.instrumentation.execution.UtConcreteExecutionRe
 
 interface JavaFuzzingContext {
     val classUnderTest: ClassId
-    val mockStrategy: MockStrategy
     val idGenerator: IdentityPreservingIdGenerator<Int>
     val valueProvider: JavaValueProvider
 

diff --git a/...-framework/src/main/kotlin/org/utbot/framework/context/simple/SimpleJavaFuzzingContext.kt b/...-framework/src/main/kotlin/org/utbot/framework/context/simple/SimpleJavaFuzzingContext.kt
@@ -1,6 +1,5 @@
 package org.utbot.framework.context.simple
 
-import org.utbot.engine.MockStrategy
 import org.utbot.framework.context.JavaFuzzingContext
 import org.utbot.framework.plugin.api.ClassId
 import org.utbot.framework.plugin.api.EnvironmentModels
@@ -15,7 +14,6 @@ import org.utbot.instrumentation.instrumentation.execution.UtConcreteExecutionRe
 
 class SimpleJavaFuzzingContext(
     override val classUnderTest: ClassId,
-    override val mockStrategy: MockStrategy,
     override val idGenerator: IdentityPreservingIdGenerator<Int>,
 ) : JavaFuzzingContext {
     override val valueProvider: JavaValueProvider =

diff --git a/...mework/src/main/kotlin/org/utbot/framework/context/spring/SpringApplicationContextImpl.kt b/...mework/src/main/kotlin/org/utbot/framework/context/spring/SpringApplicationContextImpl.kt
@@ -76,7 +76,7 @@ class SpringApplicationContextImpl(
                 fuzzingContext
                     .useMocks { type ->
                         ReplacedFuzzedTypeFlag !in type.properties &&
-                                fuzzingContext.mockStrategy.eligibleToMock(
+                                mockStrategy.eligibleToMock(
                                     classToMock = type.classId,
                                     classUnderTest = fuzzingContext.classUnderTest
                                 )