74 lines
2.7 KiB
Scala
74 lines
2.7 KiB
Scala
|
/**
|
||
|
|
* Copyright (C) 2015 Typesafe Inc. <http://www.typesafe.com>
|
||
|
|
*/
|
||
|
|
package akka.cluster.sharding.protobuf
|
||
|
|
|
||
|
|
import akka.actor.{ ExtendedActorSystem, Address }
|
||
|
|
import akka.testkit.AkkaSpec
|
||
|
|
import akka.actor.Props
|
||
|
|
import akka.cluster.sharding.ShardCoordinator
|
||
|
|
import akka.cluster.sharding.Shard
|
||
|
|
|
||
|
|
class ClusterShardingMessageSerializerSpec extends AkkaSpec {
|
||
|
|
import ShardCoordinator.Internal._
|
||
|
|
|
||
|
|
val serializer = new ClusterShardingMessageSerializer(system.asInstanceOf[ExtendedActorSystem])
|
||
|
|
|
||
|
|
val region1 = system.actorOf(Props.empty, "region1")
|
||
|
|
val region2 = system.actorOf(Props.empty, "region2")
|
||
|
|
val region3 = system.actorOf(Props.empty, "region3")
|
||
|
|
val regionProxy1 = system.actorOf(Props.empty, "regionProxy1")
|
||
|
|
val regionProxy2 = system.actorOf(Props.empty, "regionProxy2")
|
||
|
|
|
||
|
|
def checkSerialization(obj: AnyRef): Unit = {
|
||
|
|
val blob = serializer.toBinary(obj)
|
||
|
|
val ref = serializer.fromBinary(blob, serializer.manifest(obj))
|
||
|
|
ref should ===(obj)
|
||
|
|
}
|
||
|
|
|
||
|
|
"ClusterShardingMessageSerializer" must {
|
||
|
|
|
||
|
|
"be able to serializable ShardCoordinator snapshot State" in {
|
||
|
|
val state = State(
|
||
|
|
shards = Map("a" -> region1, "b" -> region2, "c" -> region2),
|
||
|
|
regions = Map(region1 -> Vector("a"), region2 -> Vector("b", "c"), region3 -> Vector.empty[String]),
|
||
|
|
regionProxies = Set(regionProxy1, regionProxy2),
|
||
|
|
unallocatedShards = Set("d"))
|
||
|
|
checkSerialization(state)
|
||
|
|
}
|
||
|
|
|
||
|
|
"be able to serializable ShardCoordinator domain events" in {
|
||
|
|
checkSerialization(ShardRegionRegistered(region1))
|
||
|
|
checkSerialization(ShardRegionProxyRegistered(regionProxy1))
|
||
|
|
checkSerialization(ShardRegionTerminated(region1))
|
||
|
|
checkSerialization(ShardRegionProxyTerminated(regionProxy1))
|
||
|
|
checkSerialization(ShardHomeAllocated("a", region1))
|
||
|
|
checkSerialization(ShardHomeDeallocated("a"))
|
||
|
|
}
|
||
|
|
|
||
|
|
"be able to serializable ShardCoordinator remote messages" in {
|
||
|
|
checkSerialization(Register(region1))
|
||
|
|
checkSerialization(RegisterProxy(regionProxy1))
|
||
|
|
checkSerialization(RegisterAck(region1))
|
||
|
|
checkSerialization(GetShardHome("a"))
|
||
|
|
checkSerialization(ShardHome("a", region1))
|
||
|
|
checkSerialization(HostShard("a"))
|
||
|
|
checkSerialization(ShardStarted("a"))
|
||
|
|
checkSerialization(BeginHandOff("a"))
|
||
|
|
checkSerialization(BeginHandOffAck("a"))
|
||
|
|
checkSerialization(HandOff("a"))
|
||
|
|
checkSerialization(ShardStopped("a"))
|
||
|
|
checkSerialization(GracefulShutdownReq(region1))
|
||
|
|
}
|
||
|
|
|
||
|
|
"be able to serializable PersistentShard snapshot state" in {
|
||
|
|
checkSerialization(Shard.State(Set("e1", "e2", "e3")))
|
||
|
|
}
|
||
|
|
|
||
|
|
"be able to serializable PersistentShard domain events" in {
|
||
|
|
checkSerialization(Shard.EntityStarted("e1"))
|
||
|
|
checkSerialization(Shard.EntityStopped("e1"))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|