Add origin to UtSpringMockMvcResultActionsModel

Author: IlyaMuravjov

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

@@ -761,6 +761,7 @@ data class SpringRepositoryId(
 
 class UtSpringMockMvcResultActionsModel(
     id: Int?,
+    origin: UtCompositeModel?,
     val status: Int,
     val errorMessage: String?,
     val contentAsString: String,
@@ -769,6 +770,7 @@ class UtSpringMockMvcResultActionsModel(
     val model: UtModel?
     // TODO add headers and other data
 ) : UtCustomModel(
+    origin = origin,
     classId = SpringModelUtils.resultActionsClassId,
     id = id,
     modelName = "mockMvcResultActions@$id"

utbot-instrumentation/src/main/kotlin/org/utbot/instrumentation/instrumentation/execution/constructors/UtModelConstructor.kt

@@ -17,6 +17,13 @@ interface UtModelConstructorInterface {
      * Constructs a UtModel from a concrete [value] with a specific [classId].
      */
     fun construct(value: Any?, classId: ClassId): UtModel
+
+    /**
+     * Constructs UtCompositeModel.
+     *
+     * Uses runtime javaClass to collect ALL fields, except final static fields, and builds this model recursively.
+     */
+    fun constructCompositeModel(value: Any): UtCompositeModel
 }
 
 /**
@@ -314,12 +321,12 @@ class UtModelConstructor(
             }
         }
 
-    /**
-     * Constructs UtCompositeModel.
-     *
-     * Uses runtime javaClass to collect ALL fields, except final static fields, and builds this model recursively.
-     */
-    private fun constructCompositeModel(value: Any, remainingDepth: Long): UtModel {
+    override fun constructCompositeModel(value: Any): UtCompositeModel = constructCompositeModel(
+        value,
+        remainingDepth = maxDepth
+    )
+
+    private fun constructCompositeModel(value: Any, remainingDepth: Long): UtCompositeModel {
         // value can be mock only if it was previously constructed from UtCompositeModel
         val isMock = objectToModelCache[value]?.isMockModel() ?: false
 
@@ -350,6 +357,9 @@ class UtModelConstructor(
     private fun withMaxDepth(newMaxDepth: Long) = object : UtModelConstructorInterface {
         override fun construct(value: Any?, classId: ClassId): UtModel =
             construct(value, classId, newMaxDepth)
+
+        override fun constructCompositeModel(value: Any): UtCompositeModel =
+            constructCompositeModel(value, newMaxDepth)
     }
 }
 

utbot-instrumentation/src/main/kotlin/org/utbot/instrumentation/instrumentation/spring/UtMockMvcResultActionsModelConstructor.kt

@@ -30,6 +30,7 @@ class UtMockMvcResultActionsModelConstructor : UtCustomModelConstructor {
 
         return UtSpringMockMvcResultActionsModel(
             id = id,
+            origin = internalConstructor.constructCompositeModel(value),
             status = responseGetStatusMethodId.method.invoke(response) as Int,
             errorMessage = responseGetErrorMessageMethodId.method.invoke(response) as String?,
             contentAsString = responseGetContentAsStringMethodId.method.invoke(response) as String,