Spring/spring-framework
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

MockRestRequestMatchers can match query params

Browse Source 
Issue: SPR-14995
This commit is contained in:
schjan79
2016-12-07 23:56:30 +01:00
committed by Rossen Stoyanchev
parent ba47d06cbb
commit ba826f1026
2 changed files with 155 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
spring-test/src/main/java/org/springframework/test/web/client/match/MockRestRequestMatchers.java
View File

@@ -18,6 +18,7 @@ package org.springframework.test.web.client.match;
import java.io.IOException;
import java.net.URI;
import java.nio.charset.Charset;
import java.util.List;
import java.util.Map;
import javax.xml.xpath.XPathExpressionException;
@@ -31,7 +32,11 @@ import org.springframework.test.util.AssertionErrors;
import org.springframework.test.web.client.MockRestServiceServer;
import org.springframework.test.web.client.RequestMatcher;
import org.springframework.util.Assert;
import org.springframework.util.MultiValueMap;
import org.springframework.web.util.UriComponentsBuilder;
import org.springframework.web.util.UriUtils;
import static java.nio.charset.StandardCharsets.UTF_8;
import static org.hamcrest.MatcherAssert.*;
import static org.springframework.test.util.AssertionErrors.*;
@@ -153,9 +158,75 @@ public abstract class MockRestRequestMatchers {
}
private static void assertHeaderValueCount(final String name, HttpHeaders headers, int expectedCount) {
List<String> actualValues = headers.get(name);
AssertionErrors.assertTrue("Expected header <" + name + ">", actualValues != null);
AssertionErrors.assertTrue("Expected header <" + name + "> to have at least <" + expectedCount +
assertMultiValueMapCount("header", name, headers, expectedCount);
}
/**
* Assert request query parameter values with the given Hamcrest matcher.
*/
@SafeVarargs
public static RequestMatcher queryParameter(final String name, final Matcher<? super String>... matchers) {
return queryParameter(name, UTF_8, matchers);
}
/**
* Assert request query parameter values with the given Hamcrest matcher.
*/
@SafeVarargs
public static RequestMatcher queryParameter(final String name, final Charset charset,
final Matcher<? super String>... matchers) {
return new RequestMatcher() {
@Override
public void match(ClientHttpRequest request) {
MultiValueMap<String, String> queryParameters = getQueryParameters(request.getURI().toString(), charset);
assertQueryParameterValueCount(name, queryParameters, matchers.length);
for (int i = 0 ; i < matchers.length; i++) {
assertThat("Request query parameter", queryParameters.get(name).get(i), matchers[i]);
}
}
};
}
/**
* Assert request query parameter values.
*/
public static RequestMatcher queryParameter(final String name, final String... expectedValues) {
return queryParameter(name, UTF_8, expectedValues);
}
/**
* Assert request query parameter values.
*/
public static RequestMatcher queryParameter(final String name, final Charset charset, final String... expectedValues) {
return new RequestMatcher() {
@Override
public void match(ClientHttpRequest request) {
MultiValueMap<String, String> queryParameters = getQueryParameters(request.getURI().toString(), charset);
assertQueryParameterValueCount(name, queryParameters, expectedValues.length);
for (int i = 0 ; i < expectedValues.length; i++) {
assertEquals("Request query parameter + [" + name + "]",
expectedValues[i], queryParameters.get(name).get(i));
}
}
};
}
private static MultiValueMap<String, String> getQueryParameters(String uri, Charset charset) {
String decodeUri = UriUtils.decode(uri, charset);
MultiValueMap<String, String> queryParameters = UriComponentsBuilder.fromUriString(decodeUri)
.build().getQueryParams();
return queryParameters;
}
private static void assertQueryParameterValueCount(final String name, MultiValueMap<String, String> queryParameters, int expectedCount) {
assertMultiValueMapCount("query parameter", name, queryParameters, expectedCount);
}
private static void assertMultiValueMapCount(String type, final String name, MultiValueMap<String, String> multiValueMap, int expectedCount) {
List<String> actualValues = multiValueMap.get(name);
assertTrue("Expected " + type + " <" + name + ">", actualValues != null);
assertTrue("Expected " + type + " <" + name + "> to have at least <" + expectedCount +
"> values but found " + actualValues, expectedCount <= actualValues.size());
}