Skip to content

Commit e26100f

Browse files
committed
Handle modelTagName for UtCustomModel
1 parent 14f5dcb commit e26100f

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/builders/SpringTestClassModelBuilder.kt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,9 @@ class SpringTestClassModelBuilder(val context: CgContext): TestClassModelBuilder
113113
is UtClassRefModel,
114114
is UtVoidModel,
115115
is UtEnumConstantModel -> {}
116-
is UtCustomModel -> currentModel.origin?.let { collectRecursively(it.wrap(), allModels) }
116+
is UtCustomModel -> currentModel.origin?.let {
117+
collectRecursively(it.wrap(currentModelWrapper.modelTagName), allModels)
118+
}
117119
is UtLambdaModel -> {
118120
currentModel.capturedValues.forEach { collectRecursively(it.wrap(), allModels) }
119121
}

0 commit comments

Comments
 (0)