org.catools.ws.rest.CHttpClient Maven / Gradle / Ivy

Go to download
package org.catools.ws.rest;

import io.restassured.config.RestAssuredConfig;
import io.restassured.internal.print.RequestPrinter;
import lombok.Getter;
import lombok.Setter;
import lombok.experimental.Accessors;
import org.catools.common.collections.CList;
import org.catools.common.collections.CSet;
import org.catools.common.extensions.verify.CVerifier;
import org.catools.common.extensions.verify.CVerify;
import org.catools.common.tests.CTest;
import org.catools.ws.enums.CHttpRequestType;
import org.catools.ws.enums.CHttpStatusCode;
import org.catools.ws.model.CHttpRequest;
import org.catools.ws.model.CHttpResponse;
import org.catools.ws.utils.CRestAssuredUtil;
import org.slf4j.Logger;

import java.io.PrintStream;
import java.util.Map;
import java.util.function.BiConsumer;

import static io.restassured.filter.log.LogDetail.ALL;

@Getter
@Setter
@Accessors(chain = true)
public abstract class CHttpClient {
  protected final Logger logger;
  protected final CVerify verify;
  protected final T testInstance;

  private final CHttpRequest request;

  public CHttpClient(T testInstance, CHttpRequestType requestType, String targetURI) {
    this(testInstance, requestType, targetURI, null);
  }

  public CHttpClient(
      T testInstance, CHttpRequestType requestType, String targetURI, String targetPath) {
    this.request = new CHttpRequest(requestType, targetURI, targetPath);
    this.testInstance = testInstance;
    this.logger = testInstance.logger;
    this.verify = testInstance.verify;
  }

  public > R setEntity(Object obj) {
    this.request.setEntity(obj);
    return (R) this;
  }

  public CHttpResponse send() {
    CHttpResponse response =
        CRestAssuredUtil.send(getConfig(), request, getRequestLoggerFilterListener());
    logger.info("Response << {}", response.toString());
    return response;
  }

  public CHttpResponse send(CHttpStatusCode expectedCode) {
    return send(expectedCode, (BiConsumer) null);
  }

  public CHttpResponse send(
      CHttpStatusCode expectedCode, BiConsumer afterCall) {
    CHttpResponse response = send();
    CVerifier verifier = new CVerifier(testInstance);
    response.StatusCode.verifyEquals(verifier, expectedCode);
    if (afterCall != null) {
      afterCall.accept(response, verifier);
    }
    verifier.verify();
    return response;
  }

  public CHttpResponse send(CHttpStatusCode expectedCode, String entityKey, Object entityValue) {
    return send(expectedCode, entityKey, entityValue, null);
  }

  public CHttpResponse send(
      CHttpStatusCode expectedCode,
      String entityKey,
      Object entityValue,
      BiConsumer afterCall) {
    return send(expectedCode, new CList<>(Map.entry(entityKey, entityValue)), afterCall);
  }

  public CHttpResponse send(
      CHttpStatusCode expectedCode, CList> expectedEntityValues) {
    return send(expectedCode, expectedEntityValues, null);
  }

  public CHttpResponse send(
      CHttpStatusCode expectedCode,
      CList> expectedEntityValues,
      BiConsumer afterCall) {
    CHttpResponse response = send();
    CVerifier verifier = new CVerifier(testInstance);
    response.StatusCode.verifyEquals(verifier, expectedCode);
    CSet entrySet = new CSet(response.ContentMap.entrySet());
    for (Map.Entry entityValue : expectedEntityValues) {
      entrySet.verifyContains(verifier, entityValue, "ContentMap matches expected value.");
    }

    if (afterCall != null) {
      afterCall.accept(response, verifier);
    }
    verifier.verify();
    return response;
  }

  protected RestAssuredConfig getConfig() {
    return RestAssuredConfig.newConfig();
  }

  private CFilterListener getRequestLoggerFilterListener() {
    return (reqSpec, responseSpec, ctx) -> {
      logger.info(
          "Request >> {}",
          RequestPrinter.print(
              reqSpec,
              reqSpec.getMethod(),
              reqSpec.getURI(),
              ALL,
              new PrintStream(PrintStream.nullOutputStream()),
              true));
    };
  }

  public > C setUrlEncodingEnabled(boolean urlEncodingEnabled) {
    this.getRequest().setUrlEncodingEnabled(urlEncodingEnabled);
    return (C) this;
  }

  public abstract O process();
}