com.github.hippoom.wiremock.contract.verifier.junit.MockMvcContractVerifier Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of wiremock-contract-verifier Show documentation

Your description of the lib

The newest version!

package com.github.hippoom.wiremock.contract.verifier.junit;

import static com.github.hippoom.wiremock.contract.verifier.mockmvc.JsonContentResultMatchers.toJsonCompareMode;
import static com.github.tomakehurst.wiremock.stubbing.StubMapping.buildFrom;
import static java.nio.file.Files.readAllBytes;

import com.github.hippoom.wiremock.contract.verifier.NoSuchContractException;
import com.github.hippoom.wiremock.contract.verifier.anntation.Contract;
import com.github.hippoom.wiremock.contract.verifier.mockmvc.JsonContentResultMatchers;
import com.github.hippoom.wiremock.contract.verifier.mockmvc.MockHttpServletRequestBuilderMapper;
import com.github.hippoom.wiremock.contract.verifier.mockmvc.ResultMatcherMapper;
import com.github.tomakehurst.wiremock.stubbing.StubMapping;

import java.io.IOException;
import java.nio.charset.Charset;
import java.nio.file.Paths;

import org.junit.rules.TestWatcher;
import org.junit.runner.Description;
import org.springframework.core.io.ClassPathResource;
import org.springframework.test.web.servlet.ResultMatcher;
import org.springframework.test.web.servlet.request.MockHttpServletRequestBuilder;

public class MockMvcContractVerifier extends TestWatcher {

    private final MockHttpServletRequestBuilderMapper requestBuilderMapper =
        new MockHttpServletRequestBuilderMapper();

    private ResultMatcherMapper resultMatcherMapper;

    private StubMapping stubMapping;

    @Override
    protected void starting(Description description) {
        Contract contract = description.getAnnotation(Contract.class);
        if (contract != null) {
            stubMapping = buildFrom(extractJsonFrom(contract));

            resultMatcherMapper = new ResultMatcherMapper(toJsonCompareMode(contract));
        }
    }

    private String extractJsonFrom(Contract contract) {
        ClassPathResource pathResource = new ClassPathResource(
            "contracts/" + contract.value());
        try {
            byte[] bytes = readAllBytes(Paths.get(pathResource.getURI()));
            return new String(bytes, Charset.forName("utf-8"));
        } catch (IOException err) {
            throw new NoSuchContractException(err.getMessage(), err);
        }
    }

    public MockHttpServletRequestBuilder requestPattern() {
        return this.requestBuilderMapper.from(stubMapping);
    }

    public ResultMatcher responseDefinition() {
        return this.resultMatcherMapper.from(stubMapping);
    }
}