GH-317, Implement HTTP request-specific handler for Azure adapter
* Handler extends AzureSpringBootRequestHandler with types HttpRequestMessage and HttpResponseMessage (which are required by Azure Functions for HTTP-triggered functions) * Handler extracts request path, query string parameters as well as the HTTP method from the HttpRequestMessage object and forwards them to the function as message headers * Handler creates an HttpResponseMessage with the response coming from the called function * Created unit tests accordingly GH-317, Implement HTTP request-specific handler for Azure adapter * Handler extends AzureSpringBootRequestHandler with types HttpRequestMessage and HttpResponseMessage (which are required by Azure Functions for HTTP-triggered functions) * Handler extracts request path, query string parameters as well as the HTTP method from the HttpRequestMessage object and forwards them to the function as message headers * Handler creates an HttpResponseMessage with the response coming from the called function * Created unit tests accordingly GH-317, Implement HTTP request-specific handler for Azure adapter * Handler extends AzureSpringBootRequestHandler with types HttpRequestMessage and HttpResponseMessage (which are required by Azure Functions for HTTP-triggered functions) * Handler extracts request path, query string parameters as well as the HTTP method from the HttpRequestMessage object and forwards them to the function as message headers * Handler creates an HttpResponseMessage with the response coming from the called function * Created unit tests accordingly
This commit is contained in:
Markus Gulden
committed by
Oleg Zhurakousky
Oleg Zhurakousky
parent
09053d3f59
commit
27344d7098
@@ -0,0 +1,288 @@
|
||||
/*
|
||||
* Copyright 2012-2019 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.cloud.function.adapter.azure;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.URI;
|
||||
import java.net.URISyntaxException;
|
||||
import java.util.Collections;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.util.function.Consumer;
|
||||
import java.util.function.Function;
|
||||
|
||||
import com.microsoft.azure.functions.HttpMethod;
|
||||
import com.microsoft.azure.functions.HttpRequestMessage;
|
||||
import com.microsoft.azure.functions.HttpResponseMessage;
|
||||
import com.microsoft.azure.functions.HttpResponseMessage.Builder;
|
||||
import com.microsoft.azure.functions.HttpStatus;
|
||||
import com.microsoft.azure.functions.HttpStatusType;
|
||||
|
||||
import org.junit.After;
|
||||
import org.junit.Test;
|
||||
import org.springframework.cloud.function.context.config.ContextFunctionCatalogAutoConfiguration;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.context.annotation.Import;
|
||||
import org.springframework.messaging.Message;
|
||||
import org.springframework.messaging.support.GenericMessage;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
|
||||
/**
|
||||
* @author Markus Gulden
|
||||
*/
|
||||
public class AzureSpringBootHttpRequestHandlerTests {
|
||||
|
||||
private AzureSpringBootHttpRequestHandler<?> handler = null;
|
||||
|
||||
<I> AzureSpringBootHttpRequestHandler<I> handler(Class<?> config) {
|
||||
AzureSpringBootHttpRequestHandler<I> handler = new AzureSpringBootHttpRequestHandler<I>(
|
||||
config);
|
||||
this.handler = handler;
|
||||
return handler;
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testWithBody() {
|
||||
AzureSpringBootHttpRequestHandler<Foo> handler = handler(
|
||||
FunctionMessageBodyConfig.class);
|
||||
HttpRequestMessageStub<Foo> request = new HttpRequestMessageStub<Foo>();
|
||||
request.setBody(new Foo("foo"));
|
||||
|
||||
HttpResponseMessage response = handler.handleRequest(request,
|
||||
new TestExecutionContext("uppercase"));
|
||||
|
||||
assertThat(response.getBody()).isInstanceOf(Bar.class);
|
||||
assertThat(response.getStatusCode()).isEqualTo(200);
|
||||
Bar body = (Bar) response.getBody();
|
||||
assertThat(body.getValue()).isEqualTo("FOO");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testWithRequestParameters() throws URISyntaxException {
|
||||
AzureSpringBootHttpRequestHandler<Foo> handler = handler(
|
||||
FunctionMessageEchoReqParametersConfig.class);
|
||||
HttpRequestMessageStub<Foo> request = new HttpRequestMessageStub<Foo>();
|
||||
request.setUri(new URI("http://localhost:8080/pathValue"));
|
||||
request.setHeaders(Collections.singletonMap("test-header", "headerValue"));
|
||||
request.setQueryParameters(Collections.singletonMap("query", "queryValue"));
|
||||
request.setHttpMethod(HttpMethod.GET);
|
||||
|
||||
HttpResponseMessage response = handler.handleRequest(request,
|
||||
new TestExecutionContext("uppercase"));
|
||||
|
||||
assertThat(response.getStatusCode()).isEqualTo(200);
|
||||
assertThat(response.getHeader("path")).isEqualTo("/pathValue");
|
||||
assertThat(response.getHeader("query")).isEqualTo("queryValue");
|
||||
assertThat(response.getHeader("test-header")).isEqualTo("headerValue");
|
||||
Bar body = (Bar) response.getBody();
|
||||
assertThat(body.getValue()).isEqualTo("body");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testWithEmptyBody() {
|
||||
AzureSpringBootHttpRequestHandler<Foo> handler = handler(
|
||||
FunctionMessageConsumerConfig.class);
|
||||
HttpRequestMessageStub<Foo> request = new HttpRequestMessageStub<Foo>();
|
||||
|
||||
HttpResponseMessage response = handler.handleRequest(request,
|
||||
new TestExecutionContext("uppercase"));
|
||||
|
||||
assertThat(response.getStatusCode()).isEqualTo(200);
|
||||
Bar body = (Bar) response.getBody();
|
||||
assertThat(body).isNull();
|
||||
}
|
||||
|
||||
@After
|
||||
public void close() throws IOException {
|
||||
if (this.handler != null) {
|
||||
this.handler.close();
|
||||
}
|
||||
}
|
||||
|
||||
@Configuration
|
||||
@Import({ ContextFunctionCatalogAutoConfiguration.class })
|
||||
protected static class FunctionMessageBodyConfig {
|
||||
|
||||
@Bean
|
||||
public Function<Message<Foo>, Message<Bar>> function() {
|
||||
return (foo -> {
|
||||
Map<String, Object> headers = new HashMap<>();
|
||||
return new GenericMessage<>(
|
||||
new Bar(foo.getPayload().getValue().toUpperCase()), headers);
|
||||
});
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@Configuration
|
||||
@Import({ ContextFunctionCatalogAutoConfiguration.class })
|
||||
protected static class FunctionMessageEchoReqParametersConfig {
|
||||
|
||||
@Bean
|
||||
public Function<Message<Void>, Message<Bar>> function() {
|
||||
return (message -> {
|
||||
Map<String, Object> headers = new HashMap<>();
|
||||
headers.put("path", message.getHeaders().get("path"));
|
||||
headers.put("query", message.getHeaders().get("query"));
|
||||
headers.put("test-header", message.getHeaders().get("test-header"));
|
||||
headers.put("httpMethod", message.getHeaders().get("httpMethod"));
|
||||
return new GenericMessage<>(new Bar("body"), headers);
|
||||
});
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@Configuration
|
||||
@Import({ ContextFunctionCatalogAutoConfiguration.class })
|
||||
protected static class FunctionMessageConsumerConfig {
|
||||
|
||||
@Bean
|
||||
public Consumer<Message<Foo>> function() {
|
||||
return (foo -> {
|
||||
});
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
class HttpRequestMessageStub<I> implements HttpRequestMessage<I> {
|
||||
|
||||
private URI uri;
|
||||
private HttpMethod httpMethod;
|
||||
private Map<String, String> headers;
|
||||
private Map<String, String> queryParameters;
|
||||
private I body;
|
||||
|
||||
public void setUri(URI uri) {
|
||||
this.uri = uri;
|
||||
}
|
||||
|
||||
public void setHttpMethod(HttpMethod httpMethod) {
|
||||
this.httpMethod = httpMethod;
|
||||
}
|
||||
|
||||
public void setHeaders(Map<String, String> headers) {
|
||||
this.headers = headers;
|
||||
}
|
||||
|
||||
public void setQueryParameters(Map<String, String> queryParameters) {
|
||||
this.queryParameters = queryParameters;
|
||||
}
|
||||
|
||||
public void setBody(I body) {
|
||||
this.body = body;
|
||||
}
|
||||
|
||||
@Override
|
||||
public URI getUri() {
|
||||
return this.uri;
|
||||
}
|
||||
|
||||
@Override
|
||||
public HttpMethod getHttpMethod() {
|
||||
return this.httpMethod;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Map<String, String> getHeaders() {
|
||||
return this.headers;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Map<String, String> getQueryParameters() {
|
||||
return this.queryParameters;
|
||||
}
|
||||
|
||||
@Override
|
||||
public I getBody() {
|
||||
return this.body;
|
||||
}
|
||||
|
||||
@Override
|
||||
public HttpResponseMessage.Builder createResponseBuilder(HttpStatusType status) {
|
||||
return new BuilderStub().status(status);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Builder createResponseBuilder(HttpStatus status) {
|
||||
return new BuilderStub().status(status);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
class BuilderStub implements Builder {
|
||||
|
||||
private HttpStatusType status;
|
||||
private Map<String, String> headers = new HashMap<>();
|
||||
private Object body;
|
||||
|
||||
@Override
|
||||
public Builder status(HttpStatusType status) {
|
||||
this.status = status;
|
||||
return this;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Builder header(String key, String value) {
|
||||
headers.put(key, value);
|
||||
return this;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Builder body(Object body) {
|
||||
this.body = body;
|
||||
return this;
|
||||
}
|
||||
|
||||
@Override
|
||||
public HttpResponseMessage build() {
|
||||
return new HttpResponseMessageStub(this.status, this.headers, this.body);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
class HttpResponseMessageStub implements HttpResponseMessage {
|
||||
|
||||
private HttpStatusType status;
|
||||
private Map<String, String> headers = new HashMap<>();
|
||||
private Object body;
|
||||
|
||||
public HttpResponseMessageStub(HttpStatusType status, Map<String, String> headers,
|
||||
Object body) {
|
||||
this.status = status;
|
||||
this.headers = headers;
|
||||
this.body = body;
|
||||
}
|
||||
|
||||
@Override
|
||||
public HttpStatusType getStatus() {
|
||||
return this.status;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getHeader(String key) {
|
||||
return this.headers.get(key);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Object getBody() {
|
||||
return this.body;
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user