uk.gov.service.payments.commons.testing.pact.consumers.PayPactProviderRule Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of testing Show documentation
There is a newer version: 1.0.20241118085938
package uk.gov.service.payments.commons.testing.pact.consumers;

import au.com.dius.pact.consumer.junit.PactProviderRule;
import au.com.dius.pact.core.model.DefaultPactReader;
import au.com.dius.pact.core.model.FileSource;
import au.com.dius.pact.core.model.RequestResponsePact;
import org.junit.runner.Description;
import org.junit.runners.model.Statement;

import java.io.File;
import java.lang.reflect.Method;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;

import static com.google.common.io.Resources.getResource;
import static java.lang.String.format;
import static java.util.Arrays.stream;
import static uk.gov.service.payments.commons.testing.port.PortFactory.findFreePort;

public class PayPactProviderRule extends PactProviderRule {

    private String methodName;
    private List pactsToDelete = new ArrayList<>();

    public PayPactProviderRule(String provider, Object target) {
        super(provider, "localhost", findFreePort(), target);
    }

    @Override
    public Statement apply(final Statement base, final Description description) {
        this.methodName = description.getMethodName();
        return new Statement() {
            @Override
            public void evaluate() throws Throwable {
                try {
                    PayPactProviderRule.super.apply(base, description).evaluate();
                } finally {
                    after();
                }
            }
        };   
    }
    
    @Override
    protected Map getPacts(String fragment) {
        HashMap pacts = new HashMap<>();
        for (Method m : target.getClass().getMethods()) {
            if (m.getName().equals(methodName)) {
                Optional.ofNullable(m.getAnnotationsByType(Pacts.class)).ifPresent(pactsAnnotation ->
                        stream(pactsAnnotation).forEach(p -> stream(p.pacts()).forEach(
                                fileName -> {
                                    if (fileName.contains(provider)) {
                                        RequestResponsePact pact = (RequestResponsePact) DefaultPactReader.INSTANCE.loadPact(new FileSource<>(new File(getResource(format("pacts/%s.json", fileName)).getFile())));
                                        pacts.put(provider, pact);
                                    }
                                    if (!p.publish()) pactsToDelete.add(format("%s.json", fileName));
                                }
                        ))
                );
            }
        }
        return pacts;
    }

    @Override
    protected void after() {
        //by deleting from project_dir/target/pacts it won't be there for the pact:publish maven plugin to publish to the pact broker. Issue has been raised here https://github.com/DiUS/pact-jvm/issues/711
        pactsToDelete.stream().forEach(s -> {
            File file = new File(Paths.get("", "target", "pacts", s).toAbsolutePath().toString());
            file.delete();
        });
    }
}