Skip to content

Commit 9a4a101

Browse files
committed
Remove setOption for DTO tests, set targetFormat when writing values in DTO output
1 parent 38818e9 commit 9a4a101

4 files changed

Lines changed: 3 additions & 13 deletions

File tree

core-tests/e2e-tests/spring/spring-rest-openapi-v3/src/test/kotlin/org/evomaster/e2etests/spring/openapi/v3/oracledisable/SSRFBaseDisableEMTest.kt

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,6 @@ class SSRFBaseDisableEMTest : SpringTestBase() {
3939
setOption(args, "schemaOracles", "false")
4040
setOption(args, "disabledOracleCodes", DefinedFaultCategory.SSRF.code.toString())
4141

42-
// TODO: Remove once EnumGene is supported for DTOs
43-
setOption(args, "dtoForRequestPayload","true")
44-
4542
val solution = initAndRun(args)
4643

4744
assertTrue(solution.individuals.isNotEmpty())

core-tests/e2e-tests/spring/spring-rest-openapi-v3/src/test/kotlin/org/evomaster/e2etests/spring/openapi/v3/security/ssrf/base/SSRFBaseEMTest.kt

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,10 +37,6 @@ class SSRFBaseEMTest : SpringTestBase() {
3737
setOption(args, "languageModelConnector", "false")
3838
setOption(args, "schemaOracles", "false")
3939

40-
// TODO: Remove once EnumGene is supported for DTOs
41-
setOption(args, "dtoForRequestPayload","true")
42-
// setOption(args, "dtoForRequestPayload","false")
43-
4440
val solution = initAndRun(args)
4541

4642
assertTrue(solution.individuals.isNotEmpty())

core-tests/e2e-tests/spring/spring-rest-openapi-v3/src/test/kotlin/org/evomaster/e2etests/spring/openapi/v3/swagger/SwaggerDescriptionEMTest.kt

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,6 @@ class SwaggerDescriptionEMTest : SpringTestBase() {
2626
500
2727
) { args: MutableList<String> ->
2828

29-
// TODO: Remove once EnumGene is supported for DTOs
30-
setOption(args, "dtoForRequestPayload","true")
31-
3229
val solution = initAndRun(args)
3330

3431
Assertions.assertTrue(solution.individuals.isNotEmpty())

core/src/main/kotlin/org/evomaster/core/output/dto/GeneToDto.kt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -109,9 +109,9 @@ class GeneToDto(
109109
if (leafGene is EnumGene<*> && it is ChoiceGene<*>) {
110110
val children = it.getViewOfChildren()
111111
val otherChoice = children.find { child -> child != leafGene }
112-
result.add(dtoOutput.getSetterStatement(dtoVarName, attributeName, "${leafGene.getValueAsPrintableString(targetFormat = null)}${getValueSuffix(otherChoice)}"))
112+
result.add(dtoOutput.getSetterStatement(dtoVarName, attributeName, "${leafGene.getValueAsPrintableString(targetFormat = outputFormat)}${getValueSuffix(otherChoice)}"))
113113
} else {
114-
result.add(dtoOutput.getSetterStatement(dtoVarName, attributeName, "${leafGene.getValueAsPrintableString(targetFormat = null)}${getValueSuffix(leafGene)}"))
114+
result.add(dtoOutput.getSetterStatement(dtoVarName, attributeName, "${leafGene.getValueAsPrintableString(targetFormat = outputFormat)}${getValueSuffix(leafGene)}"))
115115
}
116116
}
117117
}
@@ -142,7 +142,7 @@ class GeneToDto(
142142
} else {
143143
gene.getViewOfElements().forEach {
144144
val leafGene = it.getLeafGene()
145-
result.add(dtoOutput.getAddElementToListStatement(listVarName, "${leafGene.getValueAsPrintableString(targetFormat = null)}${getValueSuffix(leafGene)}"))
145+
result.add(dtoOutput.getAddElementToListStatement(listVarName, "${leafGene.getValueAsPrintableString(targetFormat = outputFormat)}${getValueSuffix(leafGene)}"))
146146
}
147147
}
148148

0 commit comments

Comments
 (0)