65 lines
2.2 KiB
Scala
65 lines
2.2 KiB
Scala
/**
|
|
* Copyright (C) 2009-2015 Typesafe Inc. <http://www.typesafe.com>
|
|
*/
|
|
|
|
package docs.io
|
|
|
|
import java.net.{ Inet6Address, InetSocketAddress, NetworkInterface, StandardProtocolFamily }
|
|
import java.nio.channels.DatagramChannel
|
|
import scala.util.Random
|
|
|
|
import akka.actor.{ ActorSystem, Props }
|
|
import akka.io.Udp
|
|
import akka.testkit.TestKit
|
|
import org.scalatest.{ BeforeAndAfter, WordSpecLike }
|
|
|
|
import scala.collection.JavaConversions.enumerationAsScalaIterator
|
|
|
|
class ScalaUdpMulticastSpec extends TestKit(ActorSystem("ScalaUdpMulticastSpec")) with WordSpecLike with BeforeAndAfter {
|
|
|
|
"listener" should {
|
|
"send message back to sink" in {
|
|
|
|
def okInterfaceToUse(iface: NetworkInterface): Boolean = {
|
|
iface.getInetAddresses.exists(_.isInstanceOf[Inet6Address]) &&
|
|
// awdl0 is a special interface on OSX that we cannot use
|
|
iface.getDisplayName != "awdl0" &&
|
|
// we do not want to use virtual docker interfaces
|
|
!iface.getDisplayName.contains("docker")
|
|
}
|
|
val Some(ipv6Iface) = NetworkInterface.getNetworkInterfaces.find(okInterfaceToUse)
|
|
|
|
// host assigned link local multicast address http://tools.ietf.org/html/rfc3307#section-4.3.2
|
|
// generate a random 32 bit multicast address with the high order bit set
|
|
val randomAddress: String = (Random.nextInt().abs.toLong | (1L << 31)).toHexString.toUpperCase
|
|
val group = randomAddress.grouped(4).mkString("FF02::", ":", "")
|
|
val port = TestUtils.temporaryUdpIpv6Port(ipv6Iface)
|
|
val msg = "ohi"
|
|
val sink = testActor
|
|
val iface = ipv6Iface.getName
|
|
|
|
val listener = system.actorOf(Props(classOf[Listener], iface, group, port, sink))
|
|
expectMsgType[Udp.Bound]
|
|
val sender = system.actorOf(Props(classOf[Sender], iface, group, port, msg))
|
|
expectMsg(msg)
|
|
|
|
// unbind
|
|
system.stop(listener)
|
|
}
|
|
}
|
|
|
|
def afterAll(): Unit = {
|
|
TestKit.shutdownActorSystem(system)
|
|
}
|
|
|
|
}
|
|
|
|
object TestUtils {
|
|
def temporaryUdpIpv6Port(iface: NetworkInterface) = {
|
|
val serverSocket = DatagramChannel.open(StandardProtocolFamily.INET6).socket()
|
|
serverSocket.bind(new InetSocketAddress(iface.getInetAddresses.nextElement(), 0))
|
|
val port = serverSocket.getLocalPort
|
|
serverSocket.close()
|
|
port
|
|
}
|
|
}
|