@@ -19,6 +19,7 @@ import org.junit.jupiter.api.AfterAll
1919import org.junit.jupiter.api.AfterEach
2020import org.junit.jupiter.api.BeforeEach
2121import org.junit.jupiter.api.Test
22+ import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable
2223import org.kodein.di.DI
2324import org.kodein.di.bindSingleton
2425import software.aws.toolkits.jetbrains.uitests.TestCIServer
@@ -30,6 +31,7 @@ import java.io.File
3031import java.io.FileOutputStream
3132import java.nio.file.Path
3233import java.nio.file.Paths
34+ import kotlin.time.Duration.Companion.minutes
3335
3436class FeatureDevTest {
3537 init {
@@ -94,6 +96,7 @@ class FeatureDevTest {
9496 }
9597
9698 @Test
99+ @EnabledIfEnvironmentVariable(named = " ENABLE_ITERATION_TEST" , matches = " true" )
97100 fun `Iterate code generation` () {
98101 val testCase = TestCase (
99102 IdeProductProvider .IC ,
@@ -114,8 +117,8 @@ class FeatureDevTest {
114117
115118 copyExistingConfig(Paths .get(" tstData" , " configAmazonQTests" ))
116119 updateGeneralSettings()
117- }.runIdeWithDriver()
118- .useDriverAndCloseIde {
120+ }.runIdeWithDriver(runTimeout = 15 .minutes )
121+ .useDriverAndCloseIde( 15 .minutes) {
119122 waitForProjectOpen()
120123 // required wait time for the system to be fully ready
121124 Thread .sleep(30000 )
@@ -126,6 +129,7 @@ class FeatureDevTest {
126129 }
127130
128131 @Test
132+ @EnabledIfEnvironmentVariable(named = " ENABLE_ITERATION_TEST" , matches = " true" )
129133 fun `Start new code generation` () {
130134 val testCase = TestCase (
131135 IdeProductProvider .IC ,
@@ -146,8 +150,8 @@ class FeatureDevTest {
146150
147151 copyExistingConfig(Paths .get(" tstData" , " configAmazonQTests" ))
148152 updateGeneralSettings()
149- }.runIdeWithDriver()
150- .useDriverAndCloseIde {
153+ }.runIdeWithDriver(runTimeout = 15 .minutes )
154+ .useDriverAndCloseIde( 15 .minutes) {
151155 waitForProjectOpen()
152156 // required wait time for the system to be fully ready
153157 Thread .sleep(30000 )
0 commit comments