org.sisioh.trinity.test.ControllerUnitTestSupport.scala Maven / Gradle / Ivy

Go to download
package org.sisioh.trinity.test

import com.twitter.finagle.Service
import com.twitter.finagle.http.{Response => FinagleResponse, Request => FinagleRequest}
import com.twitter.util.Await
import org.jboss.netty.handler.codec.http.HttpMethod
import org.sisioh.trinity.domain.io.http.{Response => IOResponse, HeaderName}
import org.sisioh.trinity.domain.mvc.action.Action
import org.sisioh.trinity.domain.mvc.filter.Filter
import org.sisioh.trinity.domain.mvc.http.{Request, Response}
import org.sisioh.trinity.domain.mvc.routing.RoutingFilter
import org.sisioh.trinity.domain.mvc.server.ServiceBuilder
import org.sisioh.trinity.domain.mvc.{Environment, GlobalSettings}
import org.sisioh.trinity.util.DurationConverters._
import org.specs2.specification.Scope
import scala.concurrent.ExecutionContext
import scala.concurrent.duration.Duration
import scala.language.reflectiveCalls
import scala.util.Try

/**
 * 単体テストをサポートするためのトレイト。
 */
trait ControllerUnitTestSupport extends ControllerTestSupport {
  self =>

  case class UnitTestContext(routingFilter: RoutingFilter,
                             filters: Seq[Filter[Request, Response, Request, Response]] = Seq.empty,
                             server: TestServer = TestServer())
                            (implicit val executor: ExecutionContext)
    extends TestContext

  protected def buildRequest
  (method: HttpMethod,
   path: String,
   content: Option[Content],
   headers: Map[HeaderName, String], timeout: Duration)
  (implicit testContext: TestContext): Try[Response] = {
    implicit val executor = testContext.executor
    val UnitTestContext(routingFilter, filters, _) = testContext
    val request = newRequest(method, path, content, headers)
    val serviceBuilder = new ServiceBuilder {
      val globalSettings: Option[GlobalSettings[Request, Response]] = self.globalSettings

      def _buildService(action: Option[Action[Request, Response]] = None)
                       (implicit executor: ExecutionContext): Service[FinagleRequest, FinagleResponse] =
        buildService(Environment.Development, action)(executor)
    }
    serviceBuilder.registerFilter(routingFilter)
    serviceBuilder.registerFilters(filters)
    val service = serviceBuilder._buildService()
    Try {
      val finagleResponse = Await.result(service(request), timeout.toTwitter)
      val r = IOResponse(finagleResponse)
      Response(r)
    }
  }

  protected abstract class WithTestScope(implicit executor: ExecutionContext) extends Scope {
    val routingFilter: RoutingFilter
    implicit lazy val testContext = UnitTestContext(routingFilter)
  }

}