skinny.test.SkinnyTestSupport.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of skinny-test_2.11 Show documentation

skinny-test

The newest version!

package skinny.test

import skinny.controller.{ SkinnySessionInjectorController, SessionInjectorController }
import skinny.SkinnyEnv
import skinny.micro.test.SkinnyMicroTests
import skinny.logging.LoggerProvider

/**
 * Skinny framework testing support
 */
trait SkinnyTestSupport extends LoggerProvider { self: SkinnyMicroTests =>

  // set skinny.env as "test"
  System.setProperty(SkinnyEnv.PropertyKey, "test")

  /**
   * Session injector controller
   */
  object SessionInjector extends SessionInjectorController {
    put("/tmp/SkinnyTestSupport/session")(update)
  }
  addFilter(SessionInjector, "/tmp/SkinnyTestSupport/session")

  object SkinnySessionInjector extends SkinnySessionInjectorController {
    put("/tmp/SkinnyTestSupport/skinnySession")(update)
  }
  addFilter(SkinnySessionInjector, "/tmp/SkinnyTestSupport/skinnySession")

  /**
   * Provides a code block with injected session.
   */
  def withSession[A](attributes: (String, AnyRef)*)(action: => A): A = session {
    val params = attributes.map {
      case (key, obj) =>
        (key, SessionInjectorController.serialize(obj))
    }

    put("/tmp/SkinnyTestSupport/session", params)(action)
  }

  /**
   * Provides a code block with injected session.
   */
  def withSkinnySession[A](attributes: (String, AnyRef)*)(action: => A): A = session {
    val params = attributes.map {
      case (key, obj) =>
        (key, SessionInjectorController.serialize(obj))
    }

    put("/tmp/SkinnyTestSupport/skinnySession", params)(action)
  }

  /**
   * Logs response body when response status is unexpected.
   */
  def logBodyUnless(expectedStatus: Int) = {
    if (status != expectedStatus) {
      logger.warn(s"Unexpected status: ${status}, body: ${body}")
    }
  }

}