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();
}