io.gatling.core.assertion.AssertionBuilders.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2011-2018 GatlingCorp (https://gatling.io)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.gatling.core.assertion

import io.gatling.commons.stats.assertion._
import io.gatling.core.config.GatlingConfiguration

class AssertionWithPath(path: AssertionPath)(implicit configuration: GatlingConfiguration) {

  def responseTime = new AssertionWithPathAndTimeMetric(path, ResponseTime)
  def allRequests = new AssertionWithPathAndCountMetric(path, AllRequests)
  def failedRequests = new AssertionWithPathAndCountMetric(path, FailedRequests)
  def successfulRequests = new AssertionWithPathAndCountMetric(path, SuccessfulRequests)
  def requestsPerSec = new AssertionWithPathAndTarget[Int](path, MeanRequestsPerSecondTarget)
}

class AssertionWithPathAndTimeMetric(path: AssertionPath, metric: TimeMetric)(implicit configuration: GatlingConfiguration) {

  private def next(selection: TimeSelection) =
    new AssertionWithPathAndTarget[Int](path, TimeTarget(metric, selection))

  def min = next(Min)
  def max = next(Max)
  def mean = next(Mean)
  def stdDev = next(StandardDeviation)
  def percentile1 = percentile(configuration.charting.indicators.percentile1)
  def percentile2 = percentile(configuration.charting.indicators.percentile2)
  def percentile3 = percentile(configuration.charting.indicators.percentile3)
  def percentile4 = percentile(configuration.charting.indicators.percentile4)
  def percentile(value: Double) = next(Percentiles(value))
}

class AssertionWithPathAndCountMetric(path: AssertionPath, metric: CountMetric) {

  def count = new AssertionWithPathAndTarget[Long](path, CountTarget(metric))
  def percent = new AssertionWithPathAndTarget[Double](path, PercentTarget(metric))
}

class AssertionWithPathAndTarget[T: Numeric](path: AssertionPath, target: Target) {

  def next(condition: Condition) =
    Assertion(path, target, condition)

  val numeric = implicitly[Numeric[T]]

  def lt(threshold: T): Assertion = next(Lt(numeric.toDouble(threshold)))
  def lte(threshold: T): Assertion = next(Lte(numeric.toDouble(threshold)))
  def gt(threshold: T): Assertion = next(Gt(numeric.toDouble(threshold)))
  def gte(threshold: T): Assertion = next(Gte(numeric.toDouble(threshold)))
  def between(min: T, max: T, inclusive: Boolean = true): Assertion = next(Between(numeric.toDouble(min), numeric.toDouble(max), inclusive))
  def is(value: T): Assertion = next(Is(numeric.toDouble(value)))
  def in(set: Set[T]): Assertion = next(In(set.map(numeric.toDouble).toList))
  def in(values: T*): Assertion = in(values.toSet)
}