Skip to content

Commit 225042c

Browse files
committed
Use outermost test class for test class util method provider and renderer context
1 parent 843d444 commit 225042c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/constructor/context/CgContext.kt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -476,7 +476,7 @@ internal data class CgContext(
476476
get() = if (generateUtilClassFile) {
477477
UtilClassFileMethodProvider
478478
} else {
479-
TestClassUtilMethodProvider(currentTestClass)
479+
TestClassUtilMethodProvider(outerMostTestClass)
480480
}
481481

482482
override lateinit var currentTestClass: ClassId

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ internal class CgRendererContext(
3333
importedClasses = context.importedClasses,
3434
importedStaticMethods = context.importedStaticMethods,
3535
classPackageName = context.testClassPackageName,
36-
generatedClass = context.currentTestClass,
36+
generatedClass = context.outerMostTestClass,
3737
utilMethodProvider = context.utilMethodProvider,
3838
codegenLanguage = context.codegenLanguage,
3939
mockFrameworkUsed = context.mockFrameworkUsed,

0 commit comments

Comments
 (0)