Skip to content

Commit 49d1bc5

Browse files
committed
Do render nested classes in Java and Kotlin
1 parent 225042c commit 49d1bc5

File tree

3 files changed

+3
-7
lines changed

3 files changed

+3
-7
lines changed

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/tree/CgElement.kt

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -185,11 +185,7 @@ data class CgTestClassBody(
185185
val testMethodRegions: List<CgExecutableUnderTestCluster>,
186186
val staticDeclarationRegions: List<CgStaticsRegion>,
187187
val nestedClassRegions: List<CgRegion<CgTestClass>>
188-
) : AbstractCgClassBody() {
189-
190-
val regions: List<CgRegion<*>>
191-
get() = testMethodRegions
192-
}
188+
) : AbstractCgClassBody()
193189

194190
/**
195191
* A class representing the IntelliJ IDEA's regions.

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/visitor/CgJavaRenderer.kt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ internal class CgJavaRenderer(context: CgRendererContext, printer: CgPrinter = C
9696

9797
override fun visit(element: CgTestClassBody) {
9898
// render regions for test methods and utils
99-
val allRegions = element.testMethodRegions + element.staticDeclarationRegions
99+
val allRegions = element.testMethodRegions + element.nestedClassRegions + element.staticDeclarationRegions
100100
for ((i, region) in allRegions.withIndex()) {
101101
if (i != 0) println()
102102

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/visitor/CgKotlinRenderer.kt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ internal class CgKotlinRenderer(context: CgRendererContext, printer: CgPrinter =
117117

118118
override fun visit(element: CgTestClassBody) {
119119
// render regions for test methods
120-
for ((i, region) in (element.testMethodRegions).withIndex()) {
120+
for ((i, region) in (element.testMethodRegions + element.nestedClassRegions).withIndex()) {
121121
if (i != 0) println()
122122

123123
region.accept(this)

0 commit comments

Comments
 (0)