48 lines
1.6 KiB
Scala
48 lines
1.6 KiB
Scala
|
/*
|
||
|
|
* Copyright (C) 2009-2014 Typesafe Inc. <http://www.typesafe.com>
|
||
|
|
*/
|
||
|
|
|
||
|
|
package akka.http.javadsl.testkit
|
||
|
|
|
||
|
|
import scala.annotation.varargs
|
||
|
|
import scala.concurrent.ExecutionContext
|
||
|
|
import scala.concurrent.duration._
|
||
|
|
import akka.stream.ActorFlowMaterializer
|
||
|
|
import akka.http.scaladsl.server
|
||
|
|
import akka.http.javadsl.model.HttpRequest
|
||
|
|
import akka.http.javadsl.server.{ Route, Directives }
|
||
|
|
import akka.http.impl.util.JavaMapping.Implicits._
|
||
|
|
import akka.http.impl.server.RouteImplementation
|
||
|
|
import akka.http.scaladsl.model.HttpResponse
|
||
|
|
import akka.http.scaladsl.server.{ RouteResult, RoutingSettings, Route ⇒ ScalaRoute }
|
||
|
|
import akka.actor.ActorSystem
|
||
|
|
import akka.event.NoLogging
|
||
|
|
import akka.http.impl.util._
|
||
|
|
|
||
|
|
abstract class RouteTest {
|
||
|
|
implicit def system: ActorSystem
|
||
|
|
implicit def materializer: ActorFlowMaterializer
|
||
|
|
implicit def executionContext: ExecutionContext = system.dispatcher
|
||
|
|
|
||
|
|
protected def awaitDuration: FiniteDuration = 500.millis
|
||
|
|
|
||
|
|
def runRoute(route: Route, request: HttpRequest): TestResponse = {
|
||
|
|
val scalaRoute = ScalaRoute.seal(RouteImplementation(route))
|
||
|
|
val result = scalaRoute(new server.RequestContextImpl(request.asScala, NoLogging, RoutingSettings(system)))
|
||
|
|
|
||
|
|
result.awaitResult(awaitDuration) match {
|
||
|
|
case RouteResult.Complete(response) ⇒ createTestResponse(response)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
@varargs
|
||
|
|
def testRoute(first: Route, others: Route*): TestRoute =
|
||
|
|
new TestRoute {
|
||
|
|
val underlying: Route = Directives.route(first, others: _*)
|
||
|
|
|
||
|
|
def run(request: HttpRequest): TestResponse = runRoute(underlying, request)
|
||
|
|
}
|
||
|
|
|
||
|
|
protected def createTestResponse(response: HttpResponse): TestResponse
|
||
|
|
}
|