@@ -18,6 +18,7 @@ import org.usvm.types.scoreClassNode
1818import org.usvm.util.ApproximationPaths
1919import org.utbot.common.utBotTempDirectory
2020import org.utbot.framework.UtSettings
21+ import org.utbot.framework.assemble.AssembleModelGenerator
2122import org.utbot.framework.codegen.domain.builtin.UtilMethodProviderPlaceholder
2223import org.utbot.framework.context.ConcreteExecutionContext
2324import org.utbot.framework.fuzzer.ReferencePreservingIntIdGenerator
@@ -28,6 +29,8 @@ import org.utbot.framework.plugin.api.UtError
2829import org.utbot.framework.plugin.api.UtFailedExecution
2930import org.utbot.framework.plugin.api.UtResult
3031import org.utbot.framework.plugin.api.util.utContext
32+ import org.utbot.framework.plugin.api.UtExecution
33+ import org.utbot.framework.plugin.api.mapper.UtModelDeepMapper
3134import org.utbot.framework.plugin.services.JdkInfoService
3235import org.utbot.instrumentation.ConcreteExecutor
3336import 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