Add specific annotations for Spring integration testing (#2281)

Author: EgorkaKulikov

utbot-framework-api/src/main/kotlin/org/utbot/framework/plugin/api/util/SpringModelUtils.kt

@@ -10,9 +10,28 @@ import org.utbot.framework.plugin.api.UtPrimitiveModel
 import org.utbot.framework.plugin.api.UtSpringContextModel
 
 object SpringModelUtils {
+ val autowiredClassId = ClassId("org.springframework.beans.factory.annotation.Autowired")
+
  val applicationContextClassId = ClassId("org.springframework.context.ApplicationContext")
  val crudRepositoryClassId = ClassId("org.springframework.data.repository.CrudRepository")
 
+ @Suppress("unused", "may be used instead of ExtendWith + BootstrapWith in future")
+ val springBootTestClassId = ClassId("org.springframework.boot.test.context.SpringBootTest")
+
+ val dirtiesContextClassId = ClassId("org.springframework.test.annotation.DirtiesContext")
+ val dirtiesContextClassModeClassId = ClassId("org.springframework.test.annotation.DirtiesContext\$ClassMode")
+ val transactionalClassId = ClassId("org.springframework.transaction.annotation.Transactional")
+ val autoConfigureTestDbClassId = ClassId("org.springframework.boot.test.autoconfigure.jdbc.AutoConfigureTestDatabase")
+
+ val runWithClassId = ClassId("org.junit.runner.RunWith")
+ val extendWithClassId = ClassId("org.junit.jupiter.api.extension.ExtendWith")
+ val springExtensionClassId = ClassId("org.springframework.test.context.junit.jupiter.SpringExtension")
+
+ val bootstrapWithClassId = ClassId("org.springframework.test.context.BootstrapWith")
+ 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
  private val persistentLibraries = listOf("javax.persistence", "jakarta.persistence")
  private fun persistentClassIds(simpleName: String) = persistentLibraries.map { ClassId("$it.$simpleName") }

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/builtin/SpringBuiltins.kt

@@ -1,8 +0,0 @@
-package org.utbot.framework.codegen.domain.builtin
-
-import org.utbot.framework.plugin.api.BuiltinClassId
-
-internal val autowiredClassId = BuiltinClassId(
- canonicalName = "org.springframework.beans.factory.annotation.Autowired",
- simpleName = "Autowired",
-)

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

@@ -7,14 +7,6 @@ import org.utbot.framework.codegen.domain.ParametrizedTestSource
 import org.utbot.framework.codegen.domain.RuntimeExceptionTestsBehaviour
 import org.utbot.framework.codegen.domain.StaticsMocking
 import org.utbot.framework.codegen.domain.TestFramework
-import org.utbot.framework.codegen.domain.models.CgAnnotation
-import org.utbot.framework.codegen.domain.models.CgExecutableCall
-import org.utbot.framework.codegen.domain.models.CgStatement
-import org.utbot.framework.codegen.domain.models.CgStatementExecutableCall
-import org.utbot.framework.codegen.domain.models.CgTestMethod
-import org.utbot.framework.codegen.domain.models.CgThisInstance
-import org.utbot.framework.codegen.domain.models.CgValue
-import org.utbot.framework.codegen.domain.models.CgVariable
 import kotlinx.collections.immutable.PersistentList
 import kotlinx.collections.immutable.PersistentMap
 import kotlinx.collections.immutable.PersistentSet
@@ -23,12 +15,10 @@ import kotlinx.collections.immutable.persistentMapOf
 import kotlinx.collections.immutable.persistentSetOf
 import org.utbot.framework.codegen.domain.UtModelWrapper
 import org.utbot.framework.codegen.domain.ProjectType
-import org.utbot.framework.codegen.domain.models.CgMethodTestSet
 import org.utbot.framework.codegen.domain.builtin.TestClassUtilMethodProvider
 import org.utbot.framework.codegen.domain.builtin.UtilClassFileMethodProvider
 import org.utbot.framework.codegen.domain.builtin.UtilMethodProvider
-import org.utbot.framework.codegen.domain.models.SimpleTestClassModel
-import org.utbot.framework.codegen.domain.models.CgParameterKind
+import org.utbot.framework.codegen.domain.models.*
 import org.utbot.framework.codegen.services.access.Block
 import org.utbot.framework.codegen.tree.EnvironmentFieldStateCache
 import org.utbot.framework.codegen.tree.importIfNeeded
@@ -304,11 +294,13 @@ interface CgContextOwner {
  }
  }
 
- fun addAnnotation(annotation: CgAnnotation) {
- if (collectedMethodAnnotations.add(annotation)) {
- importIfNeeded(annotation.classId) // TODO: check how JUnit annotations are loaded
+ fun createGetClassExpression(id: ClassId, codegenLanguage: CodegenLanguage): CgGetClass =
+ when (codegenLanguage) {
+ CodegenLanguage.JAVA -> CgGetJavaClass(id)
+ CodegenLanguage.KOTLIN -> CgGetKotlinClass(id)
+ }.also {
+ importIfNeeded(id)
  }
- }
 
  /**
  * This method sets up context for a new test class file generation and executes the given [block].

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

@@ -357,22 +357,46 @@ enum class CgTestMethodType(val displayName: String, val isThrowing: Boolean) {
 
 // Annotations
 
+enum class AnnotationTarget {
+ Class,
+
+ Method,
+
+ Field,
+}
+
 abstract class CgAnnotation : CgElement {
  abstract val classId: ClassId
+ abstract val target: AnnotationTarget
 }
 
+/**
+ * NOTE: use `StatementConstructor.addAnnotation`
+ * instead of explicit constructor call.
+ */
 class CgCommentedAnnotation(val annotation: CgAnnotation) : CgAnnotation() {
  override val classId: ClassId = annotation.classId
+ override val target: AnnotationTarget = annotation.target
 }
 
+/**
+ * NOTE: use `StatementConstructor.addAnnotation`
+ * instead of explicit constructor call.
+ */
 class CgSingleArgAnnotation(
  override val classId: ClassId,
- val argument: CgExpression
+ val argument: CgExpression,
+ override val target: AnnotationTarget,
 ) : CgAnnotation()
 
+/**
+ * NOTE: use `StatementConstructor.addAnnotation`
+ * instead of explicit constructor call.
+ */
 class CgMultipleArgsAnnotation(
  override val classId: ClassId,
- val arguments: MutableList<CgNamedAnnotationArgument>
+ val arguments: MutableList<CgNamedAnnotationArgument>,
+ override val target: AnnotationTarget,
 ) : CgAnnotation()
 
 data class CgArrayAnnotationArgument(
@@ -942,8 +966,16 @@ sealed class CgGetClass : CgReferenceExpression {
  override val type: ClassId = Class::class.id
 }
 
+/**
+ * NOTE: use `CgContext.createGetClassExpression`
+ * instead of the explicit constructor call.
+ */
 data class CgGetJavaClass(override val classId: ClassId) : CgGetClass()
 
+/**
+ * NOTE: use `CgContext.createGetClassExpression`
+ * instead of the explicit constructor call.
+ */
 data class CgGetKotlinClass(override val classId: ClassId) : CgGetClass()
 
 // Executable calls