Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright 2011-2021 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.http.request.builder
import io.gatling.commons.validation._
import io.gatling.core.check.Validator
import io.gatling.core.session._
import io.gatling.core.session.el.El
import io.gatling.http.check.HttpCheck
import io.gatling.http.check.status.HttpStatusCheckBuilder
import io.gatling.http.check.status.HttpStatusCheckMaterializer
import io.gatling.http.client.Request
import io.gatling.http.client.oauth.{ ConsumerKey, RequestToken }
import io.gatling.http.client.proxy.ProxyServer
import io.gatling.http.client.realm.Realm
import io.gatling.http.client.sign.OAuthSignatureCalculator
import io.gatling.http.client.uri.Uri
import io.gatling.http.protocol.Proxy
import io.gatling.http.util.HttpHelper
import io.gatling.http.util.HttpHelper._
import com.softwaremill.quicklens._
import io.netty.handler.codec.http.{ HttpHeaderNames, HttpHeaderValues, HttpMethod }
object CommonAttributes {
def apply(requestName: Expression[String], method: HttpMethod, urlOrURI: Either[Expression[String], Uri]): CommonAttributes =
new CommonAttributes(
requestName,
method,
urlOrURI,
disableUrlEncoding = None,
queryParams = Nil,
headers = Map.empty,
realm = None,
virtualHost = None,
proxy = None,
signatureCalculator = None,
ignoreProtocolHeaders = false
)
}
final case class CommonAttributes(
requestName: Expression[String],
method: HttpMethod,
urlOrURI: Either[Expression[String], Uri],
disableUrlEncoding: Option[Boolean],
queryParams: List[HttpParam],
headers: Map[CharSequence, Expression[String]],
realm: Option[Expression[Realm]],
virtualHost: Option[Expression[String]],
proxy: Option[ProxyServer],
signatureCalculator: Option[(Request, Session) => Validation[_]],
ignoreProtocolHeaders: Boolean
)
object RequestBuilder {
/**
* This is the default HTTP check used to verify that the response status is 2XX
*/
val DefaultHttpCheck: HttpCheck = {
val okStatusValidator: Validator[Int] = new Validator[Int] {
override val name: String = "in([200, 209], 304)"
override def apply(actual: Option[Int], displayActualValue: Boolean): Validation[Option[Int]] = actual match {
case Some(actualValue) =>
if (HttpHelper.isOk(actualValue))
actual.success
else
s"found $actualValue".failure
case _ => Validator.FoundNothingFailure
}
}
HttpStatusCheckBuilder.find.validate(okStatusValidator.expressionSuccess).build(HttpStatusCheckMaterializer.Instance)
}
private val JsonHeaderValueExpression = HttpHeaderValues.APPLICATION_JSON.toString.expressionSuccess
private val XmlHeaderValueExpression = HttpHeaderValues.APPLICATION_XML.toString.expressionSuccess
val AcceptAllHeaderValueExpression: Expression[String] = "*/*".expressionSuccess
val AcceptCssHeaderValueExpression: Expression[String] = "text/css,*/*;q=0.1".expressionSuccess
def oauth1SignatureCalculator(
consumerKey: Expression[String],
clientSharedSecret: Expression[String],
token: Expression[String],
tokenSecret: Expression[String]
): (Request, Session) => Validation[_] =
(request, session) =>
for {
ck <- consumerKey(session)
css <- clientSharedSecret(session)
tk <- token(session)
tks <- tokenSecret(session)
} yield new OAuthSignatureCalculator(new ConsumerKey(ck, css), new RequestToken(tk, tks)).accept(request)
}
abstract class RequestBuilder[B <: RequestBuilder[B]] {
def commonAttributes: CommonAttributes
private[http] def newInstance(commonAttributes: CommonAttributes): B
def queryParam(name: Expression[String], value: Expression[Any]): B = queryParam(SimpleParam(name, value))
def multivaluedQueryParam(name: Expression[String], values: Expression[Seq[Any]]): B = queryParam(MultivaluedParam(name, values))
def queryParamSeq(seq: Seq[(String, Any)]): B = queryParamSeq(tupleSeq2SeqExpression(seq))
def queryParamSeq(seq: Expression[Seq[(String, Any)]]): B = queryParam(ParamSeq(seq))
def queryParamMap(map: Map[String, Any]): B = queryParamSeq(tupleSeq2SeqExpression(map.toSeq))
def queryParamMap(map: Expression[Map[String, Any]]): B = queryParam(ParamMap(map))
private def queryParam(param: HttpParam): B = newInstance(modify(commonAttributes)(_.queryParams)(_ ::: List(param)))
/**
* Adds a header to the request
*
* @param name the name of the header
* @param value the value of the header
*/
def header(name: CharSequence, value: Expression[String]): B = newInstance(modify(commonAttributes)(_.headers)(_ + (name -> value)))
/**
* Adds several headers to the request at the same time
*
* @param newHeaders a scala map containing the headers to add
*/
def headers(newHeaders: Map[_ <: CharSequence, String]): B =
newInstance(modify(commonAttributes)(_.headers)(_ ++ newHeaders.view.mapValues(_.el[String])))
@deprecated("Please use ignoreProtocolHeaders instead. Will be removed in 3.5.0", "3.4.0")
def ignoreDefaultHeaders: B = ignoreProtocolHeaders
def ignoreProtocolHeaders: B = newInstance(modify(commonAttributes)(_.ignoreProtocolHeaders).setTo(true))
/**
* Adds Accept and Content-Type headers to the request set with "application/json" values
*/
def asJson: B =
header(HttpHeaderNames.ACCEPT, RequestBuilder.JsonHeaderValueExpression).header(HttpHeaderNames.CONTENT_TYPE, RequestBuilder.JsonHeaderValueExpression)
/**
* Adds Accept and Content-Type headers to the request set with "application/xml" values
*/
def asXml: B =
header(HttpHeaderNames.ACCEPT, RequestBuilder.XmlHeaderValueExpression).header(HttpHeaderNames.CONTENT_TYPE, RequestBuilder.XmlHeaderValueExpression)
/**
* Adds BASIC authentication to the request
*
* @param username the username needed
* @param password the password needed
*/
def basicAuth(username: Expression[String], password: Expression[String]): B = authRealm(HttpHelper.buildBasicAuthRealm(username, password))
def digestAuth(username: Expression[String], password: Expression[String]): B = authRealm(HttpHelper.buildDigestAuthRealm(username, password))
private def authRealm(realm: Expression[Realm]): B = newInstance(modify(commonAttributes)(_.realm).setTo(Some(realm)))
/**
* @param virtualHost a virtual host to override default compute one
*/
def virtualHost(virtualHost: Expression[String]): B = newInstance(modify(commonAttributes)(_.virtualHost).setTo(Some(virtualHost)))
def disableUrlEncoding: B = newInstance(modify(commonAttributes)(_.disableUrlEncoding).setTo(Some(true)))
def proxy(httpProxy: Proxy): B = newInstance(modify(commonAttributes)(_.proxy).setTo(Some(httpProxy.proxyServer)))
def sign(calculator: (Request, Session) => Validation[_]): B = newInstance(modify(commonAttributes)(_.signatureCalculator).setTo(Some(calculator)))
def signWithOAuth1(consumerKey: Expression[String], clientSharedSecret: Expression[String], token: Expression[String], tokenSecret: Expression[String]): B =
sign(RequestBuilder.oauth1SignatureCalculator(consumerKey, clientSharedSecret, token, tokenSecret))
}
