Add GCF integration tests
Add Integration Tests for GcfSpringBootHttpRequestHandler2 fix up fix build cleanup after merge Added process-based server integration test support some more refactoring remove unneeded maven deps address Dmitry and Dans feedback
This commit is contained in:
@@ -42,6 +42,11 @@
|
|||||||
<artifactId>spring-boot-starter-test</artifactId>
|
<artifactId>spring-boot-starter-test</artifactId>
|
||||||
<scope>test</scope>
|
<scope>test</scope>
|
||||||
</dependency>
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.springframework</groupId>
|
||||||
|
<artifactId>spring-web</artifactId>
|
||||||
|
<scope>test</scope>
|
||||||
|
</dependency>
|
||||||
<dependency>
|
<dependency>
|
||||||
<groupId>com.google.cloud.functions.invoker</groupId>
|
<groupId>com.google.cloud.functions.invoker</groupId>
|
||||||
<artifactId>java-function-invoker</artifactId>
|
<artifactId>java-function-invoker</artifactId>
|
||||||
|
|||||||
@@ -0,0 +1,102 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2012-2020 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.gcloud.integration;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.util.function.Function;
|
||||||
|
|
||||||
|
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 static org.springframework.cloud.function.adapter.gcloud.integration.LocalServerTestSupport.verify;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Integration tests for GCF Http Functions.
|
||||||
|
*
|
||||||
|
* @author Daniel Zou
|
||||||
|
* @author Mike Eltsufin
|
||||||
|
*/
|
||||||
|
public class FunctionInvokerIntegrationTests {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSingular() {
|
||||||
|
verify(CloudFunctionMainSingular.class, null, "hello", "HELLO");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testUppercase() throws InterruptedException, IOException {
|
||||||
|
verify(CloudFunctionMain.class, "uppercase", "hello", "HELLO");
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testFooBar() {
|
||||||
|
verify(CloudFunctionMain.class, "foobar", new Foo("Hi"), new Bar("Hi"));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Configuration
|
||||||
|
@Import({ ContextFunctionCatalogAutoConfiguration.class })
|
||||||
|
static class CloudFunctionMainSingular {
|
||||||
|
|
||||||
|
@Bean
|
||||||
|
Function<String, String> uppercase() {
|
||||||
|
return input -> input.toUpperCase();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Configuration
|
||||||
|
@Import({ ContextFunctionCatalogAutoConfiguration.class })
|
||||||
|
static class CloudFunctionMain {
|
||||||
|
|
||||||
|
@Bean
|
||||||
|
Function<String, String> uppercase() {
|
||||||
|
return input -> input.toUpperCase();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Bean
|
||||||
|
Function<Foo, Bar> foobar() {
|
||||||
|
return input -> new Bar(input.value);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private static class Foo {
|
||||||
|
|
||||||
|
String value;
|
||||||
|
|
||||||
|
Foo(String value) {
|
||||||
|
this.value = value;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private static class Bar {
|
||||||
|
|
||||||
|
String value;
|
||||||
|
|
||||||
|
Bar(String value) {
|
||||||
|
this.value = value;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@@ -0,0 +1,190 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2012-2020 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.gcloud.integration;
|
||||||
|
|
||||||
|
import java.io.BufferedReader;
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.io.InputStreamReader;
|
||||||
|
import java.io.UncheckedIOException;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Arrays;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.concurrent.CountDownLatch;
|
||||||
|
import java.util.concurrent.ExecutorService;
|
||||||
|
import java.util.concurrent.Executors;
|
||||||
|
import java.util.concurrent.Future;
|
||||||
|
import java.util.concurrent.TimeUnit;
|
||||||
|
import java.util.concurrent.atomic.AtomicInteger;
|
||||||
|
|
||||||
|
import com.google.cloud.functions.invoker.runner.Invoker;
|
||||||
|
import com.google.gson.Gson;
|
||||||
|
|
||||||
|
import org.springframework.boot.test.web.client.TestRestTemplate;
|
||||||
|
import org.springframework.cloud.function.adapter.gcloud.FunctionInvoker;
|
||||||
|
import org.springframework.http.HttpEntity;
|
||||||
|
import org.springframework.http.HttpHeaders;
|
||||||
|
import org.springframework.http.ResponseEntity;
|
||||||
|
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test support class for running tests on the local Cloud Function server.
|
||||||
|
*
|
||||||
|
* @author Daniel Zou
|
||||||
|
* @author Mike Eltsufin
|
||||||
|
*/
|
||||||
|
public class LocalServerTestSupport {
|
||||||
|
|
||||||
|
private static final Gson gson = new Gson();
|
||||||
|
|
||||||
|
private static final ExecutorService EXECUTOR = Executors.newCachedThreadPool();
|
||||||
|
|
||||||
|
private static final String SERVER_READY_STRING = "Started ServerConnector";
|
||||||
|
|
||||||
|
private static AtomicInteger nextPort = new AtomicInteger(8080);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Starts up the Cloud Function Server and executes the test
|
||||||
|
*/
|
||||||
|
public static <I, O> void verify(Class mainClass, String function, I input, O expectedOutput) {
|
||||||
|
try (ServerProcess serverProcess = LocalServerTestSupport.startServer(mainClass, function)) {
|
||||||
|
TestRestTemplate testRestTemplate = new TestRestTemplate();
|
||||||
|
|
||||||
|
HttpHeaders headers = new HttpHeaders();
|
||||||
|
|
||||||
|
ResponseEntity<String> response = testRestTemplate.postForEntity(
|
||||||
|
"http://localhost:" + serverProcess.getPort(), new HttpEntity<>(gson.toJson(input), headers),
|
||||||
|
String.class);
|
||||||
|
|
||||||
|
assertThat(response.getBody()).isEqualTo(gson.toJson(expectedOutput));
|
||||||
|
}
|
||||||
|
catch (Exception e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static ServerProcess startServer(Class<?> springApplicationMainClass, String function)
|
||||||
|
throws InterruptedException, IOException {
|
||||||
|
int port = nextPort.getAndIncrement();
|
||||||
|
|
||||||
|
String signatureType = "http";
|
||||||
|
String target = FunctionInvoker.class.getCanonicalName();
|
||||||
|
|
||||||
|
File javaHome = new File(System.getProperty("java.home"));
|
||||||
|
assertThat(javaHome.exists()).isTrue();
|
||||||
|
File javaBin = new File(javaHome, "bin");
|
||||||
|
File javaCommand = new File(javaBin, "java");
|
||||||
|
assertThat(javaCommand.exists()).isTrue();
|
||||||
|
String myClassPath = System.getProperty("java.class.path");
|
||||||
|
assertThat(myClassPath).isNotNull();
|
||||||
|
|
||||||
|
List<String> command = new ArrayList<>();
|
||||||
|
command.addAll(Arrays.asList(javaCommand.toString(), "-classpath", myClassPath, Invoker.class.getName()));
|
||||||
|
|
||||||
|
ProcessBuilder processBuilder = new ProcessBuilder().command(command).redirectErrorStream(true);
|
||||||
|
Map<String, String> environment = new HashMap<>();
|
||||||
|
environment.put("PORT", String.valueOf(port));
|
||||||
|
environment.put("K_SERVICE", "test-function");
|
||||||
|
environment.put("FUNCTION_SIGNATURE_TYPE", signatureType);
|
||||||
|
environment.put("FUNCTION_TARGET", target);
|
||||||
|
environment.put("MAIN_CLASS", springApplicationMainClass.getCanonicalName());
|
||||||
|
if (function != null) {
|
||||||
|
environment.put("spring.cloud.function.definition", function);
|
||||||
|
}
|
||||||
|
processBuilder.environment().putAll(environment);
|
||||||
|
Process serverProcess = processBuilder.start();
|
||||||
|
CountDownLatch ready = new CountDownLatch(1);
|
||||||
|
StringBuilder output = new StringBuilder();
|
||||||
|
Future<?> outputMonitorResult = EXECUTOR
|
||||||
|
.submit(() -> monitorOutput(serverProcess.getInputStream(), ready, output));
|
||||||
|
boolean serverReady = ready.await(5, TimeUnit.SECONDS);
|
||||||
|
if (!serverReady) {
|
||||||
|
serverProcess.destroy();
|
||||||
|
throw new AssertionError("Server never became ready");
|
||||||
|
}
|
||||||
|
return new ServerProcess(serverProcess, outputMonitorResult, output, port);
|
||||||
|
}
|
||||||
|
|
||||||
|
private static void monitorOutput(InputStream processOutput, CountDownLatch ready, StringBuilder output) {
|
||||||
|
try (BufferedReader reader = new BufferedReader(new InputStreamReader(processOutput))) {
|
||||||
|
String line;
|
||||||
|
while ((line = reader.readLine()) != null) {
|
||||||
|
if (line.contains(SERVER_READY_STRING)) {
|
||||||
|
ready.countDown();
|
||||||
|
}
|
||||||
|
System.out.println(line);
|
||||||
|
synchronized (output) {
|
||||||
|
output.append(line).append('\n');
|
||||||
|
}
|
||||||
|
if (line.contains("WARNING")) {
|
||||||
|
throw new AssertionError("Found warning in server output:\n" + line);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (IOException e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
throw new UncheckedIOException(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static class ServerProcess implements AutoCloseable {
|
||||||
|
|
||||||
|
private final Process process;
|
||||||
|
|
||||||
|
private final Future<?> outputMonitorResult;
|
||||||
|
|
||||||
|
private final StringBuilder output;
|
||||||
|
|
||||||
|
private final int port;
|
||||||
|
|
||||||
|
ServerProcess(Process process, Future<?> outputMonitorResult, StringBuilder output, int port) {
|
||||||
|
this.process = process;
|
||||||
|
this.outputMonitorResult = outputMonitorResult;
|
||||||
|
this.output = output;
|
||||||
|
this.port = port;
|
||||||
|
}
|
||||||
|
|
||||||
|
Process process() {
|
||||||
|
return process;
|
||||||
|
}
|
||||||
|
|
||||||
|
Future<?> outputMonitorResult() {
|
||||||
|
return outputMonitorResult;
|
||||||
|
}
|
||||||
|
|
||||||
|
String output() {
|
||||||
|
synchronized (output) {
|
||||||
|
return output.toString();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void close() {
|
||||||
|
process().destroy();
|
||||||
|
}
|
||||||
|
|
||||||
|
public int getPort() {
|
||||||
|
return port;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user