Handle modelTagName for UtCustomModel

UnitTestBot · Jul 28, 2023 · e26100f · e26100f
1 parent 14f5dcb
commit e26100f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.../kotlin/org/utbot/framework/codegen/domain/models/builders/SpringTestClassModelBuilder.kt b/.../kotlin/org/utbot/framework/codegen/domain/models/builders/SpringTestClassModelBuilder.kt
@@ -113,7 +113,9 @@ class SpringTestClassModelBuilder(val context: CgContext): TestClassModelBuilder
                 is UtClassRefModel,
                 is UtVoidModel,
                 is UtEnumConstantModel -> {}
-                is UtCustomModel -> currentModel.origin?.let { collectRecursively(it.wrap(), allModels) }
+                is UtCustomModel -> currentModel.origin?.let {
+                    collectRecursively(it.wrap(currentModelWrapper.modelTagName), allModels)
+                }
                 is UtLambdaModel -> {
                     currentModel.capturedValues.forEach { collectRecursively(it.wrap(), allModels) }
                 }