diff --git a/akka-coordination/src/test/scala/akka/coordination/lease/TestLease.scala b/akka-coordination/src/test/scala/akka/coordination/lease/TestLease.scala index e5df039b42..2f8e9c3750 100644 --- a/akka-coordination/src/test/scala/akka/coordination/lease/TestLease.scala +++ b/akka-coordination/src/test/scala/akka/coordination/lease/TestLease.scala @@ -64,16 +64,16 @@ class TestLease(settings: LeaseSettings, system: ExtendedActorSystem) extends Le val log = Logging(system, classOf[TestLease]) val probe = TestProbe()(system) - log.info("Creating lease {}", settings) - - TestLeaseExt(system).setTestLease(settings.leaseName, this) - val initialPromise = Promise[Boolean]() private val nextAcquireResult = new AtomicReference[Future[Boolean]](initialPromise.future) private val nextCheckLeaseResult = new AtomicReference[Boolean](false) private val currentCallBack = new AtomicReference[Option[Throwable] => Unit](_ => ()) + log.info("Creating lease {}", settings) + + TestLeaseExt(system).setTestLease(settings.leaseName, this) + def setNextAcquireResult(next: Future[Boolean]): Unit = nextAcquireResult.set(next)