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 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.ClusterEvent.CurrentClusterState
import akka.cluster.ClusterHeartbeatSender.Heartbeat import akka.cluster.ClusterHeartbeatSender.Heartbeat
import akka.cluster.CrossDcHeartbeatSender.ReportStatus
import akka.cluster.CrossDcHeartbeatSenderSpec.TestCrossDcHeartbeatSender 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 import akka.util.Version
object CrossDcHeartbeatSenderSpec { object CrossDcHeartbeatSenderSpec {
class TestCrossDcHeartbeatSender(probe: TestProbe) extends CrossDcHeartbeatSender { class TestCrossDcHeartbeatSender(heartbeatProbe: TestProbe) extends CrossDcHeartbeatSender {
// disable register for cluster events // disable register for cluster events
override def preStart(): Unit = {} override def preStart(): Unit = {}
override def heartbeatReceiver(address: Address): ActorSelection = { 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 { """) with ImplicitSender {
"CrossDcHeartBeatSender" should { "CrossDcHeartBeatSender" should {
"increment heart beat sequence nr" in { "increment heart beat sequence nr" in {
val probe = TestProbe()
val heartbeatProbe = TestProbe()
Cluster(system).join(Cluster(system).selfMember.address) Cluster(system).join(Cluster(system).selfMember.address)
awaitAssert(Cluster(system).selfMember.status == MemberStatus.Up) 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( underTest ! CurrentClusterState(
members = SortedSet( members = SortedSet(
Cluster(system).selfMember, Cluster(system).selfMember,
Member(UniqueAddress(Address("akka", system.name), 2L), Set("dc-dc2"), Version.Zero) Member(UniqueAddress(Address("akka", system.name), 2L), Set("dc-dc2"), Version.Zero)
.copy(status = MemberStatus.Up))) .copy(status = MemberStatus.Up)))
probe.expectMsgType[Heartbeat].sequenceNr shouldEqual 1 awaitAssert {
probe.expectMsgType[Heartbeat].sequenceNr shouldEqual 2 underTest ! ReportStatus()
expectMsgType[CrossDcHeartbeatSender.MonitoringActive]
}
heartbeatProbe.expectMsgType[Heartbeat].sequenceNr shouldEqual 1
heartbeatProbe.expectMsgType[Heartbeat].sequenceNr shouldEqual 2
} }
} }
} }