Improve rendering of UtUtils documentation

Author: EgorkaKulikov

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

@@ -124,6 +124,7 @@ open class CgClassFile(
 
 class CgClass(
     val id: ClassId,
+    val documentation: CgDocumentationComment?,
     val annotations: List<CgAnnotation>,
     val superclass: ClassId?,
     val interfaces: List<ClassId>,
@@ -149,7 +150,6 @@ class CgClass(
  */
 class CgClassBody(
     val classId: ClassId,
-    val documentation: CgDocumentationComment?,
     val methodRegions: List<CgMethodsCluster>,
     val staticDeclarationRegions: List<CgStaticsRegion>,
     val nestedClassRegions: List<CgNestedClassesRegion<*>>

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

@@ -61,6 +61,8 @@ internal class CgJavaRenderer(context: CgRendererContext, printer: CgPrinter = C
         get() = this == context.generatedClass
 
     override fun visit(element: CgClass) {
+        element.documentation?.accept(this)
+
         for (annotation in element.annotations) {
             annotation.accept(this)
         }
@@ -87,8 +89,6 @@ internal class CgJavaRenderer(context: CgRendererContext, printer: CgPrinter = C
     }
 
     override fun visit(element: CgClassBody) {
-        element.documentation?.accept(this)
-
         // render regions for test methods and utils
         val allRegions = element.methodRegions + element.nestedClassRegions + element.staticDeclarationRegions
         for ((i, region) in allRegions.withIndex()) {

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

@@ -75,6 +75,8 @@ internal class CgKotlinRenderer(context: CgRendererContext, printer: CgPrinter =
         get() = (this == context.generatedClass) || isKotlinFile
 
     override fun visit(element: CgClass) {
+        element.documentation?.accept(this)
+
         for (annotation in element.annotations) {
             annotation.accept(this)
         }
@@ -115,8 +117,6 @@ internal class CgKotlinRenderer(context: CgRendererContext, printer: CgPrinter =
     }
 
     override fun visit(element: CgClassBody) {
-        element.documentation?.accept(this)
-
         // render regions for test methods
         for ((i, region) in (element.methodRegions + element.nestedClassRegions).withIndex()) {
             if (i != 0) println()

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

@@ -49,25 +49,25 @@ fun buildClassFile(init: CgClassFileBuilder.() -> Unit) = CgClassFileBuilder().a
 
 class CgClassBuilder : CgBuilder<CgClass> {
     lateinit var id: ClassId
+    var documentation: CgDocumentationComment? = null
     val annotations: MutableList<CgAnnotation> = mutableListOf()
     var superclass: ClassId? = null
     val interfaces: MutableList<ClassId> = mutableListOf()
     var isStatic: Boolean = false
     var isNested: Boolean = false
     lateinit var body: CgClassBody
 
-    override fun build() = CgClass(id, annotations, superclass, interfaces, body, isStatic, isNested)
+    override fun build() = CgClass(id, documentation, annotations, superclass, interfaces, body, isStatic, isNested)
 }
 
 fun buildClass(init: CgClassBuilder.() -> Unit) = CgClassBuilder().apply(init).build()
 
 class CgClassBodyBuilder(val classId: ClassId) : CgBuilder<CgClassBody> {
-    var documentation: CgDocumentationComment? = null
     val methodRegions: MutableList<CgMethodsCluster> = mutableListOf()
     val staticDeclarationRegions: MutableList<CgStaticsRegion> = mutableListOf()
     val nestedClassRegions: MutableList<CgNestedClassesRegion<*>> = mutableListOf()
 
-    override fun build() = CgClassBody(classId, documentation, methodRegions, staticDeclarationRegions, nestedClassRegions)
+    override fun build() = CgClassBody(classId, methodRegions, staticDeclarationRegions, nestedClassRegions)
 }
 
 fun buildClassBody(classId: ClassId, init: CgClassBodyBuilder.() -> Unit) = CgClassBodyBuilder(classId).apply(init).build()

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/ututils/CgUtilClassConstructor.kt

@@ -27,8 +27,8 @@ internal object CgUtilClassConstructor {
             // so they will be imported once IDEA reformatting action has worked
             declaredClass = buildClass {
                 id = utilsClassId
+                documentation = utilClassKind.utilClassDocumentation(codegenLanguage)
                 body = buildClassBody(utilsClassId) {
-                    documentation = utilClassKind.utilClassDocumentation(codegenLanguage)
                     staticDeclarationRegions += CgStaticsRegion("Util methods", utilMethodProvider.utilMethodIds.map { CgUtilMethod(it) })
                     nestedClassRegions += CgAuxiliaryNestedClassesRegion(
                         nestedClasses = listOf(