Bug fixes in CgAbstractSpringTestClassConstructor (#2269)

Author: EgorkaKulikov

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

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

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/generator/SpringCodeGenerator.kt

@@ -61,8 +61,7 @@ class SpringCodeGenerator(
         logger.info { "Code generation phase started at ${now()}" }
         val astConstructor = when (springTestsType) {
             SpringTestsType.UNIT_TESTS -> CgSpringUnitTestClassConstructor(context)
-            // TODO replace with CgSpringIntegrationTestClassConstructor
-            SpringTestsType.INTEGRATION_TESTS -> CgSpringUnitTestClassConstructor(context)
+            SpringTestsType.INTEGRATION_TESTS -> CgSpringIntegrationTestClassConstructor(context)
         }
         val testClassFile = astConstructor.construct(testClassModel)
         logger.info { "Code generation phase finished at ${now()}" }

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

@@ -1,6 +1,7 @@
 package org.utbot.framework.codegen.tree
 
 import org.utbot.framework.codegen.domain.builtin.TestClassUtilMethodProvider
+import org.utbot.framework.codegen.domain.builtin.autowiredClassId
 import org.utbot.framework.codegen.domain.builtin.injectMocksClassId
 import org.utbot.framework.codegen.domain.builtin.mockClassId
 import org.utbot.framework.codegen.domain.context.CgContext
@@ -87,7 +88,11 @@ abstract class CgAbstractSpringTestClassConstructor(context: CgContext):
         groupedModelsByClassId: TypedModelWrappers,
         annotationClassId: ClassId
     ): List<CgFieldDeclaration> {
-        require(annotationClassId == injectMocksClassId || annotationClassId == mockClassId) {
+        require(
+            annotationClassId == injectMocksClassId ||
+                    annotationClassId == mockClassId ||
+                    annotationClassId == autowiredClassId
+        ) {
             error("Unexpected annotation classId -- $annotationClassId")
         }