pekko/akka-actor-tests/src/test/scala/akka/io/UdpFFIntegrationSpec.scala

/**
 * Copyright (C) 2009-2013 Typesafe Inc. <http://www.typesafe.com>
 */
package akka.io

import akka.testkit.{ TestProbe, ImplicitSender, AkkaSpec }
import akka.io.UdpFF._
import TestUtils._
import akka.util.ByteString
import java.net.InetSocketAddress
import akka.actor.ActorRef

class UdpFFIntegrationSpec extends AkkaSpec("akka.loglevel = INFO") with ImplicitSender {

  def bindUdp(handler: ActorRef): (InetSocketAddress, ActorRef) = {
    val address = temporaryServerAddress()
    val commander = TestProbe()
    commander.send(IO(UdpFF), Bind(handler, address))
    commander.expectMsg(Bound)
    (address, commander.sender)
  }

  val simpleSender: ActorRef = {
    val commander = TestProbe()
    commander.send(IO(UdpFF), SimpleSender(Nil))
    commander.expectMsg(SimpleSendReady)
    commander.sender
  }

  "The UDP Fire-and-Forget implementation" must {

    "be able to send without binding" in {
      val (serverAddress, server) = bindUdp(testActor)
      val data = ByteString("To infinity and beyond!")
      simpleSender ! Send(data, serverAddress)

      expectMsgPF() {
        case Received(d, _) ⇒
          d must be === data
      }
    }

    "be able to send with binding" in {
      val (serverAddress, _) = bindUdp(testActor)
      val (clientAddress, client) = bindUdp(testActor)
      val data = ByteString("Fly little packet!")

      client ! Send(data, serverAddress)

      expectMsgPF() {
        case Received(d, a) ⇒
          d must be === data
          a must be === clientAddress
      }
    }

  }

}