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 2d0d82a431..e049f21f3a 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 @@ -164,7 +164,8 @@ class CgClassBody( * A class representing the IntelliJ IDEA's regions. * A region is a part of code between the special starting and ending comments. * - * [header] The header of the region. + * @property header The header of the region, + * no ///region and ///endregion comments are generated if [header] is `null` */ sealed class CgRegion : CgElement { abstract val header: String? diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/renderer/CgAbstractRenderer.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/renderer/CgAbstractRenderer.kt index 39c41e3df0..0f0917cc5c 100644 --- a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/renderer/CgAbstractRenderer.kt +++ b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/renderer/CgAbstractRenderer.kt @@ -190,9 +190,10 @@ abstract class CgAbstractRenderer( private fun CgRegion<*>.render(printLineAfterContentEnd: Boolean = false) { if (content.isEmpty() || isInterrupted) return - print(regionStart) - header?.let { print(" $it") } - println() + header?.let { + print(regionStart) + println(" $it") + } if (this is CgTestMethodCluster) description?.accept(this@CgAbstractRenderer) @@ -208,7 +209,9 @@ abstract class CgAbstractRenderer( if (printLineAfterContentEnd) println() - println(regionEnd) + header?.let { + println(regionEnd) + } if (isLimitExceeded && !isInterrupted) { visit(CgSingleLineComment("Abrupt generation termination: file size exceeds configured limit (${FileUtil.byteCountToDisplaySize(UtSettings.maxTestFileSize.toLong())})")) diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/renderer/CgKotlinRenderer.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/renderer/CgKotlinRenderer.kt index 158cce48af..3dcb3f7f91 100644 --- a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/renderer/CgKotlinRenderer.kt +++ b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/renderer/CgKotlinRenderer.kt @@ -183,23 +183,6 @@ internal class CgKotlinRenderer(context: CgRendererContext, printer: CgPrinter = println("}") } - override fun visit(element: CgStaticsRegion) { - if (element.content.isEmpty()) return - - print(regionStart) - element.header?.let { print(" $it") } - println() - - for (item in element.content) { - println() - println("@JvmStatic") - item.accept(this) - } - - println(regionEnd) - } - - // Property access override fun visit(element: CgFieldAccess) { @@ -383,6 +366,12 @@ internal class CgKotlinRenderer(context: CgRendererContext, printer: CgPrinter = print(")") } + override fun visit(element: CgParameterizedTestDataProviderMethod) { + println() + println("@JvmStatic") + super.visit(element) + } + override fun renderRegularImport(regularImport: RegularImport) { val escapedImport = getEscapedImportRendering(regularImport) println("import $escapedImport$statementEnding") @@ -654,4 +643,4 @@ internal class CgKotlinRenderer(context: CgRendererContext, printer: CgPrinter = print(">") } } -} \ No newline at end of file +} diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgSpringTestClassConstructor.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgSpringTestClassConstructor.kt index cd3d751a1b..c63241a4ae 100644 --- a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgSpringTestClassConstructor.kt +++ b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgSpringTestClassConstructor.kt @@ -123,10 +123,10 @@ class CgSpringTestClassConstructor(context: CgContext): CgAbstractTestClassConst ) val methodCluster = CgMethodsCluster( - "Test run utility methods", - listOf(CgSimpleRegion("Mocking utils", listOf(beforeMethod, afterMethod))) + header = null, + listOf(CgSimpleRegion(header = null, listOf(beforeMethod, afterMethod))) ) return mockitoCloseableField to methodCluster } -} \ No newline at end of file +}