org.scalatra.test.EmbeddedJettyContainer.scala Maven / Gradle / Ivy

Go to download
package org.scalatra.test

import org.eclipse.jetty.server.{ Server, ServerConnector }
import org.scalatra.JettyCompat
import org.scalatra.test.EmbeddedJettyContainerCompat.{ configureServer, configureServletContextHandler }

trait EmbeddedJettyContainer extends JettyContainer {
  /**
   * Sets the port to listen on.  0 means listen on any available port.
   */
  def port: Int = 0

  /**
   * The port of the currently running Jetty.  May differ from port if port is 0.
   *
   * @return Some port if Jetty is currently listening, or None if it is not.
   */
  def localPort: Option[Int] = server.getConnectors collectFirst {
    case x: ServerConnector => x.getLocalPort
  }

  def contextPath = "/"

  lazy val server = new Server(port)

  lazy val servletContextHandler = {
    val handler = JettyCompat.createServletContextHandler(resourceBasePath)
    configureServletContextHandler(handler)
    handler.setContextPath(contextPath)
    handler
  }

  def start(): Unit = {
    configureServer(server)
    server.setHandler(servletContextHandler)
    server.start()
  }

  def stop(): Unit = server.stop()

  def baseUrl: String =
    server.getConnectors collectFirst {
      case conn: ServerConnector =>
        val host = Option(conn.getHost) getOrElse "localhost"
        val port = conn.getLocalPort
        require(port > 0, "The detected local port is < 1, that's not allowed")
        "http://%s:%d".format(host, port)
    } getOrElse sys.error("can't calculate base URL: no connector")
}