TestableSimulation.scala

trait TestableSimulation extends Simulation with FlatSpecLike with BeforeAndAfter {

  val props = scala.collection.mutable.Map.empty[String, String]
  for { p <- sys.props if (p._1.startsWith("gatling")) } yield { props += p }
  GatlingConfiguration.setUp(props)

  val simClass = getClass.asInstanceOf[Class[io.gatling.core.scenario.Simulation]]
  val cleanSimClassName = simClass.getSimpleName.clean
  val outputDirectoryName = configuration.core.outputDirectoryBaseName.getOrElse(cleanSimClassName)
  var selection = new Selection(simClass, outputDirectoryName, "no description")

  def describe(desc: String) = selection = new Selection(simClass, outputDirectoryName, desc)

  "Simulation" should "complete without assertion failures" in {
    val runOutcome = new Runner(selection).run
    val result = runOutcome._2 match {
      case sim: Simulation if !sim.assertions.isEmpty => {
        if (Assertion.assertThat(sim.assertions, DataReader.newInstance(outputDirectoryName))) GatlingStatusCodes.success
        else GatlingStatusCodes.assertionsFailed
      }
      case _ => GatlingStatusCodes.success
    }
    assertResult(GatlingStatusCodes.success, "simulation has assertion errors")(result)
  }
}