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

Serialization support for wrapped stream refs (#27353)

Browse source
This commit is contained in:
Johan Andrén 2019-07-16 19:11:33 +02:00 • committed by Helena Edelson
parent cdd7976c26
commit 03b8c543fc
14 changed files with 304 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

41
akka-stream-tests/src/test/scala/akka/stream/scaladsl/StreamRefsSpec.scala
View file

@ -11,6 +11,8 @@ import akka.pattern._
import akka.stream.testkit.TestPublisher
import akka.stream.testkit.scaladsl._
import akka.stream._
import akka.stream.impl.streamref.SinkRefImpl
import akka.stream.impl.streamref.SourceRefImpl
import akka.testkit.{ AkkaSpec, ImplicitSender, TestKit, TestProbe }
import akka.util.ByteString
import com.typesafe.config._
@ -160,15 +162,20 @@ object StreamRefsSpec {
}
""").withFallback(ConfigFactory.load())
}
object SnitchActor {
def props(probe: ActorRef) = Props(new SnitchActor(probe))
}
class SnitchActor(probe: ActorRef) extends Actor {
def receive = {
case msg => probe ! msg
}
}
}
class StreamRefsSpec(config: Config) extends AkkaSpec(config) with ImplicitSender {
class StreamRefsSpec extends AkkaSpec(StreamRefsSpec.config()) with ImplicitSender {
import StreamRefsSpec._
def this() {
this(StreamRefsSpec.config())
}
val remoteSystem = ActorSystem("RemoteSystem", StreamRefsSpec.config())
implicit val mat = ActorMaterializer()
@ -408,4 +415,28 @@ class StreamRefsSpec(config: Config) extends AkkaSpec(config) with ImplicitSende
}
"The StreamRefResolver" must {
"serialize and deserialize SourceRefs" in {
val probe = TestProbe()
val ref = system.actorOf(StreamRefsSpec.SnitchActor.props(probe.ref))
val sourceRef = SourceRefImpl[String](ref)
val resolver = StreamRefResolver(system)
val result = resolver.resolveSourceRef(resolver.toSerializationFormat(sourceRef))
result.asInstanceOf[SourceRefImpl[String]].initialPartnerRef ! "ping"
probe.expectMsg("ping")
}
"serialize and deserialize SinkRefs" in {
val probe = TestProbe()
val ref = system.actorOf(StreamRefsSpec.SnitchActor.props(probe.ref))
val sinkRef = SinkRefImpl[String](ref)
val resolver = StreamRefResolver(system)
val result = resolver.resolveSinkRef(resolver.toSerializationFormat(sinkRef))
result.asInstanceOf[SinkRefImpl[String]].initialPartnerRef ! "ping"
probe.expectMsg("ping")
}
}
}