Skip to content

Commit 756cb5c

Browse files
committed
Add JUnit extension to control USE_CUSTOM_TAGS setting
1 parent 6792f06 commit 756cb5c

File tree

5 files changed

+34
-8
lines changed

5 files changed

+34
-8
lines changed
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
package examples
2+
3+
import org.junit.jupiter.api.extension.AfterEachCallback
4+
import org.junit.jupiter.api.extension.BeforeEachCallback
5+
import org.junit.jupiter.api.extension.ExtensionContext
6+
import org.utbot.summary.UtSummarySettings
7+
8+
class CustomJavaDocTagsEnabler(private val enable: Boolean = true) : BeforeEachCallback, AfterEachCallback {
9+
private var previousValue = false
10+
11+
override fun beforeEach(context: ExtensionContext?) {
12+
previousValue = UtSummarySettings.USE_CUSTOM_JAVADOC_TAGS
13+
UtSummarySettings.USE_CUSTOM_JAVADOC_TAGS = enable
14+
}
15+
16+
override fun afterEach(context: ExtensionContext?) {
17+
UtSummarySettings.USE_CUSTOM_JAVADOC_TAGS = previousValue
18+
}
19+
}

utbot-summary-tests/src/test/kotlin/examples/SummaryTestCaseGeneratorTest.kt

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -50,15 +50,13 @@ open class SummaryTestCaseGeneratorTest(
5050
summaryKeys: List<String>,
5151
methodNames: List<String> = listOf(),
5252
displayNames: List<String> = listOf(),
53-
clusterInfo: List<Pair<UtClusterInfo, Int>> = listOf(),
54-
useCustomTags: Boolean = false
53+
clusterInfo: List<Pair<UtClusterInfo, Int>> = listOf()
5554
) {
5655
workaround(WorkaroundReason.HACK) {
5756
// @todo change to the constructor parameter
5857
checkSolverTimeoutMillis = 0
5958
checkNpeInNestedMethods = true
6059
checkNpeInNestedNotPrivateMethods = true
61-
UtSummarySettings.USE_CUSTOM_JAVADOC_TAGS = useCustomTags
6260
}
6361
val utMethod = UtMethod.from(method)
6462
val testSet = executionsModel(utMethod, mockStrategy)

utbot-summary-tests/src/test/kotlin/examples/controlflow/SummaryConditionsTest.kt

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,14 @@
11
package examples.controlflow
22

3+
import examples.CustomJavaDocTagsEnabler
34
import examples.SummaryTestCaseGeneratorTest
45
import org.junit.jupiter.api.Test
6+
import org.junit.jupiter.api.extension.ExtendWith
57
import org.utbot.examples.DoNotCalculate
68
import org.utbot.examples.controlflow.Conditions
79
import org.utbot.framework.plugin.api.MockStrategyApi
810

11+
@ExtendWith(CustomJavaDocTagsEnabler::class)
912
class SummaryConditionsTest : SummaryTestCaseGeneratorTest(
1013
Conditions::class
1114
) {
@@ -46,6 +49,6 @@ class SummaryConditionsTest : SummaryTestCaseGeneratorTest(
4649
val mockStrategy = MockStrategyApi.NO_MOCKS
4750
val coverage = DoNotCalculate
4851

49-
summaryCheck(method, mockStrategy, coverage, summaryKeys, methodNames, displayNames, useCustomTags = true)
52+
summaryCheck(method, mockStrategy, coverage, summaryKeys, methodNames, displayNames)
5053
}
5154
}

utbot-summary-tests/src/test/kotlin/examples/exceptions/SummaryExceptionClusteringExamplesTest.kt

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,14 @@
11
package examples.exceptions
22

3+
import examples.CustomJavaDocTagsEnabler
34
import examples.SummaryTestCaseGeneratorTest
45
import org.junit.jupiter.api.Test
6+
import org.junit.jupiter.api.extension.ExtendWith
57
import org.utbot.examples.DoNotCalculate
68
import org.utbot.examples.exceptions.ExceptionClusteringExamples
79
import org.utbot.framework.plugin.api.MockStrategyApi
810

11+
@ExtendWith(CustomJavaDocTagsEnabler::class)
912
class SummaryExceptionClusteringExamplesTest : SummaryTestCaseGeneratorTest(
1013
ExceptionClusteringExamples::class
1114
) {
@@ -69,6 +72,6 @@ class SummaryExceptionClusteringExamplesTest : SummaryTestCaseGeneratorTest(
6972
val mockStrategy = MockStrategyApi.NO_MOCKS
7073
val coverage = DoNotCalculate
7174

72-
summaryCheck(method, mockStrategy, coverage, summaryKeys, methodNames, displayNames, useCustomTags = true)
75+
summaryCheck(method, mockStrategy, coverage, summaryKeys, methodNames, displayNames)
7376
}
7477
}

utbot-summary-tests/src/test/kotlin/examples/structures/SummaryMinStackTest.kt

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,14 @@
11
package examples.structures
22

3+
import examples.CustomJavaDocTagsEnabler
34
import examples.SummaryTestCaseGeneratorTest
45
import org.junit.jupiter.api.Test
6+
import org.junit.jupiter.api.extension.ExtendWith
57
import org.utbot.examples.DoNotCalculate
68
import org.utbot.examples.structures.MinStack
79
import org.utbot.framework.plugin.api.MockStrategyApi
810

11+
@ExtendWith(CustomJavaDocTagsEnabler::class)
912
class SummaryMinStackTest : SummaryTestCaseGeneratorTest(
1013
MinStack::class
1114
) {
@@ -53,7 +56,7 @@ class SummaryMinStackTest : SummaryTestCaseGeneratorTest(
5356
val mockStrategy = MockStrategyApi.NO_MOCKS
5457
val coverage = DoNotCalculate
5558

56-
summaryCheck(method, mockStrategy, coverage, summaryKeys, methodNames, displayNames, useCustomTags = true)
59+
summaryCheck(method, mockStrategy, coverage, summaryKeys, methodNames, displayNames)
5760
}
5861

5962
@Test
@@ -92,7 +95,7 @@ class SummaryMinStackTest : SummaryTestCaseGeneratorTest(
9295
val mockStrategy = MockStrategyApi.NO_MOCKS
9396
val coverage = DoNotCalculate
9497

95-
summaryCheck(method, mockStrategy, coverage, summaryKeys, methodNames, displayNames, useCustomTags = true)
98+
summaryCheck(method, mockStrategy, coverage, summaryKeys, methodNames, displayNames)
9699
}
97100

98101
@Test
@@ -197,6 +200,6 @@ class SummaryMinStackTest : SummaryTestCaseGeneratorTest(
197200
val mockStrategy = MockStrategyApi.NO_MOCKS
198201
val coverage = DoNotCalculate
199202

200-
summaryCheck(method, mockStrategy, coverage, summaryKeys, methodNames, displayNames, useCustomTags = true)
203+
summaryCheck(method, mockStrategy, coverage, summaryKeys, methodNames, displayNames)
201204
}
202205
}

0 commit comments

Comments
 (0)