ru.tinkoff.gatling.amqp.checks.AmqpResponseCodeCheckBuilder.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gatling-amqp-plugin_2.13 Show documentation

Plugin for support performance testing with AMQP in Gatling

The newest version!

package ru.tinkoff.gatling.amqp.checks

import io.gatling.commons.validation._
import io.gatling.core.check.{Extractor, _}
import io.gatling.core.session._
import ru.tinkoff.gatling.amqp.AmqpCheck
import ru.tinkoff.gatling.amqp.request.AmqpProtocolMessage

object AmqpResponseCodeCheckBuilder {

  private type AmqpCheckMaterializer[T, P] = CheckMaterializer[T, AmqpCheck, AmqpProtocolMessage, P]

  class NotInMatcher[A](expected: Seq[A]) extends Matcher[A] {

    def name: String = expected.mkString("notIn(", ",", ")")

    protected def doMatch(actual: Option[A]): Validation[Option[A]] =
      actual match {
        case Some(actualValue) =>
          if (!expected.contains(actualValue))
            actual.success
          else
            s"found $actualValue".failure
        case _                 => Validator.FoundNothingFailure
      }
  }

  class ExtendedDefaultFindCheckBuilder[T, P, X](extractor: Expression[Extractor[P, X]], displayActualValue: Boolean)
      extends CheckBuilder.Find.Default[T, P, X](extractor, displayActualValue) {

    def notIn(expected: Expression[Seq[X]]): CheckBuilder[T, P] =
      new CheckBuilder.Final.Default[T, P, X](extractor, expected.map(new NotInMatcher(_)), displayActualValue, None, None)

    def notIn(expected: X*): CheckBuilder[T, P] = notIn(expected.expressionSuccess)
  }

  trait AmqpMessageCheckType

  val ResponseCode: ExtendedDefaultFindCheckBuilder[AmqpMessageCheckType, AmqpProtocolMessage, String] = {
    val rcExtractor = new Extractor[AmqpProtocolMessage, String] {
      val name                                                             = "responseCode"
      val arity                                                            = "find"
      def apply(prepared: AmqpProtocolMessage): Validation[Option[String]] = prepared.responseCode.success
    }.expressionSuccess

    new ExtendedDefaultFindCheckBuilder(rcExtractor, displayActualValue = true)
  }
}