TS-38628 Fix null partition

cqse · Dec 13, 2024 · 2c58ab9 · 2c58ab9
1 parent 12d4308
commit 2c58ab9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/...engine/src/main/kotlin/com/teamscale/test_impacted/test_descriptor/TestDescriptorUtils.kt b/...engine/src/main/kotlin/com/teamscale/test_impacted/test_descriptor/TestDescriptorUtils.kt
@@ -99,7 +99,7 @@ object TestDescriptorUtils {
 	@JvmStatic
 	fun getAvailableTests(
 		rootTestDescriptor: TestDescriptor,
-		partition: String?
+		partition: String
 	): AvailableTests {
 		val availableTests = AvailableTests()
 

diff --git a/...t-engine/src/test/kotlin/com/teamscale/test_impacted/engine/ImpactedTestEngineTestBase.kt b/...t-engine/src/test/kotlin/com/teamscale/test_impacted/engine/ImpactedTestEngineTestBase.kt
@@ -22,6 +22,11 @@ abstract class ImpactedTestEngineTestBase {
 
 	@Test
 	fun testEngineExecution() {
+		whenever(impactedTestsProvider.getImpactedTestsFromTeamscale(any()))
+			.thenReturn(impactedTests)
+		whenever(impactedTestsProvider.partition)
+			.thenReturn("partition")
+
 		val testEngine = createInternalImpactedTestEngine(engines)
 
 		val engineDescriptor = testEngine
@@ -33,8 +38,6 @@ abstract class ImpactedTestEngineTestBase {
 			.thenReturn(executionListener)
 		whenever(executionRequest.rootTestDescriptor)
 			.thenReturn(engineDescriptor)
-		whenever(impactedTestsProvider.getImpactedTestsFromTeamscale(any()))
-			.thenReturn(impactedTests)
 
 		testEngine.execute(executionRequest)