pekko/akka-docs-dev/rst/scala/code/docs/http/server/DirectiveExamplesSpec.scala

/*
 * Copyright (C) 2009-2014 Typesafe Inc. <http://www.typesafe.com>
 */

package docs.http.server

import akka.http.scaladsl.server._
import Directives._
import akka.http.scaladsl.testkit.ScalatestRouteTest
import org.scalatest._

class DirectiveExamplesSpec extends RoutingSpec {

  "example-1, example-2" in {
    val route: Route =
      path("order" / IntNumber) { id =>
        get {
          complete {
            "Received GET request for order " + id
          }
        } ~
          put {
            complete {
              "Received PUT request for order " + id
            }
          }
      }
    verify(route) // hide
  }

  "example-3" in {
    def innerRoute(id: Int): Route =
      get {
        complete {
          "Received GET request for order " + id
        }
      } ~
        put {
          complete {
            "Received PUT request for order " + id
          }
        }

    val route: Route = path("order" / IntNumber) { id => innerRoute(id) }
    verify(route) // hide
  }

  "example-4" in {
    val route =
      path("order" / IntNumber) { id =>
        (get | put) { ctx =>
          ctx.complete("Received " + ctx.request.method.name + " request for order " + id)
        }
      }
    verify(route) // hide
  }

  "example-5" in {
    val getOrPut = get | put
    val route =
      path("order" / IntNumber) { id =>
        getOrPut { ctx =>
          ctx.complete("Received " + ctx.request.method.name + " request for order " + id)
        }
      }
    verify(route) // hide
  }

  "example-6" in {
    val getOrPut = get | put
    val route =
      (path("order" / IntNumber) & getOrPut) { id =>
        ctx =>
          ctx.complete("Received " + ctx.request.method.name + " request for order " + id)
      }
    verify(route) // hide
  }

  "example-7" in {
    val orderGetOrPut = path("order" / IntNumber) & (get | put)
    val route =
      orderGetOrPut { id =>
        ctx =>
          ctx.complete("Received " + ctx.request.method.name + " request for order " + id)
      }
    verify(route) // hide
  }

  "example-8" in {
    val orderGetOrPut = path("order" / IntNumber) & (get | put)
    val requestMethod = extract(_.request.method)
    val route =
      orderGetOrPut { id =>
        requestMethod { m =>
          complete("Received " + m.name + " request for order " + id)
        }
      }
    verify(route) // hide
  }

  "example-9" in {
    val orderGetOrPut = path("order" / IntNumber) & (get | put)
    val requestMethod = extract(_.request.method)
    val route =
      (orderGetOrPut & requestMethod) { (id, m) =>
        complete("Received " + m.name + " request for order " + id)
      }
    verify(route) // hide
  }

  "example-A" in {
    val orderGetOrPutMethod =
      path("order" / IntNumber) & (get | put) & extract(_.request.method)
    val route =
      orderGetOrPutMethod { (id, m) =>
        complete("Received " + m.name + " request for order " + id)
      }
    verify(route) // hide
  }

  def verify(route: Route) = {
    Get("/order/42") ~> route ~> check { responseAs[String] shouldEqual "Received GET request for order 42" }
    Put("/order/42") ~> route ~> check { responseAs[String] shouldEqual "Received PUT request for order 42" }
    Get("/") ~> route ~> check { handled shouldEqual false }
  }
}