Disable test generation for synthetic methods of data classes

Author: volivan239

utbot-cli/src/main/kotlin/org/utbot/cli/GenerateTestsCommand.kt

@@ -50,8 +50,8 @@ class GenerateTestsCommand :
         help = "Specifies source code file for a generated test"
     )
         .required()
-        .check("Must exist and ends with *.java suffix") {
-            it.endsWith(".java") && Files.exists(Paths.get(it))
+        .check("Must exist and end with .java or .kt suffix") {
+            (it.endsWith(".java") || it.endsWith(".kt")) && Files.exists(Paths.get(it))
         }
 
     private val projectRoot by option(

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

@@ -363,6 +363,9 @@ val ClassId.isIterableOrMap: Boolean
 val ClassId.isEnum: Boolean
     get() = jClass.isEnum
 
+val ClassId.isData: Boolean
+    get() = kClass.isData
+
 fun ClassId.findFieldByIdOrNull(fieldId: FieldId): Field? {
     if (isNotSubtypeOf(fieldId.declaringClass)) {
         return null

utbot-framework/src/main/kotlin/org/utbot/framework/util/SyntheticMethods.kt

@@ -1,14 +1,18 @@
 package org.utbot.framework.util
 
 import org.utbot.framework.plugin.api.ExecutableId
-import org.utbot.framework.plugin.api.util.humanReadableName
+import org.utbot.framework.plugin.api.util.isData
 import org.utbot.framework.plugin.api.util.isEnum
 
-fun isKnownSyntheticMethod(method: ExecutableId): Boolean =
+fun isKnownSyntheticMethod(method: ExecutableId): Boolean {
     if (method.classId.isEnum)
-        method.humanReadableName.substringBefore('(') in KnownSyntheticMethodNames.enumSyntheticMethodNames
-    else
-        false
+        return method.name in KnownSyntheticMethodNames.enumSyntheticMethodNames
+
+    if (method.classId.isData)
+        return KnownSyntheticMethodNames.dataClassSyntheticMethodNames.any { it.matches(method.name) }
+
+    return false
+}
 
 /**
  * Contains names of methods that are always autogenerated and thus it is unlikely that
@@ -17,4 +21,14 @@ fun isKnownSyntheticMethod(method: ExecutableId): Boolean =
 private object KnownSyntheticMethodNames {
     /** List with names of enum methods that are autogenerated */
     val enumSyntheticMethodNames = listOf("values", "valueOf")
+
+    /** List with names of data classes methods that are autogenerated in Kotlin */
+    val dataClassSyntheticMethodNames = listOf(
+        "equals",
+        "hashCode",
+        "toString",
+        "copy",
+        "copy\\\$default",
+        "component[1-9][0-9]*"
+    ).map { it.toRegex() }
 }

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/util/PsiClassHelper.kt

@@ -9,6 +9,7 @@ import com.intellij.testIntegration.TestIntegrationUtils
 import org.jetbrains.kotlin.asJava.elements.KtLightMethod
 import org.jetbrains.kotlin.asJava.elements.isGetter
 import org.jetbrains.kotlin.asJava.elements.isSetter
+import org.jetbrains.kotlin.psi.KtClass
 import org.utbot.common.filterWhen
 import org.utbot.framework.UtSettings
 
@@ -22,8 +23,15 @@ private val PsiMember.isKotlinGetterOrSetter: Boolean
         return isGetter || isSetter
     }
 
-private fun Iterable<MemberInfo>.filterTestableMethods(): List<MemberInfo> = this
-    .filterWhen(UtSettings.skipTestGenerationForSyntheticMethods) { it.member !is SyntheticElement }
+// By now, we think that method in Kotlin is synthetic iff navigation to its declaration leads to its declaring class
+// rather than the method itself (because synthetic methods don't have bodies that we can navigate to)
+private val PsiMember.isSyntheticKotlinMethod: Boolean
+    get() = this is KtLightMethod && navigationElement is KtClass
+
+fun Iterable<MemberInfo>.filterTestableMethods(): List<MemberInfo> = this
+    .filterWhen(UtSettings.skipTestGenerationForSyntheticMethods) {
+        it.member !is SyntheticElement && !it.member.isSyntheticKotlinMethod
+    }
     .filterNot { it.member.isAbstract }
     .filterNot { it.member.isKotlinGetterOrSetter }