Increase timeout of long tests to avoid CI flakiness

scylladb · Aug 31, 2024 · 329d677 · 329d677
1 parent 588fcbf
commit 329d677
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/tests/src/test/scala/com/scylladb/migrator/alternator/SkippedSegmentsTest.scala b/tests/src/test/scala/com/scylladb/migrator/alternator/SkippedSegmentsTest.scala
@@ -4,11 +4,14 @@ import com.scylladb.migrator.SparkUtils.{submitSparkJob, successfullyPerformMigr
 import software.amazon.awssdk.services.dynamodb.model.{AttributeValue, PutItemRequest, ResourceNotFoundException, ScanRequest}
 
 import java.util.UUID
+import scala.concurrent.duration.{Duration, DurationInt}
 import scala.jdk.CollectionConverters._
 import scala.util.chaining._
 
 class SkippedSegmentsTest extends MigratorSuiteWithDynamoDBLocal {
 
+  override val munitTimeout: Duration = 120.seconds
+
   withTable("SkippedSegments").test("Run partial migrations") { tableName =>
     // We rely on the fact that both config files have `scanSegments: 3` and
     // complementary `skipSegments` properties

diff --git a/tests/src/test/scala/com/scylladb/migrator/alternator/ValidatorTest.scala b/tests/src/test/scala/com/scylladb/migrator/alternator/ValidatorTest.scala
@@ -3,11 +3,14 @@ package com.scylladb.migrator.alternator
 import com.scylladb.migrator.SparkUtils.{submitSparkJob, successfullyPerformMigration}
 import software.amazon.awssdk.services.dynamodb.model.{AttributeAction, AttributeValue, AttributeValueUpdate, PutItemRequest, UpdateItemRequest}
 
+import scala.concurrent.duration.{Duration, DurationInt}
 import scala.jdk.CollectionConverters._
 import scala.util.chaining._
 
 class ValidatorTest extends MigratorSuiteWithDynamoDBLocal {
 
+  override val munitTimeout: Duration = 120.seconds
+
   withTable("BasicTest").test("Validate migration") { tableName =>
     val configFile = "dynamodb-to-alternator-basic.yaml"
 

diff --git a/tests/src/test/scala/com/scylladb/migrator/scylla/ValidatorTest.scala b/tests/src/test/scala/com/scylladb/migrator/scylla/ValidatorTest.scala
@@ -5,11 +5,14 @@ import com.datastax.oss.driver.api.querybuilder.QueryBuilder.literal
 import com.datastax.oss.driver.api.querybuilder.term.Term
 import com.scylladb.migrator.SparkUtils.{submitSparkJob, successfullyPerformMigration}
 
+import scala.concurrent.duration.{Duration, DurationInt}
 import scala.jdk.CollectionConverters._
 import scala.util.chaining.scalaUtilChainingOps
 
 class ValidatorTest extends MigratorSuite(sourcePort = 9043) {
 
+  override val munitTimeout: Duration = 120.seconds
+
   withTable("BasicTest").test("Validate migration") { tableName =>
     val configFile = "cassandra-to-scylla-basic.yaml"