Added a method and class references

amandelpie · amandelpie · commit 6ada057c721b · 2022-10-05T19:49:46.000+03:00
diff --git a/utbot-summary/src/main/kotlin/org/utbot/summary/comment/customtags/fuzzer/CommentWithCustomTagForTestProducedByFuzzerBuilder.kt b/utbot-summary/src/main/kotlin/org/utbot/summary/comment/customtags/fuzzer/CommentWithCustomTagForTestProducedByFuzzerBuilder.kt
@@ -1,10 +1,13 @@
 package org.utbot.summary.comment.customtags.fuzzer
 
+import org.utbot.framework.plugin.api.ClassId
 import org.utbot.framework.plugin.api.DocCustomTagStatement
 import org.utbot.framework.plugin.api.DocStatement
 import org.utbot.framework.plugin.api.UtExecutionResult
 import org.utbot.fuzzer.FuzzedMethodDescription
 import org.utbot.fuzzer.FuzzedValue
+import org.utbot.summary.SummarySentenceConstants
+import org.utbot.summary.SummarySentenceConstants.CARRIAGE_RETURN
 import org.utbot.summary.comment.customtags.symbolic.CustomJavaDocTagProvider
 
 /**
@@ -26,19 +29,45 @@ class CommentWithCustomTagForTestProducedByFuzzerBuilder(
     }
 
     private fun buildCustomJavaDocComment(): CommentWithCustomTagForTestProducedByFuzzer {
-        /*val methodReference = getMethodReference(
-            currentMethod.declaringClass.name,
-            currentMethod.name,
-            currentMethod.parameterTypes
+        val methodReference = getMethodReference(
+            methodDescription.packageName!! + "." + methodDescription.className!!,
+            methodDescription.compilableName!!,
+            methodDescription.parameters
         )
-        val classReference = getClassReference(currentMethod.declaringClass.javaStyleName)*/
+        val classReference = getClassReference(methodDescription.packageName!! + "." +methodDescription.className!!)
 
         val javaDocComment = CommentWithCustomTagForTestProducedByFuzzer(
-            classUnderTest = methodDescription.className!!,
-            methodUnderTest = methodDescription.name,
+            classUnderTest = classReference.replace(CARRIAGE_RETURN, ""),
+            methodUnderTest = methodReference.replace(CARRIAGE_RETURN, ""),
         )
 
-
         return javaDocComment
     }
+
+    /**
+     * Returns a reference to the invoked method.
+     *
+     * It looks like {@link packageName.className#methodName(type1, type2)}.
+     *
+     * In case when an enclosing class in nested, we need to replace '$' with '.'
+     * to render the reference.
+     */
+    private fun getMethodReference(className: String, methodName: String, methodParameterTypes: List<ClassId>): String {
+        val prettyClassName: String = className.replace("$", ".")
+
+        return if (methodParameterTypes.isEmpty()) {
+            "{@link $prettyClassName#$methodName()}"
+        } else {
+            val methodParametersAsString = methodParameterTypes.joinToString(",") { it.canonicalName }
+            "{@link $prettyClassName#$methodName($methodParametersAsString)}"
+        }
+    }
+
+    /**
+     * Returns a reference to the class.
+     * Replaces '$' with '.' in case a class is nested.
+     */
+    private fun getClassReference(fullClassName: String): String {
+        return "{@link ${fullClassName.replace("$", ".")}}"
+    }
 }
