@@ -59,6 +59,7 @@ import org.utbot.framework.codegen.model.tree.CgNonStaticRunnable
5959import org.utbot.framework.codegen.model.tree.CgParameterDeclaration
6060import org.utbot.framework.codegen.model.tree.CgParameterizedTestDataProviderMethod
6161import org.utbot.framework.codegen.model.tree.CgRegion
62+ import org.utbot.framework.codegen.model.tree.CgRegularClass
6263import org.utbot.framework.codegen.model.tree.CgRegularClassBody
6364import org.utbot.framework.codegen.model.tree.CgRegularClassFile
6465import org.utbot.framework.codegen.model.tree.CgReturnStatement
@@ -71,6 +72,7 @@ import org.utbot.framework.codegen.model.tree.CgStatementExecutableCall
7172import org.utbot.framework.codegen.model.tree.CgStaticFieldAccess
7273import org.utbot.framework.codegen.model.tree.CgStaticRunnable
7374import org.utbot.framework.codegen.model.tree.CgStaticsRegion
75+ import org.utbot.framework.codegen.model.tree.CgTestClass
7476import org.utbot.framework.codegen.model.tree.CgTestClassFile
7577import org.utbot.framework.codegen.model.tree.CgTestMethod
7678import org.utbot.framework.codegen.model.tree.CgTestMethodCluster
@@ -151,6 +153,14 @@ internal abstract class CgAbstractRenderer(
151153 visit(element as AbstractCgClassFile <* >)
152154 }
153155
156+ override fun visit (element : CgRegularClass ) {
157+ visit(element as AbstractCgClass <* >)
158+ }
159+
160+ override fun visit (element : CgTestClass ) {
161+ visit(element as AbstractCgClass <* >)
162+ }
163+
154164 override fun visit (element : AbstractCgClassBody ) {
155165 visit(element as CgElement )
156166 }
0 commit comments