Allows resource startup timeout to be configured

Configurable by a new system property
disneystreaming · Feb 25, 2022 · b0f0894 · b0f0894
1 parent 7fe40e2
commit b0f0894
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/modules/framework/src-jvm/RunnerCompat.scala b/modules/framework/src-jvm/RunnerCompat.scala
@@ -7,6 +7,7 @@ import java.util.concurrent.atomic.{ AtomicBoolean, AtomicInteger }
 
 import scala.concurrent.duration._
 import scala.concurrent.{ ExecutionContext, Promise }
+import scala.util.Try
 
 import cats.data.Chain
 import cats.effect.{ Sync, _ }
@@ -119,9 +120,10 @@ trait RunnerCompat[F[_]] { self: sbt.testing.Runner =>
                   gate)
     stillRunning.set(sbtTasks.size)
 
+    val maybeTimeout = Try(System.getProperty("weaver.test.startupTimeout")).toOption.flatMap(Option(_))
     // Waiting for the resources to be allocated.
     scala.concurrent.blocking {
-      scala.concurrent.Await.result(gate.future, 120.second)
+      scala.concurrent.Await.result(gate.future, maybeTimeout.getOrElse(120.second))
     }
     sbtTasks.toArray
   }