pekko/akka-docs/rst/scala/code/docs/http/scaladsl/MarshalSpec.scala

/*
 * Copyright (C) 2009-2016 Lightbend Inc. <http://www.lightbend.com>
 */

package docs.http.scaladsl

import akka.stream.testkit.AkkaSpec

class MarshalSpec extends AkkaSpec {

  "use marshal" in {
    import scala.concurrent.Await
    import scala.concurrent.duration._
    import akka.http.scaladsl.marshalling.Marshal
    import akka.http.scaladsl.model._

    import system.dispatcher // ExecutionContext

    val string = "Yeah"
    val entityFuture = Marshal(string).to[MessageEntity]
    val entity = Await.result(entityFuture, 1.second) // don't block in non-test code!
    entity.contentType shouldEqual ContentTypes.`text/plain(UTF-8)`

    val errorMsg = "Easy, pal!"
    val responseFuture = Marshal(420 -> errorMsg).to[HttpResponse]
    val response = Await.result(responseFuture, 1.second) // don't block in non-test code!
    response.status shouldEqual StatusCodes.EnhanceYourCalm
    response.entity.contentType shouldEqual ContentTypes.`text/plain(UTF-8)`

    val request = HttpRequest(headers = List(headers.Accept(MediaTypes.`application/json`)))
    val responseText = "Plaintext"
    val respFuture = Marshal(responseText).toResponseFor(request) // with content negotiation!
    a[Marshal.UnacceptableResponseContentTypeException] should be thrownBy {
      Await.result(respFuture, 1.second) // client requested JSON, we only have text/plain!
    }
  }

}