lv.ctco.cukes.graphql.internal.PreprocessGraphQLRequestBody Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cukes-graphql Show documentation

The newest version!

package lv.ctco.cukes.graphql.internal;

import com.google.common.base.Strings;
import com.google.inject.Inject;
import com.google.inject.Singleton;
import io.restassured.response.Response;
import io.restassured.specification.RequestSpecification;
import lv.ctco.cukes.core.internal.templating.TemplatingEngine;
import lv.ctco.cukes.graphql.facade.GQLRequestFacade;
import lv.ctco.cukes.http.extension.CukesHttpPlugin;

@Singleton
public class PreprocessGraphQLRequestBody implements CukesHttpPlugin {

    @Inject
    GQLRequestFacade requestFacade;

    @Inject
    TemplatingEngine templatingEngine;

    @Override
    public void beforeRequest(RequestSpecification requestSpecification) {
        GraphQLRequest graphQLRequest = requestFacade.getGraphQLRequest();
        if (!Strings.isNullOrEmpty(graphQLRequest.getQuery())) {
            graphQLRequest.setQuery(templatingEngine.processBody(graphQLRequest.getQuery()));
        }
        if (!Strings.isNullOrEmpty(graphQLRequest.getVariables())) {
            graphQLRequest.setVariables(templatingEngine.processBody(graphQLRequest.getVariables()));
        }
        requestSpecification.body(graphQLRequest);
    }

    @Override
    public void afterRequest(Response response) {

    }
}