Support hidden fields access in code generation

Author: ArsenHD

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

@@ -164,6 +164,9 @@ internal interface CgContextOwner {
     // java.lang.reflect.Executable is a superclass of both of these types.
     var declaredExecutableRefs: PersistentMap<ExecutableId, CgVariable>
 
+    // Variables of java.lang.reflect.Field type declared in the current name scope
+    var declaredFieldRefs: PersistentMap<FieldId, CgVariable>
+
     // generated this instance for method under test
     var thisInstance: CgValue?
 
@@ -310,6 +313,7 @@ internal interface CgContextOwner {
         val prevVariableNames = existingVariableNames
         val prevDeclaredClassRefs = declaredClassRefs
         val prevDeclaredExecutableRefs = declaredExecutableRefs
+        val prevDeclaredFieldRefs = declaredFieldRefs
         val prevValueByModel = IdentityHashMap(valueByModel)
         val prevValueByModelId = valueByModelId.toMutableMap()
         return try {
@@ -318,6 +322,7 @@ internal interface CgContextOwner {
             existingVariableNames = prevVariableNames
             declaredClassRefs = prevDeclaredClassRefs
             declaredExecutableRefs = prevDeclaredExecutableRefs
+            declaredFieldRefs = prevDeclaredFieldRefs
             valueByModel = prevValueByModel
             valueByModelId = prevValueByModelId
         }
@@ -415,6 +420,7 @@ internal data class CgContext(
     override var existingVariableNames: PersistentSet<String> = persistentSetOf(),
     override var declaredClassRefs: PersistentMap<ClassId, CgVariable> = persistentMapOf(),
     override var declaredExecutableRefs: PersistentMap<ExecutableId, CgVariable> = persistentMapOf(),
+    override var declaredFieldRefs: PersistentMap<FieldId, CgVariable> = persistentMapOf(),
     override var thisInstance: CgValue? = null,
     override val methodArguments: MutableList<CgValue> = mutableListOf(),
     override val codeGenerationErrors: MutableMap<CgMethodTestSet, MutableMap<String, Int>> = mutableMapOf(),

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/constructor/tree/CgCallableAccessManager.kt

@@ -17,7 +17,6 @@ import org.utbot.framework.codegen.model.tree.CgGetJavaClass
 import org.utbot.framework.codegen.model.tree.CgValue
 import org.utbot.framework.codegen.model.tree.CgVariable
 import org.utbot.framework.codegen.model.util.at
-import org.utbot.framework.codegen.model.util.get
 import org.utbot.framework.codegen.model.util.isAccessibleFrom
 import org.utbot.framework.codegen.model.util.stringLiteral
 import org.utbot.framework.fields.ArrayElementAccess

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/constructor/tree/CgFieldStateManager.kt

@@ -67,7 +67,6 @@ import org.utbot.framework.codegen.model.tree.convertDocToCg
 import org.utbot.framework.codegen.model.tree.toStatement
 import org.utbot.framework.codegen.model.util.canBeSetIn
 import org.utbot.framework.codegen.model.util.equalTo
-import org.utbot.framework.codegen.model.util.get
 import org.utbot.framework.codegen.model.util.inc
 import org.utbot.framework.codegen.model.util.isAccessibleFrom
 import org.utbot.framework.codegen.model.util.length

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/constructor/tree/CgMethodConstructor.kt

@@ -24,7 +24,6 @@ import org.utbot.framework.codegen.model.tree.CgValue
 import org.utbot.framework.codegen.model.tree.CgVariable
 import org.utbot.framework.codegen.model.util.at
 import org.utbot.framework.codegen.model.util.canBeSetIn
-import org.utbot.framework.codegen.model.util.get
 import org.utbot.framework.codegen.model.util.inc
 import org.utbot.framework.codegen.model.util.isAccessibleFrom
 import org.utbot.framework.codegen.model.util.lessThan

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/constructor/tree/CgVariableConstructor.kt

@@ -57,9 +57,17 @@ import org.utbot.framework.plugin.api.util.isSubtypeOf
 import org.utbot.framework.plugin.api.util.objectArrayClassId
 import org.utbot.framework.plugin.api.util.objectClassId
 import fj.data.Either
+import org.utbot.framework.codegen.model.constructor.builtin.getDeclaredConstructor
+import org.utbot.framework.codegen.model.constructor.builtin.getDeclaredField
+import org.utbot.framework.codegen.model.constructor.builtin.getDeclaredMethod
 import org.utbot.framework.codegen.model.tree.CgArrayInitializer
+import org.utbot.framework.codegen.model.tree.CgGetJavaClass
 import org.utbot.framework.codegen.model.tree.CgIsInstance
+import org.utbot.framework.plugin.api.ConstructorId
+import org.utbot.framework.plugin.api.FieldId
+import org.utbot.framework.plugin.api.MethodId
 import org.utbot.framework.plugin.api.util.classClassId
+import org.utbot.framework.plugin.api.util.isPrimitive
 import java.lang.reflect.Constructor
 import java.lang.reflect.Method
 import kotlin.reflect.KFunction
@@ -113,6 +121,19 @@ interface CgStatementConstructor {
     fun doWhileLoop(condition: CgExpression, statements: () -> Unit)
     fun forEachLoop(init: CgForEachLoopBuilder.() -> Unit)
 
+    /**
+     * Create a variable of type [java.lang.reflect.Field] by the given [FieldId].
+     */
+    fun createFieldVariable(fieldId: FieldId): CgVariable
+
+
+    /**
+     * Given an [executableId] that represents method or constructor and a list of arguments for it,
+     * create a variable of type [java.lang.reflect.Method] or [java.lang.reflect.Constructor].
+     * This created variable is returned.
+     */
+    fun createExecutableVariable(executableId: ExecutableId, arguments: List<CgExpression>): CgVariable
+
     fun tryBlock(init: () -> Unit): CgTryCatch
     fun tryBlock(init: () -> Unit, resources: List<CgDeclaration>?): CgTryCatch
     fun CgTryCatch.catch(exception: ClassId, init: (CgVariable) -> Unit): CgTryCatch
@@ -284,6 +305,46 @@ internal class CgStatementConstructorImpl(context: CgContext) :
         currentBlock += buildCgForEachLoop(init)
     }
 
+    override fun createFieldVariable(fieldId: FieldId): CgVariable {
+        val declaringClass = newVar(Class::class.id) { Class::class.id[forName](fieldId.declaringClass.name) }
+        val name = fieldId.name + "Field"
+        return newVar(java.lang.reflect.Field::class.id, name) {
+            declaringClass[getDeclaredField](fieldId.name)
+        }
+    }
+
+    override fun createExecutableVariable(executableId: ExecutableId, arguments: List<CgExpression>): CgVariable {
+        val declaringClass = newVar(Class::class.id) { Class::class.id[forName](executableId.classId.name) }
+        val argTypes = (arguments zip executableId.parameters).map { (argument, parameterType) ->
+            val baseName = when (argument) {
+                is CgVariable -> "${argument.name}Type"
+                else -> "${parameterType.prettifiedName.decapitalize()}Type"
+            }
+            newVar(classCgClassId, baseName) {
+                if (parameterType.isPrimitive) {
+                    CgGetJavaClass(parameterType)
+                } else {
+                    Class::class.id[forName](parameterType.name)
+                }
+            }
+        }
+
+        return when (executableId) {
+            is MethodId -> {
+                val name = executableId.name + "Method"
+                newVar(java.lang.reflect.Method::class.id, name) {
+                    declaringClass[getDeclaredMethod](executableId.name, *argTypes.toTypedArray())
+                }
+            }
+            is ConstructorId -> {
+                val name = executableId.classId.prettifiedName.decapitalize() + "Constructor"
+                newVar(java.lang.reflect.Constructor::class.id, name) {
+                    declaringClass[getDeclaredConstructor](*argTypes.toTypedArray())
+                }
+            }
+        }
+    }
+
     override fun tryBlock(init: () -> Unit): CgTryCatch = tryBlock(init, null)
 
     override fun tryBlock(init: () -> Unit, resources: List<CgDeclaration>?): CgTryCatch =

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/constructor/util/CgStatementConstructor.kt

@@ -68,17 +68,6 @@ fun charLiteral(c: Char) = CgLiteral(charClassId, c)
 
 fun stringLiteral(string: String) = CgLiteral(stringClassId, string)
 
-// Field access
-
-// non-static fields
-operator fun CgExpression.get(fieldId: FieldId): CgFieldAccess =
-    CgFieldAccess(this, fieldId)
-
-// static fields
-// TODO: unused receiver
-operator fun ClassId.get(fieldId: FieldId): CgStaticFieldAccess =
-    CgStaticFieldAccess(fieldId)
-
 // Get array length
 
 /**

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/util/DslUtil.kt

@@ -8,7 +8,7 @@ import org.utbot.framework.plugin.api.ExecutableId
  *
  * @param packageName name of the package we check accessibility from
  */
-fun ExecutableId.isAccessibleFrom(packageName: String): Boolean {
+infix fun ExecutableId.isAccessibleFrom(packageName: String): Boolean {
     val isAccessibleFromPackageByModifiers = isPublic || (classId.packageName == packageName && (isPackagePrivate || isProtected))
 
     return classId.isAccessibleFrom(packageName) && isAccessibleFromPackageByModifiers