From 65ed044d5efa08392eb179cfc03f8eb922f4c0c5 Mon Sep 17 00:00:00 2001
From: Egor Kulikov <egor.k.kulikov@gmail.com>
Date: Tue, 1 Aug 2023 16:24:13 +0300
Subject: [PATCH] Avoid collecting useless models in
 SpringTestClassModelBuilder

---
 .../domain/models/builders/SpringTestClassModelBuilder.kt | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/builders/SpringTestClassModelBuilder.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/builders/SpringTestClassModelBuilder.kt
index 8b24281948..3a51292b7b 100644
--- a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/builders/SpringTestClassModelBuilder.kt
+++ b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/builders/SpringTestClassModelBuilder.kt
@@ -110,10 +110,8 @@ class SpringTestClassModelBuilder(val context: CgContext):
             is UtPrimitiveModel,
             is UtClassRefModel,
             is UtVoidModel,
-            is UtEnumConstantModel -> {}
-            is UtCustomModel -> currentModel.origin?.let {
-                collectRecursively(it.wrap(currentModelWrapper.modelTagName), allModels)
-            }
+            is UtEnumConstantModel,
+            is UtCustomModel-> {}
             is UtLambdaModel -> {
                 currentModel.capturedValues.forEach { collectRecursively(it.wrap(), allModels) }
             }
@@ -134,8 +132,6 @@ class SpringTestClassModelBuilder(val context: CgContext):
                 }
             }
             is UtAssembleModel -> {
-                currentModel.origin?.let { collectRecursively(it.wrap(), allModels) }
-
                 currentModel.instantiationCall.instance?.let { collectRecursively(it.wrap(), allModels) }
                 currentModel.instantiationCall.params.forEach { collectRecursively(it.wrap(), allModels) }