@@ -5,6 +5,8 @@ import org.utbot.framework.plugin.api.*
55import org.utbot.framework.plugin.api.util.voidClassId
66import org.utbot.fuzzer.FuzzedMethodDescription
77import org.utbot.fuzzer.FuzzedValue
8+ import org.utbot.summary.SummarySentenceConstants.FROM_TO_NAMES_COLON
9+ import org.utbot.summary.SummarySentenceConstants.FROM_TO_NAMES_TRANSITION
810import org.utbot.summary.UtSummarySettings
911import org.utbot.summary.comment.classic.fuzzer.SimpleCommentForTestProducedByFuzzerBuilder
1012import org.utbot.summary.comment.customtags.fuzzer.CommentWithCustomTagForTestProducedByFuzzerBuilder
@@ -33,8 +35,8 @@ class ModelBasedNameSuggester(
3335
3436 return sequenceOf(
3537 TestSuggestedInfo (
36- testName = createTestName(description, values, result),
37- displayName = createDisplayName(description, values, result),
38+ testName = if ( UtSummarySettings . GENERATE_NAMES ) createTestName(description, values, result) else null ,
39+ displayName = if ( UtSummarySettings . GENERATE_DISPLAY_NAMES ) createDisplayName(description, values, result) else null ,
3840 javaDoc = if (UtSummarySettings .GENERATE_COMMENTS ) createJavaDoc(description, values, result) else null
3941 )
4042 )
@@ -104,12 +106,17 @@ class ModelBasedNameSuggester(
104106 * Result example:
105107 * 1. **Full name**: `firstArg = 12, secondArg < 100.0, thirdArg = empty string -> throw IllegalArgumentException`
106108 * 2. **Name without appropriate information**: `arg_0 = 0 and others -> return 0`
109+ *
110+ * NOTE: The ```:``` symbol is used as a separator instead
111+ * of ```->``` if the [UtSummarySettings.GENERATE_DISPLAYNAME_FROM_TO_STYLE] is false.
107112 */
108113 private fun createDisplayName (
109114 description : FuzzedMethodDescription ,
110115 values : List <FuzzedValue >,
111116 result : UtExecutionResult ?
112117 ): String {
118+ val displayNameSeparator = if (UtSummarySettings .GENERATE_DISPLAYNAME_FROM_TO_STYLE ) FROM_TO_NAMES_TRANSITION else FROM_TO_NAMES_COLON
119+
113120 val summaries = values.asSequence()
114121 .mapIndexed { index, value ->
115122 value.summary?.replace(" %var%" , description.parameterNameMap(index) ? : " arg_$index " )
@@ -127,13 +134,13 @@ class ModelBasedNameSuggester(
127134 val returnValue = when (result) {
128135 is UtExecutionSuccess -> result.model.let { m ->
129136 when {
130- m is UtPrimitiveModel && m.classId != voidClassId -> " -> return " + m.value
131- m is UtNullModel -> " -> return null"
137+ m is UtPrimitiveModel && m.classId != voidClassId -> " $displayNameSeparator return " + m.value
138+ m is UtNullModel -> " $displayNameSeparator return null"
132139 else -> null
133140 }
134141 }
135142
136- is UtExplicitlyThrownException , is UtImplicitlyThrownException -> " -> throw ${(result as UtExecutionFailure ).exception::class .java.simpleName} "
143+ is UtExplicitlyThrownException , is UtImplicitlyThrownException -> " $displayNameSeparator throw ${(result as UtExecutionFailure ).exception::class .java.simpleName} "
137144 else -> null
138145 }
139146
0 commit comments