Skip to content

Commit e87b5b9

Browse files
Correct annotations for parameterized tests in TestNg #2384 (#2385)
1 parent 199a315 commit e87b5b9

File tree

1 file changed

+11
-8
lines changed

1 file changed

+11
-8
lines changed

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/services/framework/TestFrameworkManager.kt

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -294,10 +294,11 @@ internal class TestNgManager(context: CgContext) : TestFrameworkManager(context)
294294
}
295295

296296
override fun addDataProviderAnnotations(dataProviderMethodName: String) {
297+
val nameArgument = CgNamedAnnotationArgument("name", stringLiteral(dataProviderMethodName))
297298
statementConstructor.addAnnotation(
298-
testFramework.methodSourceAnnotationId,
299-
listOf("name" to stringLiteral(dataProviderMethodName)),
300-
Method,
299+
classId = testFramework.methodSourceAnnotationId,
300+
namedArguments = listOf(nameArgument),
301+
target = Method,
301302
)
302303
}
303304

@@ -307,11 +308,13 @@ internal class TestNgManager(context: CgContext) : TestFrameworkManager(context)
307308
}
308309

309310
override fun addParameterizedTestAnnotations(dataProviderMethodName: String?) {
310-
statementConstructor.addAnnotation(
311-
testFramework.parameterizedTestAnnotationId,
312-
listOf("dataProvider" to CgLiteral(stringClassId, dataProviderMethodName)),
313-
Method,
314-
)
311+
val dataProviderArgument =
312+
CgNamedAnnotationArgument("dataProvider", CgLiteral(stringClassId, dataProviderMethodName))
313+
statementConstructor.addAnnotation(
314+
classId = testFramework.parameterizedTestAnnotationId,
315+
namedArguments = listOf(dataProviderArgument),
316+
target = Method,
317+
)
315318
}
316319

317320
override fun passArgumentsToArgsVariable(argsVariable: CgVariable, argsArray: CgVariable, executionIndex: Int) =

0 commit comments

Comments
 (0)