io.keploy.advice.CustomGoogleResponseAdvice Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of agent Show documentation

This agent is used to mock infra calls for an API

There is a newer version: 1.4.7

package io.keploy.advice;

import io.keploy.regression.Mode;
import io.keploy.regression.context.Context;
import io.keploy.regression.context.Kcontext;
import io.keploy.googleMaps.CustomHttpResponses;
import net.bytebuddy.asm.Advice;
import okhttp3.Response;
import okio.BufferedSource;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import java.lang.reflect.Method;
import java.util.Objects;

/**
 * This class is used for intercepting method parseResponse of OkHttpPendingResult class and to record required data from
 * the method
 */
public class CustomGoogleResponseAdvice {

    /**
     * This method gets executed before method parseResponse of OkHttpPendingResult class. In record mode it saves the
     * data from the parseResponse method arguments and help in recording mocks/tests.
     */
    @Advice.OnMethodEnter
    static void enterMethods(@Advice.Origin Method method, @Advice.AllArguments Object[] obj) throws Exception {
        final Logger logger = LogManager.getLogger(CustomGoogleResponseAdvice.class);

        logger.debug("inside OnMethodEnterAdvice of CustomGoogleResponseAdvice for method: {}", method);
        Response response = (Response) obj[1];

        Kcontext kctx = Context.getCtx();

        if (kctx == null) {
            logger.debug("[CustomGoogleResponseAdvice] keploy context is null");
        } else if (kctx.getMode().getModeFromContext().equals(Mode.ModeType.MODE_RECORD)) {
            logger.debug("[CustomGoogleResponseAdvice] keploy mode: " + kctx.getMode());
            CustomHttpResponses.googleMapResponse = response;

            if (response.body() != null) {
                final BufferedSource source = Objects.requireNonNull(response.body()).source();
                source.request(Integer.MAX_VALUE);
                okio.ByteString snapshot = source.buffer().snapshot();
                String body = "";
                if (!response.body().contentType().type().contains("image")) {
                    logger.debug("not an image");
                    body = snapshot.utf8();
                }
                CustomHttpResponses.googleMapResBody = body;
            }
        } else if (kctx.getMode().getModeFromContext().equals(Mode.ModeType.MODE_OFF) || kctx.getMode().getModeFromContext().equals(Mode.ModeType.MODE_TEST)) {
            logger.debug("[CustomGoogleResponseAdvice] keploy mode: " + kctx.getMode());
        }
    }
}