com.github.tomakehurst.wiremock.matching.UrlPattern Java Examples

/**
 * Make sure the plugin fails with too many redirects
 * @throws Exception if anything goes wrong
 */
@Test(expected = TaskExecutionException.class)
public void tooManyRedirects() throws Exception {
    UrlPattern up1 = urlPathEqualTo("/" + REDIRECT);
    redirectWireMockRule.stubFor(get(up1)
            .withQueryParam("r", matching("[0-9]+"))
            .willReturn(aResponse()
                    .withStatus(HttpServletResponse.SC_FOUND)
                    .withTransformer("redirect", "redirects", 51)));

    Download t = makeProjectAndTask();
    t.src(redirectWireMockRule.url(REDIRECT) + "?r=52");
    File dst = folder.newFile();
    t.dest(dst);
    t.execute();
}
 

/**
 * Tests if the plugin can handle one redirect
 * @throws Exception if anything goes wrong
 */
@Test
public void oneRedirect() throws Exception {
    UrlPattern up1 = urlEqualTo("/" + REDIRECT);
    wireMockRule.stubFor(get(up1)
            .willReturn(aResponse()
                    .withStatus(HttpServletResponse.SC_FOUND)
                    .withHeader("Location", wireMockRule.url(TEST_FILE_NAME))));

    UrlPattern up2 = urlEqualTo("/" + TEST_FILE_NAME);
    wireMockRule.stubFor(get(up2)
            .willReturn(aResponse()
                    .withBody(CONTENTS)));

    Download t = makeProjectAndTask();
    t.src(wireMockRule.url(REDIRECT));
    File dst = folder.newFile();
    t.dest(dst);
    t.execute();

    String dstContents = FileUtils.readFileToString(dst);
    assertEquals(CONTENTS, dstContents);

    wireMockRule.verify(1, getRequestedFor(up1));
    wireMockRule.verify(1, getRequestedFor(up2));
}
 

private static void stubRequest(String method, String endpoint, int status, Collection<Pair> headers, String body) {
	UrlPattern urlPattern = urlEqualTo(endpoint);
	MappingBuilder request = request(method, urlPattern);

	ResponseDefinitionBuilder response = aResponse().withStatus(status);

	HttpHeaders httpHeaders = new HttpHeaders();

	for (Pair header : headers) {
		String name = header.getO1();
		List<String> values = header.getO2();
		HttpHeader h = new HttpHeader(name, values);
		httpHeaders = httpHeaders.plus(h);
	}

	response.withHeaders(httpHeaders);

	if (body != null) {
		response.withBody(body);
	}

	stubFor(request.willReturn(response));
}
 

/**
 * Verify that a given request has been triggered.
 *
 * @param endpoint Request endpoint.
 * @param method Request method.
 */
static void assertUploadRequest(String endpoint, HttpMethod method, File file) {
	UrlPattern urlPattern = urlEqualTo(endpoint);
	RequestMethod rqMethod = new RequestMethod(method.name());
	RequestPatternBuilder rq = new RequestPatternBuilder(rqMethod, urlPattern)
		.withAllRequestBodyParts(new MultipartValuePatternBuilder()
			.withName(file.getName())
			.withBody(new BinaryEqualToPattern(TestUtils.readFile(file)))
		);

	WireMock.verify(1, rq);
}
 

/**
 * Tests if the plugin can handle circular redirects
 * @throws Exception if anything goes wrong
 */
@Test(expected = TaskExecutionException.class)
public void circularRedirect() throws Exception {
    UrlPattern up1 = urlPathEqualTo("/" + REDIRECT);
    wireMockRule.stubFor(get(up1)
            .willReturn(aResponse()
                    .withStatus(HttpServletResponse.SC_FOUND)
                    .withHeader("Location", "/" + REDIRECT)));

    Download t = makeProjectAndTask();
    t.src(wireMockRule.url(REDIRECT));
    File dst = folder.newFile();
    t.dest(dst);
    t.execute();
}
 

/**
 * Tests if the plugin can handle ten redirects
 * @throws Exception if anything goes wrong
 */
@Test
public void tenRedirect() throws Exception {
    UrlPattern up1 = urlPathEqualTo("/" + REDIRECT);
    redirectWireMockRule.stubFor(get(up1)
            .withQueryParam("r", matching("[0-9]+"))
            .willReturn(aResponse()
                    .withStatus(HttpServletResponse.SC_FOUND)
                    .withTransformer("redirect", "redirects", 10)));

    UrlPattern up2 = urlEqualTo("/" + TEST_FILE_NAME);
    redirectWireMockRule.stubFor(get(up2)
            .willReturn(aResponse()
                    .withBody(CONTENTS)));

    Download t = makeProjectAndTask();
    t.src(redirectWireMockRule.url(REDIRECT) + "?r=10");
    File dst = folder.newFile();
    t.dest(dst);
    t.execute();

    String dstContents = FileUtils.readFileToString(dst);
    assertEquals(CONTENTS, dstContents);

    verify(10, getRequestedFor(up1));
    verify(1, getRequestedFor(up2));
}
 

