66 lines
2.1 KiB
Scala
66 lines
2.1 KiB
Scala
|
/*
|
||
|
|
* Copyright (C) 2016 Lightbend Inc. <http://www.lightbend.com>
|
||
|
|
*/
|
||
|
|
package akka.remote.artery
|
||
|
|
|
||
|
|
import akka.actor.ActorRef
|
||
|
|
import akka.testkit.{ AkkaSpec, TestProbe }
|
||
|
|
import akka.util.ByteString
|
||
|
|
import scala.concurrent.duration._
|
||
|
|
|
||
|
|
class MetadataContainerSpec extends AkkaSpec {
|
||
|
|
|
||
|
|
"MetadataContainer" should {
|
||
|
|
"parse, given empty map" in {
|
||
|
|
val map = new MetadataMap[ByteString]
|
||
|
|
val container = new MetadataMapRendering(map)
|
||
|
|
|
||
|
|
val rendered = container.render()
|
||
|
|
val back = MetadataMapParsing.parseRaw(rendered.asByteBuffer)
|
||
|
|
|
||
|
|
map.toString() should ===(back.metadataMap.toString())
|
||
|
|
}
|
||
|
|
"parse, given 1 allocated in map" in {
|
||
|
|
val map = new MetadataMap[ByteString]
|
||
|
|
val container = new MetadataMapRendering(map)
|
||
|
|
map.set(1, ByteString("!!!"))
|
||
|
|
|
||
|
|
val rendered = container.render()
|
||
|
|
val back = MetadataMapParsing.parseRaw(rendered.asByteBuffer)
|
||
|
|
|
||
|
|
map.toString() should ===(back.metadataMap.toString())
|
||
|
|
}
|
||
|
|
|
||
|
|
"apply, given 3 allocated in map" in {
|
||
|
|
val map = new MetadataMap[ByteString]
|
||
|
|
val container = new MetadataMapRendering(map)
|
||
|
|
map.set(1, ByteString("!!!"))
|
||
|
|
map.set(10, ByteString("??????"))
|
||
|
|
map.set(31, ByteString("........."))
|
||
|
|
|
||
|
|
val p = TestProbe()
|
||
|
|
|
||
|
|
def testInstrument(id: Int): RemoteInstrument = {
|
||
|
|
new RemoteInstrument {
|
||
|
|
override def identifier: Byte = id.toByte
|
||
|
|
override def remoteMessageSent(recipient: ActorRef, message: Object, sender: ActorRef): ByteString = ???
|
||
|
|
override def remoteMessageReceived(recipient: ActorRef, message: Object, sender: ActorRef, metadata: ByteString): Unit =
|
||
|
|
p.ref ! s"${identifier}-${metadata.utf8String}"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
val instruments = Vector(
|
||
|
|
testInstrument(1), testInstrument(31), testInstrument(10)
|
||
|
|
)
|
||
|
|
|
||
|
|
val rendered = container.render()
|
||
|
|
|
||
|
|
val mockEnvelope = new ReusableInboundEnvelope
|
||
|
|
MetadataMapParsing.applyAllRemoteMessageReceivedRaw(instruments, mockEnvelope, rendered.asByteBuffer)
|
||
|
|
|
||
|
|
p.expectMsgAllOf("1-!!!", "10-??????", "31-.........")
|
||
|
|
p.expectNoMsg(100.millis)
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|