raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 11

harden CrossDcHeartbeatSenderSpec - wait until active (#31219)

Browse source
This commit is contained in:
Renato Cavalcanti 2022-03-08 18:51:33 +01:00 • committed by GitHub
parent 1371cb6a35
commit ad5864c6d8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

28
akka-cluster/src/test/scala/akka/cluster/CrossDcHeartbeatSenderSpec.scala
View file

@ -6,20 +6,25 @@ package akka.cluster
import scala.collection.immutable.SortedSet
import akka.actor.{ ActorSelection, Address, Props }
import akka.actor.ActorSelection
import akka.actor.Address
import akka.actor.Props
import akka.cluster.ClusterEvent.CurrentClusterState
import akka.cluster.ClusterHeartbeatSender.Heartbeat
import akka.cluster.CrossDcHeartbeatSender.ReportStatus
import akka.cluster.CrossDcHeartbeatSenderSpec.TestCrossDcHeartbeatSender
import akka.testkit.{ AkkaSpec, ImplicitSender, TestProbe }
import akka.testkit.AkkaSpec
import akka.testkit.ImplicitSender
import akka.testkit.TestProbe
import akka.util.Version
object CrossDcHeartbeatSenderSpec {
class TestCrossDcHeartbeatSender(probe: TestProbe) extends CrossDcHeartbeatSender {
class TestCrossDcHeartbeatSender(heartbeatProbe: TestProbe) extends CrossDcHeartbeatSender {
// disable register for cluster events
override def preStart(): Unit = {}
override def heartbeatReceiver(address: Address): ActorSelection = {
context.actorSelection(probe.ref.path)
context.actorSelection(heartbeatProbe.ref.path)
}
}
}
@ -36,18 +41,25 @@ class CrossDcHeartbeatSenderSpec extends AkkaSpec("""
""") with ImplicitSender {
"CrossDcHeartBeatSender" should {
"increment heart beat sequence nr" in {
val probe = TestProbe()
val heartbeatProbe = TestProbe()
Cluster(system).join(Cluster(system).selfMember.address)
awaitAssert(Cluster(system).selfMember.status == MemberStatus.Up)
val underTest = system.actorOf(Props(new TestCrossDcHeartbeatSender(probe)))
val underTest = system.actorOf(Props(new TestCrossDcHeartbeatSender(heartbeatProbe)))
underTest ! CurrentClusterState(
members = SortedSet(
Cluster(system).selfMember,
Member(UniqueAddress(Address("akka", system.name), 2L), Set("dc-dc2"), Version.Zero)
.copy(status = MemberStatus.Up)))
probe.expectMsgType[Heartbeat].sequenceNr shouldEqual 1
probe.expectMsgType[Heartbeat].sequenceNr shouldEqual 2
awaitAssert {
underTest ! ReportStatus()
expectMsgType[CrossDcHeartbeatSender.MonitoringActive]
}
heartbeatProbe.expectMsgType[Heartbeat].sequenceNr shouldEqual 1
heartbeatProbe.expectMsgType[Heartbeat].sequenceNr shouldEqual 2
}
}
}