/**
 * Verify that a given request has been triggered.
 *
 * @param endpoint Request endpoint.
 * @param method Request method.
 * @param cookies Cookies sent in HTTP request.
 */
static void assertRequestWithCookies(String endpoint, HttpMethod method, Iterable<Pair> cookies) {
	UrlPattern urlPattern = urlEqualTo(endpoint);
	RequestMethod rqMethod = new RequestMethod(method.name());
	RequestPatternBuilder rq = new RequestPatternBuilder(rqMethod, urlPattern);

	for (Pair cookie : cookies) {
		String cookieName = cookie.getO1();
		String cookieValue = cookie.getO2().get(0);
		rq.withCookie(cookieName, equalTo(cookieValue));
	}

	WireMock.verify(1, rq);
}
 

/**
 * Verify that a given request has been triggered.
 *
 * @param endpoint Request endpoint.
 * @param method Request method.
 * @param body Request body.
 */
static void assertRequestWithBody(String endpoint, HttpMethod method, String body) {
	UrlPattern urlPattern = urlEqualTo(endpoint);
	RequestMethod rqMethod = new RequestMethod(method.name());
	RequestPatternBuilder rq = new RequestPatternBuilder(rqMethod, urlPattern);
	rq.withRequestBody(equalTo(body));
	WireMock.verify(1, rq);
}
 

/**
 * Verify that a given request has been triggered.
 *
 * @param endpoint Request endpoint.
 * @param method Request method.
 */
static void assertRequest(String endpoint, HttpMethod method) {
	UrlPattern urlPattern = urlEqualTo(endpoint);
	RequestMethod rqMethod = new RequestMethod(method.name());
	RequestPatternBuilder rq = new RequestPatternBuilder(rqMethod, urlPattern);
	WireMock.verify(1, rq);
}
 

public void stubPost(UrlPattern url, String responseFile) throws Exception {
    wireMockServer.stubFor(post(url)
            .withHeader("Authorization", equalTo("Basic dXNlcjpwYXNz"))
            .willReturn(aResponse()
                    .withStatus(HttpStatus.SC_OK)
                    .withHeader("Content-Type", "application/json")
                    .withBody(IOUtils.toString(getClass().getResourceAsStream(responseFile)))));
}
 

public void stubGet(UrlPattern url, String responseFile) throws Exception {
    wireMockServer.stubFor(get(url)
            .willReturn(aResponse()
                    .withStatus(HttpStatus.SC_OK)
                    .withHeader("Content-Type", "application/json")
                    .withBody(IOUtils.toString(getClass().getResourceAsStream(responseFile)))));
}
 

private static UrlPattern getSchemaPattern(final Integer id) {
    return WireMock.urlPathEqualTo(SCHEMA_BY_ID_PATTERN + id);
}
 

public void stubGet(UrlPattern url, ResponseDefinitionBuilder responseDefinitionBuilder) {
    wireMockServer.stubFor(get(url)
            .withHeader("Authorization", equalTo("Basic dXNlcjpwYXNz"))
            .willReturn(responseDefinitionBuilder));
}
 

BasicMappingBuilder(RequestMethod method, UrlPattern urlPattern) {
	this.requestPatternBuilder = new RequestPatternBuilder(method, urlPattern);
}
 

private UrlPattern requestPattern(Operation operation) {
	return urlPathEqualTo(operation.getRequest().getUri().getPath());
}
 

private static UrlPattern getSubjectVersionsPattern(final String subject) {
    return WireMock.urlEqualTo(ALL_SUBJECT_PATTERN + "/" + subject + "/versions");
}
 

private static UrlPattern getSubjectPattern(final String subject) {
    return WireMock.urlEqualTo(ALL_SUBJECT_PATTERN + "/" + subject);
}
 

/**
 * Verify that a given request has been triggered.
 *
 * @param endpoint Request endpoint.
 * @param method Request method.
 * @param headerName Header name.
 * @param headerValue Header value.
 */
static void assertRequestWithHeader(String endpoint, HttpMethod method, String headerName, String headerValue) {
	UrlPattern urlPattern = urlEqualTo(endpoint);
	RequestMethod rqMethod = new RequestMethod(method.name());
	RequestPatternBuilder rq = new RequestPatternBuilder(rqMethod, urlPattern);
	rq.withHeader(headerName, equalTo(headerValue));
	WireMock.verify(1, rq);
}