Construct assemble models for classes with val-properties in constructor

volivan239 · volivan239 · commit e106126c3c9d · 2022-10-26T19:22:15.000+03:00
diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/assemble/AssembleModelGenerator.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/assemble/AssembleModelGenerator.kt
@@ -254,9 +254,6 @@ class AssembleModelGenerator(private val basePackageName: String) {
                     if (fieldId.isStatic) {
                         throw AssembleException("Static field $fieldId can't be set in an object of the class $classId")
                     }
-                    if (fieldId.isFinal) {
-                        throw AssembleException("Final field $fieldId can't be set in an object of the class $classId")
-                    }
                     if (!fieldId.type.isAccessibleFrom(basePackageName)) {
                         throw AssembleException(
                             "Field $fieldId can't be set in an object of the class $classId because its type is inaccessible"
@@ -266,6 +263,10 @@ class AssembleModelGenerator(private val basePackageName: String) {
                     if (fieldId in constructorInfo.affectedFields ||
                             (fieldId !in constructorInfo.setFields && !fieldModel.hasDefaultValue())
                     ) {
+                        // Kotlin can set final fields in constructor, so we should only check for final fields not affected by constructor
+                        if (fieldId.isFinal) {
+                            throw AssembleException("Final field $fieldId can't be set in an object of the class $classId")
+                        }
                         val assembledModel = assembleModel(fieldModel)
                         val modifierCall = modifierCall(this, fieldId, assembledModel)
                         callChain.add(modifierCall)
