0e4d41ad33
* Add scalafix plugin for jdk 9. * Add command alias sortImports. * Excludes some sources from SortImports. * Update SortImports to 0.4.0 * Sort imports with `sortImports` command.
87 lines
2.2 KiB
Scala
87 lines
2.2 KiB
Scala
/*
|
|
* Copyright (C) 2009-2020 Lightbend Inc. <https://www.lightbend.com>
|
|
*/
|
|
|
|
package akka.cluster.ddata
|
|
|
|
import com.typesafe.config.ConfigFactory
|
|
import org.scalatest.BeforeAndAfterAll
|
|
import org.scalatest.matchers.should.Matchers
|
|
import org.scalatest.wordspec.AnyWordSpecLike
|
|
|
|
import akka.actor.Actor
|
|
import akka.actor.ActorSystem
|
|
import akka.actor.Props
|
|
import akka.actor.Stash
|
|
import akka.testkit.ImplicitSender
|
|
import akka.testkit.TestKit
|
|
|
|
object LocalConcurrencySpec {
|
|
|
|
final case class Add(s: String)
|
|
|
|
object Updater {
|
|
val key = ORSetKey[String]("key")
|
|
}
|
|
|
|
class Updater extends Actor with Stash {
|
|
|
|
implicit val selfUniqueAddress: SelfUniqueAddress = DistributedData(context.system).selfUniqueAddress
|
|
|
|
val replicator = DistributedData(context.system).replicator
|
|
|
|
def receive = {
|
|
case s: String =>
|
|
val update = Replicator.Update(Updater.key, ORSet.empty[String], Replicator.WriteLocal)(_ :+ s)
|
|
replicator ! update
|
|
}
|
|
}
|
|
}
|
|
|
|
class LocalConcurrencySpec(_system: ActorSystem)
|
|
extends TestKit(_system)
|
|
with AnyWordSpecLike
|
|
with Matchers
|
|
with BeforeAndAfterAll
|
|
with ImplicitSender {
|
|
import LocalConcurrencySpec._
|
|
|
|
def this() =
|
|
this(
|
|
ActorSystem(
|
|
"LocalConcurrencySpec",
|
|
ConfigFactory.parseString("""
|
|
akka.actor.provider = "cluster"
|
|
akka.remote.classic.netty.tcp.port=0
|
|
akka.remote.artery.canonical.port = 0
|
|
""")))
|
|
|
|
override def afterAll(): Unit = {
|
|
shutdown(system)
|
|
}
|
|
|
|
val replicator = DistributedData(system).replicator
|
|
|
|
"Updates from same node" must {
|
|
|
|
"be possible to do from two actors" in {
|
|
val updater1 = system.actorOf(Props[Updater](), "updater1")
|
|
val updater2 = system.actorOf(Props[Updater](), "updater2")
|
|
|
|
val numMessages = 100
|
|
for (n <- 1 to numMessages) {
|
|
updater1 ! s"a$n"
|
|
updater2 ! s"b$n"
|
|
}
|
|
|
|
val expected = ((1 to numMessages).map("a" + _) ++ (1 to numMessages).map("b" + _)).toSet
|
|
awaitAssert {
|
|
replicator ! Replicator.Get(Updater.key, Replicator.ReadLocal)
|
|
val ORSet(elements) = expectMsgType[Replicator.GetSuccess[_]].get(Updater.key)
|
|
elements should be(expected)
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
}
|