diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/CgElement.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/CgElement.kt index f0e45d4b63..522dffd3eb 100644 --- a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/CgElement.kt +++ b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/CgElement.kt @@ -173,7 +173,7 @@ class CgFieldDeclaration( val ownerClassId: ClassId, val declaration: CgDeclaration, val annotation: CgAnnotation? = null, - val visibility: VisibilityModifier = VisibilityModifier.PUBLIC, + val visibility: VisibilityModifier = VisibilityModifier.PRIVATE, ) : CgElement /** diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgAbstractSpringTestClassConstructor.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgAbstractSpringTestClassConstructor.kt index 4e7f4761b8..287f18f915 100644 --- a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgAbstractSpringTestClassConstructor.kt +++ b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgAbstractSpringTestClassConstructor.kt @@ -37,8 +37,6 @@ abstract class CgAbstractSpringTestClassConstructor(context: CgContext): fields += constructClassFields(testClassModel) clearUnwantedVariableModels() - methodRegions += constructAdditionalMethods() - for ((testSetIndex, testSet) in testClassModel.methodTestSets.withIndex()) { updateCurrentExecutable(testSet.executableId) withTestSetIdScope(testSetIndex) { @@ -51,6 +49,8 @@ abstract class CgAbstractSpringTestClassConstructor(context: CgContext): } } + methodRegions += constructAdditionalMethods() + if (currentTestClass == outerMostTestClass) { val utilEntities = collectUtilEntities() // If utilMethodProvider is TestClassUtilMethodProvider, then util entities should be declared