Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
SpringModelUtils cleanup
  • Loading branch information
EgorkaKulikov committed Jun 28, 2023
commit 7cb15f29cd969ba93d5107c1f9930292c9e8eed2
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,8 @@ object SpringModelUtils {
val springExtensionClassId = ClassId("org.springframework.test.context.junit.jupiter.SpringExtension")

val bootstrapWithClassId = ClassId("org.springframework.test.context.BootstrapWith")
val springBootTestContextBootstrapper = ClassId("org.springframework.boot.test.context.SpringBootTestContextBootstrapper")
val springBootTestContextBootstrapperClassId =
ClassId("org.springframework.boot.test.context.SpringBootTestContextBootstrapper")


// most likely only one persistent library is on the classpath, but we need to be able to work with either of them
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ import org.utbot.framework.plugin.api.util.SpringModelUtils.autowiredClassId
import org.utbot.framework.plugin.api.util.SpringModelUtils.bootstrapWithClassId
import org.utbot.framework.plugin.api.util.SpringModelUtils.dirtiesContextClassId
import org.utbot.framework.plugin.api.util.SpringModelUtils.dirtiesContextClassModeClassId
import org.utbot.framework.plugin.api.util.SpringModelUtils.springBootTestContextBootstrapper
import org.utbot.framework.plugin.api.util.SpringModelUtils.springBootTestContextBootstrapperClassId
import org.utbot.framework.plugin.api.util.SpringModelUtils.transactionalClassId
import org.utbot.framework.plugin.api.util.utContext

Expand All @@ -29,9 +29,7 @@ class CgSpringIntegrationTestClassConstructor(context: CgContext) : CgAbstractSp

private fun collectSpringSpecificAnnotations() {
testFrameworkManager.addAnnotationForSpringRunner()
if (annotationTypeIsAccessible(bootstrapWithClassId)) {
addAnnotation(bootstrapWithClassId, createGetClassExpression(springBootTestContextBootstrapper), Class)
}
addAnnotation(bootstrapWithClassId, createGetClassExpression(springBootTestContextBootstrapperClassId), Class)

addAnnotation(
classId = dirtiesContextClassId,
Expand Down