Skip to content

Commit 0518a20

Browse files
committed
Add assemble models in USVM symbolic engine
1 parent 607d5d7 commit 0518a20

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

utbot-framework/src/main/kotlin/org/utbot/engine/UsvmSymbolicEngine.kt

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ import org.usvm.types.scoreClassNode
1818
import org.usvm.util.ApproximationPaths
1919
import org.utbot.common.utBotTempDirectory
2020
import org.utbot.framework.UtSettings
21+
import org.utbot.framework.assemble.AssembleModelGenerator
2122
import org.utbot.framework.codegen.domain.builtin.UtilMethodProviderPlaceholder
2223
import org.utbot.framework.context.ConcreteExecutionContext
2324
import org.utbot.framework.fuzzer.ReferencePreservingIntIdGenerator
@@ -28,6 +29,8 @@ import org.utbot.framework.plugin.api.UtError
2829
import org.utbot.framework.plugin.api.UtFailedExecution
2930
import org.utbot.framework.plugin.api.UtResult
3031
import org.utbot.framework.plugin.api.util.utContext
32+
import org.utbot.framework.plugin.api.UtExecution
33+
import org.utbot.framework.plugin.api.mapper.UtModelDeepMapper
3134
import org.utbot.framework.plugin.services.JdkInfoService
3235
import org.utbot.instrumentation.ConcreteExecutor
3336
import org.utbot.instrumentation.instrumentation.execution.UtConcreteExecutionResult
@@ -112,6 +115,13 @@ object UsvmSymbolicEngine {
112115
runStandardConcreteExecution(concreteExecutor, executableId, initialState)
113116
}
114117

118+
if(utResult is UtExecution) {
119+
val assembleModelGenerator = AssembleModelGenerator(executableId.classId.packageName)
120+
utResult.mapModels(UtModelDeepMapper { model ->
121+
assembleModelGenerator.createAssembleModels(listOf(model)).getValue(model)
122+
})
123+
}
124+
115125
utResult?.let {
116126
collectedExecutions.add(executableId to it)
117127
}

0 commit comments

Comments
 (0)